applying black to scripts

slack-bot/traffic-scraper/dev/sendToSQS_old.py

@@ -26,7 +26,7 @@ inst = getTrafficInfo.getTrafficURL(motorway).findIncidents() \
 queueURL = 'https://sqs.eu-west-1.amazonaws.com/745437999005/slack-bot.fifo'
 
 # message = sqs.send_message(QueueUrl=queueURL,
-#                            MessageBody='string',
+#                            MessageBofdy='string',
 #                            MessageGroupId='slack-bot-motorway')
 
 # response = sqs.receive_message(QueueUrl=queueURL,
@@ -51,6 +51,10 @@ queueURL = 'https://sqs.eu-west-1.amazonaws.com/745437999005/slack-bot.fifo'
 #     # print(encoded.encode())
 #     # print(base64.b64decode(encoded.encode()).decode())
 
+queueAttributes = sqs\
+    .get_queue_attributes(QueueUrl=queueURL,
+                          AttributeNames=['ApproximateNumberOfMessages'])
+
 
 response = sqs.receive_message(QueueUrl=queueURL,
                                MaxNumberOfMessages=10)

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

@@ -4,6 +4,7 @@ import base64
 
 class receiveFromSQS(object):
     """docstring for receiveFromSQS"""
+
     def __init__(self, session, queueURL):
         super(receiveFromSQS, self).__init__()
         self.session = session
@@ -15,25 +16,47 @@ class receiveFromSQS(object):
         session = boto3.Session(profile_name=profileName)
         return cls(session, queueURL)
 
+    def getQueueLength(self):
+        attributeNames = ['ApproximateNumberOfMessages']
+        self.queueAttributes = self.sqs.get_queue_attributes(
+            QueueUrl=self.queueURL,
+            AttributeNames=attributeNames,
+        )
+        self.queueLength = int(
+            self.queueAttributes['Attributes'][
+                'ApproximateNumberOfMessages'
+            ]
+        )
+        return self.queueLength
+
     def _receiveSQSMessage(self, maxNumberOfMessages=10):
-        response = self.sqs\
-            .receive_message(QueueUrl=self.queueURL,
-                             MaxNumberOfMessages=maxNumberOfMessages)
+        response = self.sqs.receive_message(
+            QueueUrl=self.queueURL,
+            MaxNumberOfMessages=maxNumberOfMessages,
+        )
         return response
 
     def _extractMessageFromSQS(self):
         pass
 
-    def receiveMessage(self, b64=True, MaxNumberOfMessages=10, delete=False,
-                       endString='END'):
+    def receiveMessage(
+        self,
+        b64=True,
+        MaxNumberOfMessages=10,
+        delete=False,
+        endString='END',
+    ):
         while True:
-            response = self.sqs\
-                .receive_message(QueueUrl=self.queueURL,
-                                 MaxNumberOfMessages=MaxNumberOfMessages)
+            response = self.sqs.receive_message(
+                QueueUrl=self.queueURL,
+                MaxNumberOfMessages=MaxNumberOfMessages,
+            )
             if 'Messages' in response:
                 for message in response['Messages']:
                     if b64:
-                        decoded = base64.b64decode(message['Body'])
+                        decoded = base64.b64decode(
+                            message['Body']
+                        )
                         msg = decoded.decode()
                         print(msg)
                         if msg == endString:
@@ -46,12 +69,14 @@ class receiveFromSQS(object):
                 print('Queue is empty')
                 break
 
-    def receiveAllMessages(self, b64=True, delete=None):
+    def receiveAllMessages(self, b64=True, delete=False):
         while True:
             self.response = self._receiveSQSMessage()
 
 
-inst = receiveFromSQS.createSession(profileName='plex-aws',
+inst = receiveFromSQS.createSession(
+    profileName='plex-aws',
     queueURL='https://sqs.eu-west-1.amazonaws'
-                                    '.com/745437999005/slack-bot.fifo')
-inst.receiveMessage()
+    '.com/745437999005/slack-bot.fifo',
+)
+inst.getQueueLength()

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

@@ -47,4 +47,4 @@ instM = getTrafficInfo.getTrafficURL('M62').findIncidents() \
     .getIncidentInformation().generateOutput()
 
 for item in instM.output:
-    inst.sendMessage(message=item, messageGroupId='slack-bot-M62', dedup=True)
+    inst.sendMessage(message=item, messageGroupId='slack-bot-M62')