54 lines
1.6 KiB
Python
54 lines
1.6 KiB
Python
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: object, **kwargs: object) -> 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})
|