Update api/logger.py

api/logger.py

@@ -2,11 +2,11 @@ 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
+        # Include the chat_id in the log message if provided
+        chat_id = self.extra.get('chat_id', 'N/A')
+        return f"[Chat ID: {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)
@@ -16,14 +16,6 @@ def setup_logger(name, chat_id=None):
         console_handler = logging.StreamHandler()
         console_handler.setFormatter(formatter)
         logger.addHandler(console_handler)
-        
-        # File Handler - Error Level (uncomment if needed)
-        # error_file_handler = logging.FileHandler('error.log')
-        # 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
+    
+    # Ensure we return a LoggerAdapter with or without chat_id
+    return ChatIDLoggerAdapter(logger, {'chat_id': chat_id})