65 lines
2.1 KiB
Python
65 lines
2.1 KiB
Python
|
import logging
|
||
|
|
from pathlib import Path
|
||
|
|
from .config import Logging
|
||
|
|
from logging.handlers import RotatingFileHandler
|
||
|
|
|
||
|
|
|
||
|
|
class ExcludeWatchFilesFilter(logging.Filter):
|
||
|
|
def filter(self, record):
|
||
|
|
# 如果日志记录来自 watchfiles 模块，则不记录（返回 False）
|
||
|
|
return not record.name.startswith("watchfiles")
|
||
|
|
|
||
|
|
|
||
|
|
def setup_logging(log_config: Logging):
|
||
|
|
log_level = getattr(logging, log_config["level"], logging.INFO)
|
||
|
|
log_format = log_config["format"]
|
||
|
|
# 创建文件夹
|
||
|
|
log_file_path = Path(log_config["file"])
|
||
|
|
log_file_dir = log_file_path.parent
|
||
|
|
if log_file_dir != Path("."):
|
||
|
|
log_file_dir.mkdir(parents=True, exist_ok=True)
|
||
|
|
|
||
|
|
# 设置特定日志级别的过滤
|
||
|
|
logging.getLogger("watchfiles").setLevel(logging.WARNING)
|
||
|
|
|
||
|
|
# 配置根日志记录器
|
||
|
|
root_logger = logging.getLogger()
|
||
|
|
root_logger.setLevel(log_level)
|
||
|
|
|
||
|
|
# 创建格式化器
|
||
|
|
formatter = logging.Formatter(log_format)
|
||
|
|
|
||
|
|
# 创建轮转文件处理器 - 限制文件大小为 10MB，保留 5 个备份文件
|
||
|
|
max_bytes = log_config.get("max_size", 10 * 1024 * 1024) # 默认 10MB
|
||
|
|
backup_count = log_config.get("backup_count", 5) # 默认保留 5 个备份
|
||
|
|
|
||
|
|
file_handler = RotatingFileHandler(
|
||
|
|
log_config["file"],
|
||
|
|
encoding="utf-8",
|
||
|
|
maxBytes=max_bytes,
|
||
|
|
backupCount=backup_count,
|
||
|
|
)
|
||
|
|
file_handler.setFormatter(formatter)
|
||
|
|
file_handler.addFilter(ExcludeWatchFilesFilter())
|
||
|
|
|
||
|
|
# 创建控制台处理器并添加过滤器
|
||
|
|
console_handler = logging.StreamHandler()
|
||
|
|
console_handler.setFormatter(formatter)
|
||
|
|
console_handler.addFilter(ExcludeWatchFilesFilter())
|
||
|
|
|
||
|
|
# 添加处理器到根日志记录器
|
||
|
|
root_logger.addHandler(file_handler)
|
||
|
|
root_logger.addHandler(console_handler)
|
||
|
|
|
||
|
|
print(
|
||
|
|
f"日志系统已配置 - 级别: {logging.getLevelName(log_level)}, 文件: {log_config['file']} "
|
||
|
|
f"(最大：{max_bytes / 1024 / 1024:.1f}MB, 备份：{backup_count})"
|
||
|
|
)
|
||
|
|
|
||
|
|
|
||
|
|
def get_logger(name: str) -> logging.Logger:
|
||
|
|
logger = logging.getLogger(name)
|
||
|
|
# 为每个logger添加过滤器
|
||
|
|
logger.addFilter(ExcludeWatchFilesFilter())
|
||
|
|
return logger
|