init

shared/logger/logger.py

@@ -0,0 +1,55 @@
+import logging
+import subprocess
+import sys
+
+import pydantic
+from pydantic.fields import Field
+
+from .console_formatter import ColoredConsoleFormatter
+from .json_formatter import JSONFormatter
+
+
+def get_git_commit() -> str:
+    try:
+        return subprocess.check_output(['git', 'rev-parse', '--short', 'HEAD'], text=True).strip()
+    except Exception:
+        return 'unknown'
+
+
+class LoggerConfig(pydantic.BaseModel):
+    APP_NAME: str
+    APP_VERSION: str = Field(default_factory=get_git_commit)
+    LEVEL: int = logging.INFO
+    PRETTY_CONSOLE: bool = False
+
+
+def init(config: LoggerConfig) -> None:
+    root_logger = logging.getLogger()
+    root_logger.setLevel(config.LEVEL)
+    root_logger.handlers.clear()
+
+    handler = logging.StreamHandler(sys.stdout)
+    handler.setLevel(config.LEVEL)
+
+    formatter = ColoredConsoleFormatter() if config.PRETTY_CONSOLE else JSONFormatter()
+    handler.setFormatter(formatter)
+    root_logger.addHandler(handler)
+
+    for logger_name in ['uvicorn', 'uvicorn.access', 'uvicorn.error']:
+        uvicorn_logger = logging.getLogger(logger_name)
+        uvicorn_logger.handlers = [handler]
+        uvicorn_logger.propagate = False
+
+    old_factory = logging.getLogRecordFactory()
+
+    def record_factory(*args, **kwargs) -> logging.LogRecord:
+        record = old_factory(*args, **kwargs)
+        record.app_name = config.APP_NAME
+        record.app_version = config.APP_VERSION
+        return record
+
+    logging.setLogRecordFactory(record_factory)
+
+    logging.info(
+        'Logger initialized', extra={'app': config.APP_NAME, 'version': config.APP_VERSION}
+    )