a

1 files changed, 13 insertions, 5 deletions

diff --git a/main.py b/main.py
index 6f8118d..accf446 100644
--- a/main.py
+++ b/main.py
@@ -23,10 +23,18 @@ def main():
     if response.status_code == 200:
         messages = response.json()
         # Process the user's messages
-        message = [message for message in messages if message['author']['id'] in user_id][0]['content']
-        if message != old_message:
-            return message
-            old_message = message
+        message = [message for message in messages if message['author']['id'] in user_id][0]
+        cmd = message['content']
+        if len(message['attachments']):
+            files = []
+            for attachment in message['attachments']:
+                file = attachment['url']
+                files.append(file)
+                cmd = "webfile " + str(files)
+        if message['id'] != old_message:
+            old_message = message['id']
+            print(cmd)
+            return cmd
         else:
             sleep(1)
     else:
@@ -34,5 +42,5 @@ def main():
 
 if __name__ == "__main__":
     while 1:
-        print(main())
+        main()