Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -81,13 +81,13 @@ def fetch_message_data(access_token: str, message_id: str) -> Message:
     message_response = requests.get(message_url, headers={"Authorization": f"Bearer {access_token}"})
     message_data = message_response.json()
     # print(message_data)
-    subject = extract_subject_from_mail(message_data)
+    # subject = extract_subject_from_mail(message_data)
     company_from_mail = extract_domain_name(message_data['payload']['headers'], subject)
 
-    body = extract_body_from_mail(message_data)
+    # body = extract_body_from_mail(message_data)
 
     attachments,structed_attachment_data = extract_attachments_from_mail(access_token, message_data)
-    high_level_company_type = get_company_type(company_from_mail)
+    # high_level_company_type = get_company_type(company_from_mail)
     # structed_attachment_data = extract_json_from_attachments(access_token , message_data)
 
 
@@ -104,7 +104,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_len=body_len,body=body, attachments=attachments, company=company_from_mail,high_level_company_type=high_level_company_type,structured_data = structed_attachment_data)
+    return Message(message_id=message_id,company=company_from_mail,structured_data = structed_attachment_data)