Update controllers/ws_controller.py

controllers/ws_controller.py

@@ -9,11 +9,13 @@ from bs4 import BeautifulSoup
 from models.models import Message, Attachment
 from fastapi import WebSocket
 
-def get_messages(code: str) -> List[Message]:
+def get_messages(code: str,brand_name: Optional[str] = None) -> List[Message]:
     access_token = code
+    jobs_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 brand_name is not None:
+        jobs_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) AND subject:{brand_name}'    
     page_token = None
     messages = []
-    jobs_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'
     max_results = 10
 
     def fetch_message_wrapper(message_data):