67 lines
1.9 KiB
Python
67 lines
1.9 KiB
Python
|
"""
|
|||
|
|
loguru 统一日志配置 + InterceptHandler 拦截标准库 logging。
|
|||
|
|
"""
|
|||
|
|
|
|||
|
|
import logging
|
|||
|
|
import sys
|
|||
|
|
|
|||
|
|
from loguru import logger
|
|||
|
|
|
|||
|
|
|
|||
|
|
class InterceptHandler(logging.Handler):
|
|||
|
|
"""Route standard-library logging messages into loguru."""
|
|||
|
|
|
|||
|
|
def emit(self, record: logging.LogRecord) -> None:
|
|||
|
|
try:
|
|||
|
|
level = logger.level(record.levelname).name
|
|||
|
|
except ValueError:
|
|||
|
|
level = record.levelno
|
|||
|
|
|
|||
|
|
frame, depth = logging.currentframe(), 2
|
|||
|
|
while frame and frame.f_code.co_filename == logging.__file__:
|
|||
|
|
frame = frame.f_back
|
|||
|
|
depth += 1
|
|||
|
|
|
|||
|
|
logger.opt(depth=depth, exception=record.exc_info).log(level, record.getMessage())
|
|||
|
|
|
|||
|
|
|
|||
|
|
def setup_logging() -> None:
|
|||
|
|
"""Call once at application startup, before any other import that logs."""
|
|||
|
|
logger.remove()
|
|||
|
|
|
|||
|
|
logger.add(
|
|||
|
|
sys.stderr,
|
|||
|
|
level="INFO",
|
|||
|
|
format=(
|
|||
|
|
"<green>{time:YYYY-MM-DD HH:mm:ss.SSS}</green> | "
|
|||
|
|
"<level>{level: <8}</level> | "
|
|||
|
|
"<cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> | "
|
|||
|
|
"{extra[request_id]} | "
|
|||
|
|
"<level>{message}</level>"
|
|||
|
|
),
|
|||
|
|
backtrace=True,
|
|||
|
|
diagnose=False,
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
# 将根 logger 重定向到 loguru，不再使用 basicConfig（文档要求：统一走 loguru）
|
|||
|
|
root = logging.getLogger()
|
|||
|
|
root.handlers = [InterceptHandler()]
|
|||
|
|
root.setLevel(0)
|
|||
|
|
|
|||
|
|
for name in (
|
|||
|
|
"uvicorn",
|
|||
|
|
"uvicorn.error",
|
|||
|
|
"uvicorn.access",
|
|||
|
|
"sqlalchemy.engine",
|
|||
|
|
"celery",
|
|||
|
|
"celery.worker",
|
|||
|
|
):
|
|||
|
|
logging.getLogger(name).handlers = [InterceptHandler()]
|
|||
|
|
|
|||
|
|
logger.configure(extra={"request_id": "-"})
|
|||
|
|
|
|||
|
|
|
|||
|
|
def get_logger(name: str) -> logging.Logger:
|
|||
|
|
"""获取具名 logger，统一走 loguru 拦截。各模块应使用此函数而非直接 import logging。"""
|
|||
|
|
return logging.getLogger(name)
|