Patched /home/runner/work/patchwork/patchwork/patchwork/common/utils/…

…dependency.py
patched-codes · Nov 15, 2024 · cc42f04 · cc42f04
1 parent 8187002
commit cc42f04
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/patchwork/common/utils/dependency.py b/patchwork/common/utils/dependency.py
@@ -7,9 +7,12 @@
     "notification": ["slack_sdk"],
 }
 
+__ALLOWED_MODULES = {module for modules in __DEPENDENCY_GROUPS.values() for module in modules}
 
 @lru_cache(maxsize=None)
 def import_with_dependency_group(name):
+    if name not in __ALLOWED_MODULES:
+        raise ImportError(f"Module {name} is not allowed to be imported.")
     try:
         return importlib.import_module(name)
     except ImportError: