Handle messages with no MessageAttributes in message

renanvieira · Jul 9, 2020 · 5de026f · 5de026f
1 parent 528d84b
commit 5de026f
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/phoenix_letter/main.py b/src/phoenix_letter/main.py
@@ -66,10 +66,14 @@ def main(args=None):
 
         for message in received_response['Messages']:
             print("Sending message to '{}'".format(args.destination))
-
-            send_response = sqs_client.send_message(QueueUrl=destination_queue_url,
+
+            if 'MessageAttributes' in message:
+                send_response = sqs_client.send_message(QueueUrl=destination_queue_url,
                                                     MessageBody=message['Body'],
                                                     MessageAttributes=message['MessageAttributes'])
+            else: 
+                send_response = sqs_client.send_message(QueueUrl=destination_queue_url,
+                                                    MessageBody=message['Body'])
 
             print("Deleting message from '{}'".format(args.source))
             sqs_client.delete_message(QueueUrl=source_queue_url,