Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -64,15 +64,9 @@ def fetch_message_data(access_token: str, message_id: str) -> Message:
 
     attachments = extract_attachments_from_mail(access_token, message_data)
     body_len = 0
-    attachment_len = 0
     if body is not None :
         body_len = len(body)
 
-    if attachments is not None:
-        attachment_len = len(attachments)
-
-    
-        
     # print("subject: ")
     # print(subject)
     # print("company name: ")
@@ -82,7 +76,7 @@ def fetch_message_data(access_token: str, message_id: str) -> Message:
     # print("Printing attachment Data: ")
     # print(attachments)
     # print("Completed this mail.")
-    return Message(message_id=message_id, body=body, attachments=attachments, company=company_from_mail , body_len=body_len , attachment_len=attachment_len)
+    return Message(message_id=message_id, body=body, attachments=attachments, company=company_from_mail , body_len=body_len)
 
 
 def extract_subject_from_mail(message_data: dict) -> str:
@@ -143,7 +137,7 @@ def extract_attachments_from_mail(access_token: str, message_data: dict) -> List
                 attachment_id = part["body"]["attachmentId"]
                 attachment_data = fetch_attachment_data(access_token, message_data["id"], attachment_id)
                 filename = part.get("filename", "untitled.txt")
-                attachments.append(Attachment(filename=filename, data=attachment_data.get("data", "")))
+                attachments.append(Attachment(filename=filename, data=attachment_data.get("data", ""),attachment_len = len(attachment_data.get("data", ""))))
     return attachments