fix breaking change in stomp.py

Author: bgeisberger

example_emails.py

@@ -47,11 +47,11 @@
 
 
 class STOMPListener(stomp.ConnectionListener):
-    def on_error(self, headers, body):
-        print('report_error', body)
+    def on_error(self, frame):
+        print('report_error', frame.body)
 
-    def on_message(self, headers, body):
-        msg_json = json.loads(body)
+    def on_message(self, frame):
+        msg_json = json.loads(frame.body)
 
         for field, allowed in ALLOWLIST_FILTERS.items():
             if msg_json.get(field) not in allowed:
@@ -70,10 +70,10 @@ def on_message(self, headers, body):
         makedirs(path, exist_ok=True)
 
         if WRITE_RAW_MAILS:
-            with open(path_join(path, str(hash(body)) + '.eml'), 'wb') as f:
+            with open(path_join(path, str(hash(frame.body)) + '.eml'), 'wb') as f:
                 f.write(b64decode(msg_json['original_message_base64_encoded']))
         else:
-            with open(path_join(path, str(hash(body)) + '.json'), 'w') as f:
+            with open(path_join(path, str(hash(frame.body)) + '.json'), 'w') as f:
                 json.dump(msg_json, f)
 
 

example_files_json.py

@@ -39,11 +39,11 @@
 
 
 class STOMPListener(stomp.ConnectionListener):
-    def on_error(self, headers, body):
-        print('report_error', body)
+    def on_error(self, frame):
+        print('report_error', frame.body)
 
-    def on_message(self, headers, body):
-        msg_json = json.loads(body)
+    def on_message(self, frame):
+        msg_json = json.loads(frame.body)
 
         for field, allowed in ALLOWLIST_FILTERS.items():
             if msg_json.get(field) not in allowed:
@@ -58,10 +58,10 @@ def on_message(self, headers, body):
         makedirs(path, exist_ok=True)
 
         if WRITE_RAW_FILES:
-            with open(path_join(path, str(hash(body)) + '.file'), 'wb') as f:
+            with open(path_join(path, str(hash(frame.body)) + '.file'), 'wb') as f:
                 f.write(b64decode(msg_json['attachment_base64_encoded']))
         else:
-            with open(path_join(path, str(hash(body)) + '.json'), 'w') as f:
+            with open(path_join(path, str(hash(frame.body)) + '.json'), 'w') as f:
                 json.dump(msg_json, f)
 
 

example_files_raw.py

@@ -19,16 +19,16 @@
 
 
 class STOMPListener(stomp.ConnectionListener):
-    def on_error(self, headers, body):
-        print('report_error', body)
+    def on_error(self, frame):
+        print('report_error', frame.body)
 
-    def on_message(self, headers, body):
+    def on_message(self, frame):
         msg_date = datetime.now()
         path = msg_date.strftime(OUTPUT_DIRECTORY)
         makedirs(path, exist_ok=True)
 
-        with open(path_join(path, str(hash(body)) + '.file'), 'wb') as f:
-            f.write(b64decode(body))
+        with open(path_join(path, str(hash(frame.body)) + '.file'), 'wb') as f:
+            f.write(b64decode(frame.body))
 
 
 def listen():

example_ipdomains.py

@@ -33,11 +33,11 @@
 
 
 class STOMPListener(stomp.ConnectionListener):
-    def on_error(self, headers, body):
-        print('report_error', body)
+    def on_error(self, frame):
+        print('report_error', frame.body)
 
-    def on_message(self, headers, body):
-        msg_json = json.loads(body)
+    def on_message(self, frame):
+        msg_json = json.loads(frame.body)
 
         for field, allowed in ALLOWLIST_FILTERS.items():
             if msg_json.get(field) not in allowed:
@@ -51,7 +51,7 @@ def on_message(self, headers, body):
         path = msg_date.strftime(OUTPUT_DIRECTORY)
         makedirs(path, exist_ok=True)
 
-        with open(path_join(path, str(hash(body)) + '.json'), 'w') as f:
+        with open(path_join(path, str(hash(frame.body)) + '.json'), 'w') as f:
             json.dump(msg_json, f)
 
 

example_urls_json.py

@@ -33,11 +33,11 @@
 
 
 class STOMPListener(stomp.ConnectionListener):
-    def on_error(self, headers, body):
-        print('report_error', body)
+    def on_error(self, frame):
+        print('report_error', frame.body)
 
-    def on_message(self, headers, body):
-        msg_json = json.loads(body)
+    def on_message(self, frame):
+        msg_json = json.loads(frame.body)
 
         if 'url_tld' not in msg_json:
             msg_json['url_tld'] = urlparse(msg_json.get('url')).hostname.split('.')[-1]

example_urls_raw.py

@@ -18,16 +18,16 @@
 
 
 class STOMPListener(stomp.ConnectionListener):
-    def on_error(self, headers, body):
-        print('report_error', body)
+    def on_error(self, frame):
+        print('report_error', frame.body)
 
-    def on_message(self, headers, body):
+    def on_message(self, frame):
         msg_date = datetime.now()
         path = msg_date.strftime(OUTPUT_FILE)
         makedirs(dirname(path), exist_ok=True)
 
         with open(path, 'a') as f:
-            f.write(body + '\n')
+            f.write(frame.body + '\n')
 
 
 def listen():