Merge "Improve Remote Python Executor error handling and allow for structured response"

1 files changed, 8 insertions, 4 deletions

diff --git a/ms/command-executor/src/main/python/utils.py b/ms/command-executor/src/main/python/utils.py
index 4314b287d..a3748eb17 100644
--- a/ms/command-executor/src/main/python/utils.py
+++ b/ms/command-executor/src/main/python/utils.py
@@ -16,7 +16,7 @@
 from google.protobuf.timestamp_pb2 import Timestamp
 
 import proto.CommandExecutor_pb2 as CommandExecutor_pb2
-
+import json
 
 def get_blueprint_id(request):
     blueprint_name = request.identifiers.blueprintName
@@ -24,7 +24,7 @@ def get_blueprint_id(request):
     return blueprint_name + '/' + blueprint_version
 
 
-def build_response(request, results, is_success=True):
+def build_response(request, log_results, payload_return, is_success=False):
     if is_success:
         status = CommandExecutor_pb2.SUCCESS
     else:
@@ -32,5 +32,9 @@ def build_response(request, results, is_success=True):
 
     timestamp = Timestamp()
     timestamp.GetCurrentTime()
-    return CommandExecutor_pb2.ExecutionOutput(requestId=request.requestId, response=results, status=status,
-                                               timestamp=timestamp)
+
+    if 'cds_return_code' in payload_return:
+        payload_return.pop('cds_return_code')
+    payload_str = json.dumps(payload_return)
+    return CommandExecutor_pb2.ExecutionOutput(requestId=request.requestId, response=log_results, status=status,
+                                               payload=payload_str, timestamp=timestamp)