updating receiveFromSQS.py

slack-bot/traffic-scraper/prd/sendToSQS.py

@@ -4,16 +4,18 @@ import hashlib
 from datetime import datetime
 import os
 import sys
+
 sys.path.append(os.getcwd())
 from pullTrafficInfo import getTrafficInfo
 
 
 class sendToSQS(object):
     """docstring for sendToSQS"""
+
     def __init__(self, session, queueURL):
         super(sendToSQS, self).__init__()
         self.session = session
-        self.sqs = session.client('sqs')
+        self.sqs = session.client("sqs")
         self.queueURL = queueURL
 
     @classmethod
@@ -22,29 +24,39 @@ class sendToSQS(object):
         return cls(session, queueURL)
 
     def sendMessage(self, message, messageGroupId, b64=True, dedup=False):
-        currentTime = datetime.now().strftime('%H:%M:%S.%f')
+        currentTime = datetime.now().strftime("%H:%M:%S.%f")
         if b64:
             message = (base64.b64encode(message.encode())).decode()
         if not dedup:
             dedupId = hashlib.md5((message + currentTime).encode()).hexdigest()
-            msg = self.sqs.send_message(QueueUrl=self.queueURL,
-                                        MessageBody=message,
-                                        MessageGroupId=messageGroupId,
-                                        MessageDeduplicationId=dedupId)
+            msg = self.sqs.send_message(
+                QueueUrl=self.queueURL,
+                MessageBody=message,
+                MessageGroupId=messageGroupId,
+                MessageDeduplicationId=dedupId,
+            )
         else:
-            msg = self.sqs.send_message(QueueUrl=self.queueURL,
-                                        MessageBody=message,
-                                        MessageGroupId=messageGroupId)
+            msg = self.sqs.send_message(
+                QueueUrl=self.queueURL,
+                MessageBody=message,
+                MessageGroupId=messageGroupId,
+            )
         if msg is not None:
             print(msg["MessageId"])
 
 
-inst = sendToSQS.createSession(profileName='plex-aws',
-                               queueURL='https://sqs.eu-west-1.amazonaws.com'
-                                        '/745437999005/slack-bot.fifo')
+inst = sendToSQS.createSession(
+    profileName="plex-aws",
+    queueURL="https://sqs.eu-west-1.amazonaws.com"
+    "/745437999005/slack-bot.fifo",
+)
 
-instM = getTrafficInfo.getTrafficURL('M62').findIncidents() \
-    .getIncidentInformation().generateOutput()
+instM = (
+    getTrafficInfo.getTrafficURL("M62")
+    .findIncidents()
+    .getIncidentInformation()
+    .generateOutput()
+)
 
 for item in instM.output:
-    inst.sendMessage(message=item, messageGroupId='slack-bot-M62')
+    inst.sendMessage(message=item, messageGroupId="slack-bot-M62")