Allow blocklist to apply at the module level

CybercentreCanada · Nov 9, 2023 · 2d44805 · 2d44805
1 parent 89ec1ff
commit 2d44805
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/configextractor/main.py b/configextractor/main.py
@@ -103,6 +103,8 @@ def find_venv(path: str) -> str:
                             parser_site_packages = os.path.join(root, "site-packages")
                             sys.path.insert(1, parser_site_packages)
                             break
+                if block_regex and block_regex.match(module_name):
+                    continue
                 try:
                     module = importlib.import_module(module_name)
                 except Exception as e: