Omkar008 commited on
Commit
067db0f
1 Parent(s): 83d8c1d

Update controllers/ws_controller.py

Browse files
Files changed (1) hide show
  1. controllers/ws_controller.py +2 -0
controllers/ws_controller.py CHANGED
@@ -197,6 +197,8 @@ def extract_attachments_from_mail(access_token: str, message_data: dict) -> List
197
  attachment_id = part["body"]["attachmentId"]
198
  attachment_data = fetch_attachment_data(access_token, message_data["id"], attachment_id)
199
  filename = part.get("filename", "untitled.txt")
 
 
200
  data = attachment_data.get("data", "")
201
  raw_text=ut.extract_text_from_attachment(filename , data)
202
  struct_data = ut.strcuture_document_data(raw_text)
 
197
  attachment_id = part["body"]["attachmentId"]
198
  attachment_data = fetch_attachment_data(access_token, message_data["id"], attachment_id)
199
  filename = part.get("filename", "untitled.txt")
200
+ if filename.endswith(".zip"):
201
+ continue
202
  data = attachment_data.get("data", "")
203
  raw_text=ut.extract_text_from_attachment(filename , data)
204
  struct_data = ut.strcuture_document_data(raw_text)