diff --git a/bignbit/save_cnm_message.py b/bignbit/save_cnm_message.py index 4be1865..d1056cd 100644 --- a/bignbit/save_cnm_message.py +++ b/bignbit/save_cnm_message.py @@ -31,7 +31,7 @@ def process(self): """ pobit_audit_bucket = self.config['pobit_audit_bucket'] print(pobit_audit_bucket) - print(self.input.get('cnm_sent')) + print(self.input) collection_name = self.config['collection'] granule_ur = self.config['granule_ur'] @@ -39,7 +39,7 @@ def process(self): print(collection_name) print(granule_ur) - cnm_content = self.input + cnm_content = self.input['cnm'] print(cnm_content) print(cnm_content) diff --git a/bignbit/send_to_gitc.py b/bignbit/send_to_gitc.py index f45407a..4e3f42e 100644 --- a/bignbit/send_to_gitc.py +++ b/bignbit/send_to_gitc.py @@ -51,13 +51,9 @@ def process(self): image_set = ImageSet(**self.input['image_set']) gitc_id = image_set.name - response_payload = {} - response_payload = self.input.copy() - cnm_message = notify_gitc(image_set, cmr_provider, gitc_id, collection_name) - response_payload['cnm_sent'] = cnm_message - return response_payload + return cnm_message def notify_gitc(image_set: ImageSet, cmr_provider: str, gitc_id: str, collection_name: str): diff --git a/terraform/state_machine_definition.tpl b/terraform/state_machine_definition.tpl index a8bdbeb..fb069c1 100644 --- a/terraform/state_machine_definition.tpl +++ b/terraform/state_machine_definition.tpl @@ -448,6 +448,7 @@ "Type": "Task", "Resource": "arn:aws:states:::lambda:invoke", "TimeoutSeconds": 86400, + "ResultPath": "$.cnm", "Next": "SaveCNMMessage" }, "SaveCNMMessage": { @@ -461,7 +462,7 @@ "granule_ur": "{$.granule_ur}", "pobit_audit_bucket": "${PobitAuditBucket}", "cumulus_message": { - "input": "{$}" + "input": "{$.payload}" } } }