From 1937b1f676bb3b203a4d19ecf373df34c0c61806 Mon Sep 17 00:00:00 2001
From: Nadine Wieters <nadine.wieters@awi.de>
Date: Mon, 18 Mar 2024 10:45:14 +0100
Subject: [PATCH] Added phase and run_from_batch_script as new esm_runscripts
 arguments. New values for argument task.

---
 src/esm_runscripts/cli.py | 36 +++++++++++++++++++++++++++++++++---
 1 file changed, 33 insertions(+), 3 deletions(-)

diff --git a/src/esm_runscripts/cli.py b/src/esm_runscripts/cli.py
index c7372ac52..61cb096c7 100644
--- a/src/esm_runscripts/cli.py
+++ b/src/esm_runscripts/cli.py
@@ -10,6 +10,7 @@
 import argparse
 import logging
 import os
+import pdb
 import sys
 
 from loguru import logger
@@ -96,10 +97,24 @@ def parse_shargs():
     parser.add_argument(
         "-t",
         "--task",
-        help="The task to run. Choose from: prepcompute, post, couple, tidy",
+        help="The task to run. Choose from: start, restart, continue, ...",
+        default="start",
+    )
+
+    parser.add_argument(
+        "-a",
+        "--phase",
+        help="The workflow phase to run. Choose from: prepcompute, compute, tidy, ...",
         default="unknown",
     )
 
+    parser.add_argument(
+        "-w",
+        "--workflow",
+        help="The workflow to run. Choose from: default (sim_cluster???), ...",
+        default="sim_cluster",
+    )
+
     parser.add_argument(
         "-i",
         "--inspect",
@@ -110,7 +125,7 @@ def parse_shargs():
     parser.add_argument(
         "-p",
         "--pid",
-        help="The PID of the task to observe.",
+        help="The PID of the phase to observe.",
         default=-666,
     )
 
@@ -162,6 +177,12 @@ def parse_shargs():
         action="store_true",
     )
 
+    parser.add_argument(
+        "--run-from-batch-script",
+        default=False,
+        action="store_true",
+    )
+
     return parser.parse_args()
 
 
@@ -202,8 +223,12 @@ def main():
         update_filetypes = parsed_args["update_filetypes"]
     if "expid" in parsed_args:
         expid = parsed_args["expid"]
+    if "phase" in parsed_args:
+        jobtype = parsed_args["phase"]
+    #    if "workflow" in parsed_args:
+    #        workflow = parsed_args["workflow"]
     if "task" in parsed_args:
-        jobtype = parsed_args["task"]
+        task = parsed_args["task"]
     if "verbose" in parsed_args:
         verbose = parsed_args["verbose"]
     if "inspect" in parsed_args:
@@ -224,6 +249,8 @@ def main():
         no_motd = parsed_args["no_motd"]
     if "ignore_config_warnings" in parsed_args:
         ignore_config_warnings = parsed_args["ignore_config_warnings"]
+    if "run_from_batch_script" in parsed_args:
+        run_from_batch_script = parsed_args["run_from_batch_script"]
 
     command_line_config = {}
     command_line_config["check"] = check
@@ -235,6 +262,8 @@ def main():
     command_line_config["current_date"] = start_date
     command_line_config["run_number"] = run_number
     command_line_config["jobtype"] = jobtype
+    #    command_line_config["workflow"] = workflow
+    command_line_config["task"] = task
     command_line_config["last_jobtype"] = ARGS.last_jobtype
     command_line_config["verbose"] = verbose
     command_line_config["inspect"] = inspect
@@ -243,6 +272,7 @@ def main():
     command_line_config["ignore_config_warnings"] = ignore_config_warnings
     if modify_config_file:
         command_line_config["modify_config_file"] = modify_config_file
+    command_line_config["run_from_batch_script"] = run_from_batch_script
 
     # runscript_from_cmdline = filter(lambda x: x.endswith(".yaml"), sys.argv)
     # runscript_from_cmdline = list(runscript_from_cmdline)[0]