Exclusively build for pwdata and filter out modules

pycg/processing/postprocessor.py

@@ -30,6 +30,8 @@ def __init__(
         self,
         input_file,
         modname,
+        exclusives,
+        ignored_mods,
         import_manager,
         scope_manager,
         def_manager,
@@ -38,6 +40,8 @@ def __init__(
         modules_analyzed=None,
     ):
         super().__init__(input_file, modname, modules_analyzed)
+        self.exclusives = exclusives
+        self.ignored_mods = ignored_mods
         self.import_manager = import_manager
         self.scope_manager = scope_manager
         self.def_manager = def_manager
@@ -329,6 +333,8 @@ def update_parent_classes(self, defi):
     def analyze_submodules(self):
         super().analyze_submodules(
             PostProcessor,
+            self.exclusives,
+            self.ignored_mods,
             self.import_manager,
             self.scope_manager,
             self.def_manager,

pycg/processing/preprocessor.py

@@ -30,6 +30,8 @@
         self,
         filename,
         modname,
+        exclusives,
+        ignored_mods,
         import_manager,
         scope_manager,
         def_manager,
@@ -40,6 +42,8 @@
         super().__init__(filename, modname, modules_analyzed)
 
         self.modname = modname
+        self.exclusives = exclusives
+        self.ignored_mods = ignored_mods
         self.mod_dir = "/".join(self.filename.split("/")[:-1])
 
         self.import_manager = import_manager
@@ -71,6 +75,8 @@
         super().analyze_submodule(
             PreProcessor,
             modname,
+            self.exclusives,
+            self.ignored_mods,
             self.import_manager,
             self.scope_manager,
             self.def_manager,
@@ -81,6 +87,10 @@
 
     def visit_Module(self, node):
         def iterate_mod_items(items, const):
+            items = [
+                i for i in items if not \
+                    any(ignored_mod in i for ignored_mod in self.ignored_mods)
+            ]
             for item in items:
                 defi = self.def_manager.get(item)
                 if not defi:
@@ -205,6 +215,14 @@
             tgt_name = import_item.asname if import_item.asname else import_item.name
             imported_name = self.import_manager.handle_import(src_name, level)
 
+            # Limit to exclusive module
+            if src_name.split(".")[0] not in self.exclusives:
+                continue
+
+            # Skip ignored modules
+            if tgt_name in self.ignored_mods:
+                continue
+
             if not imported_name:
                 add_external_def(src_name, tgt_name)
                 continue
@@ -223,6 +241,9 @@
 
         # handle all modules that were not analyzed
         for modname in self.import_manager.get_imports(self.modname):
+            if modname.split(".")[0] not in self.exclusives:
+                continue
+
             fname = self.import_manager.get_filepath(modname)
 
             if not fname:

pycg/pycg.py

@@ -35,9 +35,13 @@
 
 
 class CallGraphGenerator(object):
-    def __init__(self, entry_points, package, max_iter, operation):
+    def __init__(
+        self, entry_points, package, exclusives, ignored_mods, max_iter, operation
+    ):
         self.entry_points = entry_points
         self.package = package
+        self.exclusives = exclusives
+        self.ignored_mods = ignored_mods
         self.state = None
         self.max_iter = max_iter
         self.operation = operation
@@ -162,6 +166,8 @@ def analyze(self):
         self.do_pass(
             PreProcessor,
             True,
+            self.exclusives,
+            self.ignored_mods,
             self.import_manager,
             self.scope_manager,
             self.def_manager,
@@ -179,6 +185,8 @@ def analyze(self):
             self.do_pass(
                 PostProcessor,
                 False,
+                self.exclusives,
+                self.ignored_mods,
                 self.import_manager,
                 self.scope_manager,
                 self.def_manager,