Omkar008 commited on
Commit
41d97d6
1 Parent(s): 03a6290

Update get_gmail_data.py

Browse files
Files changed (1) hide show
  1. get_gmail_data.py +6 -3
get_gmail_data.py CHANGED
@@ -135,10 +135,13 @@ class GmailDataExtractor:
135
  message_id = message.get("id")
136
  if not message_id:
137
  return None, None, [], False
138
-
139
  message_data = self.__fetch_message_data(message_id)
140
- subject = message_data.get('payload', {}).get('headers', {}).get('value', '')
141
-
 
 
 
142
  body = ''
143
  links = []
144
  has_attachment = False
 
135
  message_id = message.get("id")
136
  if not message_id:
137
  return None, None, [], False
138
+ subject=''
139
  message_data = self.__fetch_message_data(message_id)
140
+ if 'payload' in message_data and 'headers' in message_data['payload']:
141
+ headers = message_data['payload']['headers']
142
+ for header in headers:
143
+ if header['name'] == 'Subject':
144
+ subject = header['value']
145
  body = ''
146
  links = []
147
  has_attachment = False