tracer: add PyPy support to get_var_name().

Fixes #141.
2019-07-09 00:29:01 -07:00 · 2019-07-09 00:29:01 -07:00 · 62b3e36612
commit 62b3e36612
parent 367ad5aee7
2 changed files with 11 additions and 1 deletions
--- a/nmigen/tracer.py
+++ b/nmigen/tracer.py
@ -20,7 +20,12 @@ def get_var_name(depth=2, default=_raise_exception):

    code = frame.f_code
    call_index = frame.f_lasti
-    call_opc   = opname[code.co_code[call_index]]
+    while True:
+        call_opc   = opname[code.co_code[call_index]]
+        if call_opc in ("EXTENDED_ARG"):
+            call_index += 2
+        else:
+            break
    if call_opc not in ("CALL_FUNCTION", "CALL_FUNCTION_KW", "CALL_FUNCTION_EX", "CALL_METHOD"):
        return None