Update test.py

test.py

@@ -61,14 +61,41 @@ async def auth_google(request: Request):
     # print(response.json())
     # print(access_token_new)
     user_info = requests.get("https://www.googleapis.com/oauth2/v1/userinfo", headers={"Authorization": f"Bearer {access_token_new}"})
-    # user_query = data.get("user_query") 
-    query = "subject:receipt OR subject:receipts has:attachment"
-    gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={query}"
-    gmail_response = requests.get(gmail_url, headers={"Authorization": f"Bearer {access_token_new}"})
-    messages = gmail_response.json().get("messages", [])
-    print(messages)
-    print("Printing gmail response")
-    print(gmail_response.json())
+    # user_query = data.get("user_query")
+    page_token = None
+    messages = []
+    jobs_query = "subject:IOS developer has:attachment"
+
+    while True:
+        # Construct Gmail API request with pageToken
+        gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={jobs_query}"
+        if page_token:
+            gmail_url += f"&pageToken={page_token}"
+
+        gmail_response = requests.get(gmail_url, headers={"Authorization": f"Bearer {access_token_new}"})
+        gmail_data = gmail_response.json()
+
+        # Check if there are messages in the response
+        if "messages" in gmail_data:
+            messages.extend(gmail_data["messages"])
+
+        # Check if there are more pages
+        if "nextPageToken" in gmail_data:
+            page_token = gmail_data["nextPageToken"]
+        else:
+            break  # No more pages, exit the loop
+
+
+    print(len(messages))
+
+    
+    # query = "subject:receipt OR subject:receipts has:attachment"
+    # gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={query}"
+    # gmail_response = requests.get(gmail_url, headers={"Authorization": f"Bearer {access_token_new}"})
+    # messages = gmail_response.json().get("messages", [])
+    # print(messages)
+    # print("Printing gmail response")
+    # print(gmail_response.json())
     # Fetch attachments from the first message
     attachments = []
     attachment_no = 0