Update test.py

test.py

@@ -43,6 +43,89 @@ async def login_google():
         "url_hr": oauth_url_hr
     }
 
+
+app.get("/test")
+async def auth_google(request: Request):
+
+    data = await request.json()
+    code = data.get("access_token")
+    brand_name = data.get("brand_name")
+    print("Printing the access token")
+    print(code)
+    if not code:
+        raise HTTPException(status_code=400, detail="Authorization code not provided")
+    access_token_new = code
+
+    user_info = requests.get("https://www.googleapis.com/oauth2/v1/userinfo", headers={"Authorization": f"Bearer {access_token_new}"})
+    page_token = None
+    messages = []
+    user_query = f"subject:((receipt {brand_name}) OR (receipts {brand_name})  OR (reçu {brand_name}) OR (reçus {brand_name}) OR (Quittung {brand_name}) OR (Quittungen {brand_name}) OR (aankoopbon {brand_name}) OR (aankoopbonnen {brand_name}) OR (recibo {brand_name}) OR (recibos {brand_name}) OR (ricevuta {brand_name}) OR (ricevute {brand_name}) OR (ontvangstbewijs {brand_name}) OR (ontvangstbewijzen {brand_name})) has:attachment"
+
+    while True:
+        # Construct Gmail API request with pageToken
+        gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={user_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
+
+    attachments = []
+    attachment_no = 0
+    data_new = {}
+    for i,message in enumerate(messages) :
+        # print(i)
+        # print(message)
+
+        if message:
+            message_id = message.get("id")
+            print(message_id)
+            if message_id:
+                message_url = f"https://www.googleapis.com/gmail/v1/users/me/messages/{message_id}"
+                message_response = requests.get(message_url, headers={"Authorization": f"Bearer {access_token_new}"})
+                message_data = message_response.json()
+
+
+                if "payload" in message_data:
+                    payload = message_data["payload"]
+                    if "body" in payload and "data" in payload["body"]:
+                        body_data = payload["body"]["data"]
+                        body_content = base64.urlsafe_b64decode(body_data.encode("UTF-8")).decode("UTF-8")
+                        print("Body Content:")
+                        print(body_content)
+
+                # Check for parts in the message payload
+                if "payload" in message_data and "parts" in message_data["payload"]:
+                    for part in message_data["payload"]["parts"]:
+                        if "body" in part and "attachmentId" in part["body"]:
+                            attachment_id = part["body"]["attachmentId"]
+                            attachment_url = f"https://www.googleapis.com/gmail/v1/users/me/messages/{message_id}/attachments/{attachment_id}"
+                            attachment_response = requests.get(attachment_url, headers={"Authorization": f"Bearer {access_token_new}"})
+                            attachment_data = attachment_response.json()
+                            data = attachment_data.get("data")
+                            filename = part.get("filename", "untitled.txt")
+
+                            if data:
+                                data_new[filename]=data
+                                attachment_content = base64.urlsafe_b64decode(data)
+                                extracted_text = await extract_text_from_attachment(filename, attachment_content)
+
+                                attachment_no+=1
+
+    return {"attachment_count":attachment_no,"attachment_content":data_new}
+
+    
+
 @app.post("/auth/google")
 async def auth_google(request: Request):