updated spiffworkflow-connector-command w/ burnettk

sartography · Oct 17, 2023 · 1dc289c · 1dc289c
1 parent 53fcd3e
commit 1dc289c
Show file tree

Hide file tree

Showing 2 changed files with 13 additions and 5 deletions.
diff --git a/poetry.lock b/poetry.lock
diff --git a/src/spiffworkflow_proxy/blueprint.py b/src/spiffworkflow_proxy/blueprint.py
@@ -56,10 +56,18 @@ def do_command(plugin_display_name: str, command_name: str) -> Response:
             status=500
         )
 
-    status_code = int(result['status'])
-    return_response = result["response"]
-    response = json.dumps(return_response)
-    return Response(response, mimetype=result["mimetype"], status=status_code)
+    if "command_response_version" in result and result["command_response_version"] > 1:  # type: ignore
+        response = json.dumps(result)
+        return Response(response, mimetype='application/json', status=200)
+    else:
+        status_code = 200
+        if 'status' in result:
+            status_code = int(result['status'])  # type: ignore
+        if isinstance(result["response"], dict):  # type: ignore
+            response = json.dumps(result["response"])  # type: ignore
+        else:
+            response = result["response"]  # type: ignore
+        return Response(response, mimetype=result["mimetype"], status=status_code)  # type: ignore
 
 
 @proxy_blueprint.route("/v1/auths")