71 lines
1.5 KiB
Python
71 lines
1.5 KiB
Python
from loguru import logger as _logger
|
|
import sys
|
|
import os
|
|
from app.utils.config import LogConfig
|
|
|
|
config = LogConfig()
|
|
LOG_DIR = config.log_dir
|
|
os.makedirs(LOG_DIR, exist_ok=True)
|
|
|
|
|
|
_logger.remove()
|
|
|
|
console_format = (
|
|
"<green>{time:YYYY-MM-DD HH:mm:ss}</green>| "
|
|
"<level>{level: <8}</level> | "
|
|
"<yellow>{extra[module]}</yellow>- {message}"
|
|
)
|
|
|
|
_logger.add(
|
|
sink=sys.stderr,
|
|
format=console_format,
|
|
level=config.level,
|
|
colorize=True,
|
|
enqueue=True,
|
|
)
|
|
|
|
|
|
|
|
|
|
file_format = "{time:YYYY-MM-DD HH:mm:ss}| {level: <8} | {extra[module]}- {message}"
|
|
|
|
_logger.add(
|
|
sink=os.path.join(LOG_DIR, "app_{time:YYYY-MM-DD}.log"),
|
|
format=file_format,
|
|
level=config.writelevel,
|
|
rotation="00:00",
|
|
retention="7 days",
|
|
colorize=False,
|
|
encoding="utf-8",
|
|
enqueue=True,
|
|
)
|
|
|
|
_logger = _logger.patch(lambda record: record["extra"].setdefault("module", "未知模块"))
|
|
|
|
|
|
def get_logger(module_name: str):
|
|
"""
|
|
获取一个绑定 module 名的日志器
|
|
:param module_name: 模块名称，如 "用户管理"
|
|
:return: 绑定后的 logger
|
|
"""
|
|
return _logger.bind(module=module_name)
|
|
|
|
|
|
__all__ = ["get_logger"]
|
|
|
|
|
|
if __name__ == "__main__":
|
|
logger = get_logger("test1")
|
|
logger.debug("调试信息（只在控制台显示）")
|
|
logger.info("这是普通信息（控制台 + 文件）")
|
|
logger.warning("这是警告")
|
|
logger.error("发生错误")
|
|
logger2 = get_logger("test2")
|
|
logger2.error("发生错误")
|
|
|
|
try:
|
|
_ = 1 / 0
|
|
except Exception:
|
|
logger.exception("捕获异常")
|