86 lines
2.1 KiB
Python
86 lines
2.1 KiB
Python
import os
|
||
import logging
|
||
from logging.handlers import RotatingFileHandler
|
||
from datetime import datetime
|
||
|
||
_loggers = {}
|
||
|
||
|
||
def get_logger(name='app'):
|
||
"""
|
||
获取日志记录器
|
||
|
||
Args:
|
||
name: 日志名称（对应日志文件名）
|
||
|
||
Returns:
|
||
logging.Logger: 日志记录器
|
||
"""
|
||
if name in _loggers:
|
||
return _loggers[name]
|
||
|
||
# 日志目录
|
||
log_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), 'logs')
|
||
os.makedirs(log_dir, exist_ok=True)
|
||
|
||
# 日志文件路径
|
||
log_file = os.path.join(log_dir, f'{name}.log')
|
||
|
||
# 创建 logger
|
||
logger = logging.getLogger(name)
|
||
logger.setLevel(logging.INFO)
|
||
logger.propagate = False
|
||
|
||
# 避免重复添加 handler
|
||
if logger.handlers:
|
||
_loggers[name] = logger
|
||
return logger
|
||
|
||
# 日志格式
|
||
formatter = logging.Formatter(
|
||
'%(asctime)s | %(levelname)-7s | %(message)s',
|
||
datefmt='%Y-%m-%d %H:%M:%S'
|
||
)
|
||
|
||
# 文件 handler（按大小轮转，最多 5 个备份，每个 10MB）
|
||
file_handler = RotatingFileHandler(
|
||
log_file,
|
||
maxBytes=10 * 1024 * 1024,
|
||
backupCount=5,
|
||
encoding='utf-8'
|
||
)
|
||
file_handler.setLevel(logging.INFO)
|
||
file_handler.setFormatter(formatter)
|
||
logger.addHandler(file_handler)
|
||
|
||
# 控制台 handler
|
||
console_handler = logging.StreamHandler()
|
||
console_handler.setLevel(logging.INFO)
|
||
console_handler.setFormatter(formatter)
|
||
logger.addHandler(console_handler)
|
||
|
||
_loggers[name] = logger
|
||
return logger
|
||
|
||
|
||
def log_info(message, module='app'):
|
||
"""记录信息日志"""
|
||
get_logger(module).info(message)
|
||
|
||
|
||
def log_error(message, module='app'):
|
||
"""记录错误日志"""
|
||
get_logger(module).error(message)
|
||
import traceback
|
||
get_logger(module).error(traceback.format_exc())
|
||
|
||
|
||
def log_warning(message, module='app'):
|
||
"""记录警告日志"""
|
||
get_logger(module).warning(message)
|
||
|
||
|
||
def log_debug(message, module='app'):
|
||
"""记录调试日志"""
|
||
get_logger(module).debug(message)
|