refactor: improve logger setup and integrate caching for performance

Author: janezlapajne

source/core/__init__.py

@@ -1,4 +1,3 @@
-from .logger import logger
-from .settings import settings
+from .settings import logger, settings
 
 __all__ = ["logger", "settings"]

source/core/logger.py

@@ -1,19 +1,23 @@
 import logging
 import sys
+from functools import lru_cache
 
-from loguru import logger
-
-from .settings import settings
+import loguru
+from loguru import logger as _logger
 
 
 class InterceptHandler(logging.Handler):
     def emit(self, record: logging.LogRecord) -> None:
-        logger_opt = logger.opt(depth=7, exception=record.exc_info)
+        logger_opt = _logger.opt(depth=7, exception=record.exc_info)
         logger_opt.log(record.levelname, record.getMessage())
 
 
-LOGGING_LEVEL = logging.DEBUG if settings.debug else logging.INFO
-logging.basicConfig(
-    handlers=[InterceptHandler(level=LOGGING_LEVEL)], level=LOGGING_LEVEL
-)
-logger.configure(handlers=[{"sink": sys.stderr, "level": LOGGING_LEVEL}])
+@lru_cache()
+def setup_logger(debug: bool = False) -> loguru._Logger:  # type: ignore
+    LOGGING_LEVEL = logging.DEBUG if debug else logging.INFO
+    logging.basicConfig(
+        handlers=[InterceptHandler(level=LOGGING_LEVEL)],
+        level=LOGGING_LEVEL,
+    )
+    _logger.configure(handlers=[{"sink": sys.stderr, "level": LOGGING_LEVEL}])
+    return _logger

source/core/settings.py

@@ -5,6 +5,8 @@
 from pydantic import Field, field_validator
 from pydantic_settings import BaseSettings, SettingsConfigDict
 
+from .logger import setup_logger
+
 DOTENV_PATH = ".env"
 
 BASE_DIR = Path(__file__).parent.parent.parent.absolute()
@@ -85,3 +87,4 @@ def set_artifacts_dir(cls, v, values):
 
 
 settings = Settings()
+logger = setup_logger(settings.debug)