Add class def

nline · Jan 2, 2024 · be897e5 · be897e5
1 parent 4e3a48d
commit be897e5
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pycg/processing/cgprocessor.py b/pycg/processing/cgprocessor.py
@@ -178,6 +178,9 @@ def create_ext_edge(name, ext_modname):
             #           self.call_graph.add_edge(self.current_ns, dec_name)
 
             if pointer_def.get_type() == utils.constants.CLS_DEF:
+                # Class def
+                self.call_graph.add_edge(self.current_method, pointer)
+
                 init_ns = self.find_cls_fun_ns(pointer, utils.constants.CLS_INIT)
 
                 for ns in init_ns: