From 23e1139c212aa449059ae82f975555142e20e5ec Mon Sep 17 00:00:00 2001
From: Fabian <fab.hof@gmx.de>
Date: Tue, 19 Mar 2024 09:39:35 +0100
Subject: [PATCH] snakefile + helpers: separate scenario and rdir getter

---
 Snakefile           |  5 +++--
 scripts/_helpers.py | 17 ++++++++++++-----
 2 files changed, 15 insertions(+), 7 deletions(-)

diff --git a/Snakefile b/Snakefile
index 4a41622e..3e8875d3 100644
--- a/Snakefile
+++ b/Snakefile
@@ -10,7 +10,7 @@ from snakemake.utils import min_version
 
 min_version("8.5")
 
-from scripts._helpers import path_provider, copy_default_files, process_run_config
+from scripts._helpers import path_provider, copy_default_files, get_scenarios, get_rdir
 
 
 copy_default_files(workflow)
@@ -21,7 +21,8 @@ configfile: "config/config.yaml"
 
 
 run = config["run"]
-RDIR = process_run_config(run)
+scenarios = get_scenarios(run)
+RDIR = get_rdir(run)
 
 logs = path_provider("logs/", RDIR, run["shared_resources"])
 benchmarks = path_provider("benchmarks/", RDIR, run["shared_resources"])
diff --git a/scripts/_helpers.py b/scripts/_helpers.py
index 2bacf5cd..9b390380 100644
--- a/scripts/_helpers.py
+++ b/scripts/_helpers.py
@@ -39,14 +39,21 @@ def copy_default_files(workflow):
             copyfile(template, target)
 
 
-def process_run_config(run):
-    scenarios = run.get("scenarios", {})
-    if run["name"] and scenarios.get("enable"):
-        fn = Path(scenarios["file"])
+def get_scenarios(run):
+    scenario_config = run.get("scenarios", {})
+    if run["name"] and scenario_config.get("enable"):
+        fn = Path(scenario_config["file"])
         scenarios = yaml.safe_load(fn.read_text())
-        RDIR = "{run}/"
         if run["name"] == "all":
             run["name"] = list(scenarios.keys())
+        return scenarios
+    return {}
+
+
+def get_rdir(run):
+    scenario_config = run.get("scenarios", {})
+    if run["name"] and scenario_config.get("enable"):
+        RDIR = "{run}/"
     elif run["name"]:
         RDIR = run["name"] + "/"
     else: