Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -63,6 +63,16 @@ def fetch_message_data(access_token: str, message_id: str) -> Message:
     body = extract_body_from_mail(message_data)
 
     attachments = extract_attachments_from_mail(access_token, message_data)
+    body_len = 0
+    attachments_len = 0
+    if body is not None :
+        body_len = len(body)
+
+    if attachments is not None:
+        attachments_len = len(attachments)
+
+    
+        
     # print("subject: ")
     # print(subject)
     # print("company name: ")
@@ -72,7 +82,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)
+    return Message(message_id=message_id, body=body, attachments=attachments, company=company_from_mail , body_len , attachments_len)
 
 
 def extract_subject_from_mail(message_data: dict) -> str: