completed full run script

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

@@ -127,34 +127,15 @@ class receiveFromSQS(object):
         )
         return self
 
-    def generateOutput(self, type='json'):
-        if type == 'json':
-            self.output = json.dumps(self.messages)
-        return self.output
+    # def generateOutput(self, outputType='json'):
+    #     if outputType == 'json':
+    #         self.output = json.dumps(self.messages)
+    #     return self.output
 
     def savetoDisk(self, path):
         self.timeNow = datetime.now().strftime('%d-%m-%Y_%H:%M:%S')
-        if self.output is None:
-            self.generateOutput()
         if len(self.messages) > 0:
             with open(f'{path}/{self.timeNow}.json', 'w+') as outputFile:
-                outputFile.write(self.output)
+                json.dump(self.messages, outputFile)
         else:
             print('No messages to save')
-
-
-inst = receiveFromSQS.createSession(
-    profileName='plex-aws',
-    queueURL='https://sqs.eu-west-1.amazonaws'
-    '.com/745437999005/slack-bot.fifo',
-)
-
-output = inst.receiveNMessages(numberOfMessages=4).generateOutput(type='json')
-inst.savetoDisk(
-    '/Users/dtomlinson/OneDrive - William Hill'
-    ' Organisation Limited/Mac/git_repos/python-VM/slack-bot/traffic-scraper'
-    '/prd'
-)
-# inst.receiveAllMessages()
-for item in inst.messages:
-    print(item)