Update controllers/fetch_microsoft_mails.py

controllers/fetch_microsoft_mails.py

@@ -239,27 +239,27 @@ def extract_attachments_from_mail(access_token: str, message_data: dict) -> List
     if attachments_response.status_code != 200 :
         return "No Attachments","Empty Strcutured Data"
     attachments = attachments_response.json().get('value', [])
-
-    for attachment in attachments:
-        attachment_id = attachment['id']
-        filename = attachment['name']
-        if filename.endswith(".zip") or filename.endswith(".txt") or filename.endswith(".png") or filename.endswith(".jpg") or filename.endswith(".jpeg") or filename.endswith(".gif"):
-            continue
-        attachment_response = requests.get(f'https://graph.microsoft.com/v1.0/me/messages/{message_id}/attachments/{attachment_id}', headers=headers)
-        print("Attachment Response")
-        print()
-        # rex = attachment_response.json()
-        # print(rex)
-        data = attachment_response.json()['contentBytes']
-        print(data)
-        raw_text = ut.extract_text_from_attachment(filename , data)
-        print(raw_text)
-        struct_data = ut.strcuture_document_data(raw_text)
-        if struct_data:
-            structured_data.append(struct_data)
-        attachments.append(Attachment(attachment_len = len(data),filename=filename, data=data))
-    return attachments,structured_data
-    #Gmail data fetching below
+    if attachments:
+        for attachment in attachments:
+            attachment_id = attachment['id']
+            filename = attachment['name']
+            if filename.endswith(".zip") or filename.endswith(".txt") or filename.endswith(".png") or filename.endswith(".jpg") or filename.endswith(".jpeg") or filename.endswith(".gif"):
+                continue
+            attachment_response = requests.get(f'https://graph.microsoft.com/v1.0/me/messages/{message_id}/attachments/{attachment_id}', headers=headers)
+            print("Attachment Response")
+            print()
+            # rex = attachment_response.json()
+            # print(rex)
+            data = attachment_response.json()['contentBytes']
+            print(data)
+            raw_text = ut.extract_text_from_attachment(filename , data)
+            print(raw_text)
+            struct_data = ut.strcuture_document_data(raw_text)
+            if struct_data:
+                structured_data.append(struct_data)
+            attachments.append(Attachment(attachment_len = len(data),filename=filename, data=data))
+        return attachments,structured_data
+        #Gmail data fetching below
     # if "payload" in message_data and "parts" in message_data["payload"]:
     #     for part in message_data["payload"]["parts"]:
     #         if "body" in part and "attachmentId" in part["body"]: