Omkar008 commited on
Commit
dc7215a
1 Parent(s): d15cbc7

Update get_gmail_data.py

Browse files
Files changed (1) hide show
  1. get_gmail_data.py +2 -2
get_gmail_data.py CHANGED
@@ -47,9 +47,9 @@ class GmailDataExtractor:
47
  """currently not implementing jwt for testing purposes
48
  replace every access_token with jwt function directly which returns the access token"""
49
  access_token = self.__jwt
50
- receipt_query = f"subject:"your order" OR subject:receipts OR subject:receipt OR subject:invoice OR subject:invoice OR category:purchases"
51
  if self.__user_input is not None:
52
- receipt_query = f"(label:^smartlabel_receipt OR (subject:your AND subject:order) OR subject:receipts OR subject:receipt OR subject:invoice OR subject:invoice)) AND subject:{self.__user_input}"
53
  gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={receipt_query}"
54
  def __fetch_page(url):
55
  response = requests.get(url, headers={"Authorization": f"Bearer {access_token}"})
 
47
  """currently not implementing jwt for testing purposes
48
  replace every access_token with jwt function directly which returns the access token"""
49
  access_token = self.__jwt
50
+ receipt_query = f'subject:"your order" OR subject:receipts OR subject:receipt OR subject:invoice OR subject:invoice OR category:purchases'
51
  if self.__user_input is not None:
52
+ receipt_query = f'(subject:"your order" OR subject:receipts OR subject:receipt OR subject:invoice OR subject:invoice OR category:purchases) AND subject:{self.__user_input}'
53
  gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={receipt_query}"
54
  def __fetch_page(url):
55
  response = requests.get(url, headers={"Authorization": f"Bearer {access_token}"})