Add 'allow' and 'disallow' setting to each manager

Signed-off-by: Shreekara <sshreeka@redhat.com>
thoth-station · Oct 11, 2022 · 8b9d789 · 8b9d789
1 parent 356af8e
commit 8b9d789
Show file tree

Hide file tree

Showing 2 changed files with 18 additions and 0 deletions.
diff --git a/kebechet/managers/thoth_advise/thoth_advise.py b/kebechet/managers/thoth_advise/thoth_advise.py
@@ -316,6 +316,15 @@ def run(self, labels: list, analysis_id=None):
                     runtime_environments = [
                         e["name"] for e in thoth_config["runtime_environments"]
                     ]
+                    managers = thoth_config.get("managers")
+                    disallow_run_times = managers[0].get("disallow", None)
+                    allowed_run_times = managers[0].get("allow", None)
+                    if disallow_run_times:
+                        runtime_environments = list(
+                            set(runtime_environments) ^ set(disallow_run_times)
+                        )
+                    if allowed_run_times:
+                        runtime_environments = allowed_run_times
                 else:
                     runtime_environments = [
                         thoth_config["runtime_environments"][0]["name"]

diff --git a/kebechet/managers/update/update.py b/kebechet/managers/update/update.py
@@ -728,6 +728,15 @@ def run(self, labels: list = []) -> Optional[dict]:
                 else:
                     runtime_environments = [None]
 
+            managers = thoth_config.content.get("managers")
+            disallow_run_times = managers[0].get("disallow", None)
+            allowed_run_times = managers[0].get("allow", None)
+            if disallow_run_times:
+                runtime_environments = list(
+                    set(runtime_environments) ^ set(disallow_run_times)
+                )
+            if allowed_run_times:
+                runtime_environments = allowed_run_times
             results: dict = {}
             for e in runtime_environments:
                 self.runtime_environment = e or "default"