🎨 better error handle

Wytamma · Nov 9, 2023 · 4d7fdcc · 4d7fdcc
1 parent aebfe42
commit 4d7fdcc
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/snk/cli/subcommands/run.py b/snk/cli/subcommands/run.py
@@ -288,8 +288,11 @@ def _save_dag(self, snakemake_args: List[str], filename: Path):
             except SystemExit:  # Catch SystemExit exception to prevent termination
                 pass
         try:
+            snakemake_output = snakemake_output.getvalue()
+            if "snakemake_dag" not in snakemake_output:
+                self.error("Could not generate dag!", exit=True)
             # discard everything before digraph snakemake_dag
-            filtered_lines = "digraph snakemake_dag" + snakemake_output.getvalue().split("snakemake_dag")[1]
+            filtered_lines = "digraph snakemake_dag" + snakemake_output.split("snakemake_dag")[1]
             echo_process = subprocess.Popen(
                 ["echo", filtered_lines], stdout=subprocess.PIPE
             )