Properly set workflow type for rerun (#41)

Co-authored-by: Ziyang Li <[email protected]>

omics/cli/rerun/__main__.py

@@ -163,6 +163,10 @@ def get_run_resources(logs, run):
         rqst["nextToken"] = token
     return sorted(resources, key=lambda x: x.get("creationTime"))
 
+def get_workflow_type(run):
+    if not run.get("workflow", None) or len(run["workflow"].split(":")) < 5:
+        die(f"Failed to retrieve workflow type from run {run['arn']}")
+    return "READY2RUN" if not run["workflow"].split(":")[4] else "PRIVATE"
 
 def start_run_request(run, opts={}):
     """Build StartRun request"""
@@ -178,18 +182,17 @@ def set_param(rqst, key, key0, val=None):
     rqst = {}
     if opts.get("--workflow-id"):
         set_param(rqst, "workflowId", "--workflow-id")
-        if opts.get("--workflow-type"):
-            rqst["workflowType"] = opts["--workflow-type"]
     elif opts.get("--run-id"):
         set_param(rqst, "runId", "--run-id")
     elif run.get("run"):
         set_param(rqst, "runId", None, run["run"].split("/")[-1])
     else:
         set_param(rqst, "workflowId", None, run["workflow"].split("/")[-1])
-        if opts.get("--workflow-type"):
-            rqst["workflowType"] = opts["--workflow-type"]
-        elif not run["workflow"].split(":")[4]:
-            rqst["workflowType"] = "READY2RUN"
+
+    if opts.get("--workflow-type"):
+        set_param(rqst, "workflowType", "--workflow-type")
+    else:
+        rqst["workflowType"] = get_workflow_type(run)
 
     set_param(rqst, "roleArn", "--role-arn")
     set_param(rqst, "name", "--name")