updating receive and send

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

@@ -15,6 +15,15 @@ class receiveFromSQS(object):
         session = boto3.Session(profile_name=profileName)
         return cls(session, queueURL)
 
+    def _receiveSQSMessage(self, maxNumberOfMessages=10):
+        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'):
         while True:
@@ -37,6 +46,10 @@ class receiveFromSQS(object):
                 print('Queue is empty')
                 break
 
+    def receiveAllMessages(self, b64=True, delete=None):
+        while True:
+            self.response = self._receiveSQSMessage()
+
 
 inst = receiveFromSQS.createSession(profileName='plex-aws',
                                     queueURL='https://sqs.eu-west-1.amazonaws'