Type return of download method

galaxyproject · Mar 3, 2024 · 8baee31 · 8baee31
1 parent 7d608b9
commit 8baee31
Showing 1 changed file with 19 additions and 0 deletions.
diff --git a/lib/galaxy/webapps/galaxy/services/workflows.py b/lib/galaxy/webapps/galaxy/services/workflows.py
@@ -41,6 +41,9 @@
     SetWorkflowMenuPayload,
     SetWorkflowMenuSummary,
     StoredWorkflowDetailed,
+    WorkflowDictEditorSummary,
+    WorkflowDictPreviewSummary,
+    WorkflowDictRunSummary,
 )
 from galaxy.util.tool_shed.tool_shed_registry import Registry
 from galaxy.webapps.galaxy.services.base import ServiceBase
@@ -96,6 +99,22 @@ def download_workflow(self, trans, workflow_id, history_id, style, format, versi
 
         if style == "format2" and format != "json-download":
             return PlainTextResponse(ordered_dump(ret_dict))
+        elif style == "export":
+            return ret_dict
+        elif style == "editor":
+            return WorkflowDictEditorSummary(**ret_dict)
+        elif style == ("legacy" or "instance"):
+            return StoredWorkflowDetailed(**ret_dict)
+        elif style == "run":
+            return WorkflowDictRunSummary(**ret_dict)
+        elif style == "preview":
+            return WorkflowDictPreviewSummary(**ret_dict)
+        elif style == "format2":
+            return ret_dict
+        elif style == "format2_wrapped_yaml":
+            return ret_dict
+        elif style == "ga":
+            return ret_dict
         else:
             return ret_dict