Add Support for Calls to External Class Instances (#61)

vitsalis · May 30, 2023 · bc4d41e · bc4d41e
1 parent 373751b
commit bc4d41e
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/pycg/processing/base.py b/pycg/processing/base.py
@@ -216,7 +216,10 @@ def decode_node(self, node):
                     return_ns = utils.join_ns(
                         called_def.get_ns(), utils.constants.RETURN_NAME
                     )
-                elif called_def.get_type() == utils.constants.CLS_DEF:
+                elif (
+                    called_def.get_type() == utils.constants.CLS_DEF
+                    or called_def.get_type() == utils.constants.EXT_DEF
+                ):
                     return_ns = called_def.get_ns()
                 defi = self.def_manager.get(return_ns)
                 if defi: