Update get_gmail_data.py

get_gmail_data.py

@@ -56,24 +56,28 @@ class GmailDataExtractor:
         receipt_query = f'(subject:"your order" OR subject:receipts OR subject:receipt OR  subject: aankoopbon  OR subject:reçu OR subject:invoice OR subject:invoice OR category:purchases)'
         # if self.__user_input is not None:
         #     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})&maxResults=15'
-        gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?maxResults=10&q={receipt_query}"
-        def __fetch_page(url):
-            response = requests.get(url, headers={"Authorization": f"Bearer {access_token}"})
-            response.raise_for_status()  # Raise error if the request fails
-            data = response.json()
-            return data.get("messages", []), data.get("nextPageToken")
-
-        messages = []
-        page_token = None
-        try:
-            while True:
-                url = f"{gmail_url}&pageToken={page_token}" if page_token else gmail_url
-                page_messages, page_token = __fetch_page(url)
-                messages.extend(page_messages)
-                if not page_token:
-                    break
-        except requests.RequestException as e:
-            raise RuntimeError(f"Error fetching messages from Gmail API: {str(e)}")
+        gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={receipt_query}&maxResults=10"
+        gmail_response = requests.get(gmail_url, headers={"Authorization": f"Bearer {access_token}"})
+        gmail_data = gmail_response.json()
+        messages=[]
+        messages.extend(gmail_data)
+        # def __fetch_page(url):
+        #     response = requests.get(url, headers={"Authorization": f"Bearer {access_token}"})
+        #     response.raise_for_status()  # Raise error if the request fails
+        #     data = response.json()
+        #     return data.get("messages", []), data.get("nextPageToken")
+
+        # messages = []
+        # page_token = None
+        # try:
+        #     while True:
+        #         url = f"{gmail_url}&pageToken={page_token}" if page_token else gmail_url
+        #         page_messages, page_token = __fetch_page(url)
+        #         messages.extend(page_messages)
+        #         if not page_token:
+        #             break
+        # except requests.RequestException as e:
+        #     raise RuntimeError(f"Error fetching messages from Gmail API: {str(e)}")
 
         print(len(messages))
         return messages