wip: replace AppLogger with AppStructLogger

Author: grillazz

app/api/health.py

@@ -6,9 +6,9 @@
 from starlette.concurrency import run_in_threadpool
 
 from app.services.smtp import SMTPEmailService
-from app.utils.logging import AppLogger
+from app.utils.logging import AppStructLogger
 
-logger = AppLogger().get_logger()
+logger = AppStructLogger().get_logger()
 
 router = APIRouter()
 

app/api/ml.py

@@ -4,9 +4,9 @@
 from fastapi.responses import StreamingResponse
 
 from app.services.llm import get_llm_service
-from app.utils.logging import AppLogger
+from app.utils.logging import AppStructLogger
 
-logger = AppLogger().get_logger()
+logger = AppStructLogger().get_logger()
 
 router = APIRouter()
 

app/api/stuff.py

@@ -5,9 +5,9 @@
 from app.database import get_db
 from app.models.stuff import Stuff
 from app.schemas.stuff import StuffResponse, StuffSchema
-from app.utils.logging import AppLogger
+from app.utils.logging import AppStructLogger
 
-logger = AppLogger().get_logger()
+logger = AppStructLogger().get_logger()
 
 router = APIRouter(prefix="/v1/stuff")
 

app/api/user.py

@@ -7,9 +7,9 @@
 from app.models.user import User
 from app.schemas.user import TokenResponse, UserLogin, UserResponse, UserSchema
 from app.services.auth import create_access_token
-from app.utils.logging import AppLogger
+from app.utils.logging import AppStructLogger
 
-logger = AppLogger().get_logger()
+logger = AppStructLogger().get_logger()
 
 router = APIRouter(prefix="/v1/user")
 

app/database.py

@@ -3,9 +3,9 @@
 from sqlalchemy.ext.asyncio import async_sessionmaker, create_async_engine
 
 from app.config import settings as global_settings
-from app.utils.logging import setup_structlog
+from app.utils.logging import AppStructLogger
 
-logger = setup_structlog()
+logger = AppStructLogger().get_logger()
 
 engine = create_async_engine(
     global_settings.asyncpg_url.unicode_string(),

app/main.py

@@ -15,9 +15,9 @@
 from app.config import settings as global_settings
 from app.redis import get_redis
 from app.services.auth import AuthBearer
-from app.utils.logging import setup_structlog
+from app.utils.logging import AppStructLogger
 
-logger = setup_structlog()
+logger = AppStructLogger().get_logger()
 templates = Jinja2Templates(directory=Path(__file__).parent.parent / "templates")
 
 @asynccontextmanager

app/models/base.py

@@ -6,9 +6,9 @@
 from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import DeclarativeBase, declared_attr
 
-from app.utils.logging import AppLogger
+from app.utils.logging import AppStructLogger
 
-logger = AppLogger().get_logger()
+logger = AppStructLogger().get_logger()
 
 
 class Base(DeclarativeBase):

app/services/auth.py

@@ -6,9 +6,9 @@
 
 from app.config import settings as global_settings
 from app.models.user import User
-from app.utils.logging import AppLogger
+from app.utils.logging import AppStructLogger
 
-logger = AppLogger().get_logger()
+logger = AppStructLogger().get_logger()
 
 
 async def get_from_redis(request: Request, key: str):

app/services/smtp.py

@@ -7,10 +7,10 @@
 from pydantic import EmailStr
 
 from app.config import settings as global_settings
-from app.utils.logging import AppLogger
+from app.utils.logging import AppStructLogger
 from app.utils.singleton import SingletonMetaNoArgs
 
-logger = AppLogger().get_logger()
+logger = AppStructLogger().get_logger()
 
 
 @define

app/utils/logging.py

@@ -5,30 +5,9 @@
 
 import orjson
 import structlog
-from rich.console import Console
-from rich.logging import RichHandler
 from whenever._whenever import Instant
 
-from app.utils.singleton import SingletonMeta
-
-
-class AppLogger(metaclass=SingletonMeta):
-    _logger = None
-
-    def __init__(self):
-        self._logger = logging.getLogger(__name__)
-
-    def get_logger(self):
-        return self._logger
-
-
-class RichConsoleHandler(RichHandler):
-    def __init__(self, width=200, style=None, **kwargs):
-        super().__init__(
-            console=Console(color_system="256", width=width, style=style, stderr=True),
-            **kwargs,
-        )
-
+from app.utils.singleton import SingletonMetaNoArgs
 
 
 # TODO: merge this wrapper with the one in structlog under one hood of AppLogger
@@ -50,30 +29,39 @@ def flush(self):
     def close(self):
         self.handler.close()
 
+# @define
+class AppStructLogger(metaclass=SingletonMetaNoArgs):
+    _logger = None
 
-def setup_structlog() -> structlog.BoundLogger:
-    log_date = Instant.now().py_datetime().strftime("%Y%m%d")
-    log_path = Path(f"{log_date}_{os.getpid()}.log")
-    handler = RotatingFileHandler(
-        filename=log_path,
-        mode="a",  # text mode
-        maxBytes=10 * 1024 * 1024,
-        backupCount=5,
-        encoding="utf-8"
-    )
-    file_like = BytesToTextIOWrapper(handler)
-    structlog.configure(
-        cache_logger_on_first_use=True,
-        wrapper_class=structlog.make_filtering_bound_logger(logging.INFO),
-        processors=[
-            structlog.contextvars.merge_contextvars,
-            structlog.processors.add_log_level,
-            structlog.processors.format_exc_info,
-            structlog.processors.TimeStamper(fmt="iso", utc=True),
-            structlog.processors.JSONRenderer(serializer=orjson.dumps),
-        ],
-        logger_factory=structlog.BytesLoggerFactory(
-            file=file_like
+    def __init__(self):
+        _log_date = Instant.now().py_datetime().strftime("%Y%m%d")
+        _log_path = Path(f"{_log_date}_{os.getpid()}.log")
+        _handler = RotatingFileHandler(
+            filename=_log_path,
+            mode="a",  # text mode
+            maxBytes=10 * 1024 * 1024,
+            backupCount=5,
+            encoding="utf-8"
+        )
+        structlog.configure(
+            cache_logger_on_first_use=True,
+            wrapper_class=structlog.make_filtering_bound_logger(logging.INFO),
+            processors=[
+                structlog.contextvars.merge_contextvars,
+                structlog.processors.add_log_level,
+                structlog.processors.format_exc_info,
+                structlog.processors.TimeStamper(fmt="iso", utc=True),
+                structlog.processors.JSONRenderer(serializer=orjson.dumps),
+            ],
+            logger_factory=structlog.BytesLoggerFactory(
+                file=BytesToTextIOWrapper(_handler)
+            )
         )
-    )
-    return structlog.get_logger()
+        self._logger = structlog.get_logger()
+
+    def get_logger(self) -> structlog.BoundLogger:
+        """
+        Returns:
+            structlog.BoundLogger: The configured logger instance.
+        """
+        return self._logger