Update controllers/ws_controller_1.py

controllers/ws_controller_1.py

@@ -144,7 +144,7 @@ async def get_messages(code: str, websocket: WebSocket,start, brand_name: Option
     
     page_token = None
     processed_count = 0
-
+    skipped_count = 0
     async with aiohttp.ClientSession() as session:
         while True:
             gmail_url = f"https://www.googleapis.com/gmail/v1/users/me/messages?q={g_query}&maxResults=50"
@@ -165,21 +165,19 @@ async def get_messages(code: str, websocket: WebSocket,start, brand_name: Option
             
             for future in asyncio.as_completed(processing_tasks):
                 processed_message = await future
-                # await websocket.send_text("MESSAGE_START")
-                await websocket.send_text(json.dumps(
-                   
-                   processed_message.to_json(),
-                    
-                ))
-                await websocket.send_text("NEXT_MESSAGE")
-                
-                processed_count += 1
+                if processed_message is not None:
+                    await websocket.send_text(json.dumps(processed_message.to_json()))
+                    await websocket.send_text("NEXT_MESSAGE")
+                    processed_count += 1
+                else:
+                    skipped_count += 1
 
             if "nextPageToken" in gmail_data:
                 page_token = gmail_data["nextPageToken"]
             else:
                 break
-
+    logging.info(f"Processed Message {processed_count}")
+    logging.info(f"Skipped Message {skipped_count}")
 async def send_message_in_chunks(websocket: WebSocket, message_json: dict, chunk_size: int):
     message_str = json.dumps(message_json)
     for i in range(0, len(message_str), chunk_size):
@@ -218,6 +216,7 @@ async def process_message(session: aiohttp.ClientSession, access_token: str, mes
     else:
         # If 'payload' or 'headers' is missing, use a default value or extract from another field
         company_from_mail = "unknown"  # or some other default value
+        return None
         logging.warning(f"Payload or headers missing for message ID: {message_data.get('id', 'unknown')}")
 
     body = extract_body_from_mail(message_data)