Update api/logger.py

api/logger.py

@@ -1,15 +1,16 @@
 import logging
 
+class ChatIDLoggerAdapter(logging.LoggerAdapter):
+    def process(self, msg, kwargs):
+        # Prepend chat_id to every log message
+        return f"[Chat ID: {self.extra['chat_id']}] {msg}", kwargs
+
 def setup_logger(name, chat_id=None):
+    # Set up base logger
     logger = logging.getLogger(name)
     if not logger.handlers:
         logger.setLevel(logging.INFO)
-        
-        # Modify the formatter to include chat_id if available
-        if chat_id:
-            formatter = logging.Formatter(f'%(asctime)s - %(name)s - %(levelname)s - [Chat ID: {chat_id}] - %(message)s')
-        else:
-            formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
+        formatter = logging.Formatter('%(asctime)s - %(name)s - %(levelname)s - %(message)s')
         
         # Console handler
         console_handler = logging.StreamHandler()
@@ -21,5 +22,8 @@ def setup_logger(name, chat_id=None):
         # error_file_handler.setFormatter(formatter)
         # error_file_handler.setLevel(logging.ERROR)
         # logger.addHandler(error_file_handler)
-    
+
+    # Wrap the logger with ChatIDLoggerAdapter if chat_id is provided
+    if chat_id:
+        return ChatIDLoggerAdapter(logger, {'chat_id': chat_id})
     return logger