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

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

Hide file tree

Showing 2 changed files with 28 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,21 @@ def run(self, labels: list, analysis_id=None):
                     runtime_environments = [
                         e["name"] for e in thoth_config["runtime_environments"]
                     ]
+                    managers = thoth_config.get("managers")
+                    for mgr in managers:
+                        if mgr.get("name") == "thoth-advise":
+                            disallow_run_times = mgr.get("disallow", None)
+                            allowed_run_times = mgr.get("allow", None)
+                    if disallow_run_times:
+                        runtime_environments = list(
+                            set(runtime_environments) - set(disallow_run_times)
+                        )
+                    if allowed_run_times:
+                        runtime_environments = list(
+                            set(runtime_environments).intersection(
+                                set(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,19 @@ def run(self, labels: list = []) -> Optional[dict]:
                 else:
                     runtime_environments = [None]
 
+            managers = thoth_config.content.get("managers")
+            for mgr in managers:
+                if mgr.get("name") == "update":
+                    disallow_run_times = mgr.get("disallow", None)
+                    allowed_run_times = mgr.get("allow", None)
+            if disallow_run_times:
+                runtime_environments = list(
+                    set(runtime_environments) - set(disallow_run_times)
+                )
+            if allowed_run_times:
+                runtime_environments = list(
+                    set(runtime_environments).intersection(set(allowed_run_times))
+                )
             results: dict = {}
             for e in runtime_environments:
                 self.runtime_environment = e or "default"