fix: logging

Author: luchog01

alembic/env.py

@@ -1,8 +1,8 @@
 import asyncio
+import importlib
 import os
 import sys
 from logging.config import fileConfig
-import importlib
 from pathlib import Path
 
 # Add the project root directory to the Python path
@@ -20,7 +20,9 @@
 src_path = Path(__file__).parent.parent / "api" / "src"
 for path in src_path.rglob("*.py"):
     if path.name != "__init__.py":
-        module_path = str(path.relative_to(Path(__file__).parent.parent)).replace(os.sep, ".")[:-3]
+        module_path = str(path.relative_to(Path(__file__).parent.parent)).replace(
+            os.sep, "."
+        )[:-3]
         try:
             importlib.import_module(module_path)
         except Exception as e:

api/core/logging.py

@@ -1,48 +1,18 @@
 import logging
 import sys
-from datetime import datetime
 
 from api.core.config import settings
 
 
-class SimpleFormatter(logging.Formatter):
-    """Simple formatter with colored output for console."""
-
-    COLORS = {
-        "DEBUG": "\033[36m",  # Cyan
-        "INFO": "\033[32m",  # Green
-        "WARNING": "\033[33m",  # Yellow
-        "ERROR": "\033[31m",  # Red
-        "RESET": "\033[0m",  # Reset
-    }
-
-    def format(self, record: logging.LogRecord) -> str:
-        color = self.COLORS.get(record.levelname, "")
-        reset = self.COLORS["RESET"]
-
-        # Format: [time] [level] [module:line] message
-        return (
-            f"{color}[{datetime.now().strftime('%H:%M:%S')}] "
-            f"[{record.levelname}] [{record.name}:{record.lineno}] "
-            f"{record.getMessage()}{reset}"
-        )
-
-
 def setup_logging() -> None:
     """Set up logging configuration."""
-    # Create and configure console handler
-    console_handler = logging.StreamHandler(sys.stdout)
-    console_handler.setFormatter(SimpleFormatter())
-
-    # Configure root logger
-    root_logger = logging.getLogger()
-    root_logger.setLevel(logging.DEBUG if settings.DEBUG else logging.INFO)
-    root_logger.addHandler(console_handler)
-
-    # Configure uvicorn loggers
-    for logger_name in ("uvicorn", "uvicorn.access", "uvicorn.error"):
-        logger = logging.getLogger(logger_name)
-        logger.handlers = [console_handler]
+    format_string = "[%(asctime)s] [%(levelname)s] [%(name)s] %(message)s"
+    logging.basicConfig(
+        level=logging.DEBUG if settings.DEBUG else logging.INFO,
+        format=format_string,
+        datefmt="%H:%M:%S",
+        stream=sys.stdout,
+    )
 
 
 def get_logger(name: str) -> logging.Logger: