Update test.py

test.py

@@ -71,7 +71,8 @@ async def test_google(code:str):
     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"
     # user_query = f"{brand_name} label:^smartlabel_receipt"
-    user_query = """("invoice" OR (("tracking" OR "track") AND ("delivery" OR "package"))) OR (subject:order OR subject:receipt OR subject:receipts OR subject:invoice OR subject:invoice)"""
+    user_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:chanel"
+    # user_query = """("invoice" OR (("tracking" OR "track") AND ("delivery" OR "package"))) OR (subject:order OR subject:receipt OR subject:receipts OR subject:invoice OR subject:invoice)"""
     while True:
         # Construct Gmail API request with pageToken
         
@@ -116,17 +117,22 @@ async def test_google(code:str):
                 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}"})
                 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)
-                        print()
-                        print()
+                print(message_data)
+
+            if 'payload' in message and 'headers' in message['payload']:
+                headers = message['payload']['headers']
+                for header in headers:
+                    if header['name'] == 'Subject':
+                        subject = header['value']
+                
+                if 'parts' in message['payload']:
+                    parts = message['payload']['parts']
+                    for part in parts:
+                        if part['mimeType'] == 'text/plain':
+                            body_data = part['body']['data']
+                            body = base64.urlsafe_b64decode(body_data).decode()
+            print("Subject:", subject)
+            print("Body:", body)
 
                 # Check for parts in the message payload
                 if "payload" in message_data and "parts" in message_data["payload"]: