From 62b3e36612504aa5fab4182541a240807aafde6e Mon Sep 17 00:00:00 2001
From: Jacob Lifshay <programmerjake@gmail.com>
Date: Tue, 9 Jul 2019 00:29:01 -0700
Subject: [PATCH] tracer: add PyPy support to get_var_name().

Fixes #141.
---
 .travis.yml      | 5 +++++
 nmigen/tracer.py | 7 ++++++-
 2 files changed, 11 insertions(+), 1 deletion(-)

diff --git a/.travis.yml b/.travis.yml
index fea7d80..22af903 100644
--- a/.travis.yml
+++ b/.travis.yml
@@ -3,6 +3,7 @@ language: python
 python:
   - "3.6"
   - "3.7"
+  - "pypy3"
 cache:
   directories:
     - "$HOME/.ccache"
@@ -23,3 +24,7 @@ script:
   - python setup.py develop
   - coverage run -m unittest discover
   - codecov
+matrix:
+  fast_finish: true
+  allow_failures:
+  - python: "pypy3"
diff --git a/nmigen/tracer.py b/nmigen/tracer.py
index 49c9650..eb8a71e 100644
--- 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
 
-- 
2.30.2