should match to old

mkorpela · Oct 13, 2022 · 8359651 · 8359651
1 parent b3f90ff
commit 8359651
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 13 deletions.
diff --git a/overrides/overrides.py b/overrides/overrides.py
@@ -146,19 +146,23 @@ def _get_base_classes(frame, namespace):
 def _get_base_class_names(frame) -> List[List[str]]:
     """Get baseclass names from the code object"""
     extends: List[Tuple[str, str]] = []
+    add_last_step = True
     for instruction in dis.get_instructions(frame.f_code):
         if instruction.offset > frame.f_lasti:
             break
         if instruction.opcode not in dis.hasname:
             continue
+        if not add_last_step:
+            extends = []
+            add_last_step = True
         if instruction.opname == "LOAD_NAME":
             extends.append(("name", instruction.argval))
         elif instruction.opname == "LOAD_ATTR":
             extends.append(("attr", instruction.argval))
         elif instruction.opname == "LOAD_GLOBAL":
             extends.append(("name", instruction.argval))
         else:
-            extends = []
+            add_last_step = False
 
     items: List[List[str]] = []
     previous_item: List[str] = []

diff --git a/tests/test_get_base_class_names.py b/tests/test_get_base_class_names.py