updating sendToSQS.py

slack-bot/sendToSQS.py

@@ -6,15 +6,48 @@ from pullTrafficInfo import getTrafficInfo
 import base64
 import emoji
 
+motorway = 'M62'
+
 session = boto3.Session(profile_name='plex-aws')
-sqs = session.resource('sqs')
+sqs = session.client('sqs')
 
 # queue = sqs.create_queue(QueueName='test', Attributes={'DelaySeconds': '5'})
 
-inst = getTrafficInfo.getTrafficURL('A50').findIncidents() \
+inst = getTrafficInfo.getTrafficURL(motorway).findIncidents() \
     .getIncidentInformation().generateOutput()
 
-inst.output
+# encoded = base64.b64encode(inst.output[0].encode())
+# decoded = base64.b64decode(encoded).decode()
 
-encoded = base64.b64encode(inst.output[0].encode())
+queueURL = 'https://sqs.eu-west-1.amazonaws.com/745437999005/slack-bot.fifo'
-decoded = base64.b64decode(encoded).decode()
+
+# message = sqs.send_message(QueueUrl=queueURL,
+#                            MessageBody='string',
+#                            MessageGroupId='slack-bot-motorway')
+
+# response = sqs.receive_message(QueueUrl=queueURL,
+#                                MaxNumberOfMessages=5)
+# if 'Messages' in response:
+#     for message in response['Messages']:
+#         print(message['Body'])
+# else:
+#     print('Queue is empty')
+
+for item in inst.output:
+    encoded = (base64.b64encode(item.encode())).decode()
+    message = sqs.send_message(QueueUrl=queueURL,
+                               MessageBody=encoded,
+                               MessageGroupId=f'slack-bot-{motorway}')
+    response = sqs.receive_message(QueueUrl=queueURL,
+                                   MaxNumberOfMessages=5)
+    if 'Messages' in response:
+        for message in response['Messages']:
+            # print(message['Body'])
+            # break
+            decoded = base64.b64decode(message['Body'].encode())
+            print(decoded.decode())
+    else:
+        print('Queue is empty')
+    # print(encoded)
+    # print(encoded.encode())
+    # print(base64.b64decode(encoded.encode()).decode())