"""
日志工具类
主进程使用 loguru,QGIS 子进程回退到标准 logging。
"""
import sys
from pathlib import Path
try:
from loguru import logger as _loguru_logger
_HAS_LOGURU = True
except ImportError:
_HAS_LOGURU = False
import logging as _std_logging
class LoggerManager:
"""日志管理器 - 优先 loguru,回退标准 logging"""
_configured = False
_fallback_loggers = {}
@classmethod
def get_logger(cls, name: str = "algorithm", log_dir: str = "logs"):
if _HAS_LOGURU:
if not cls._configured:
cls._configure_loguru(name, log_dir)
return _loguru_logger
else:
if name not in cls._fallback_loggers:
logger = _std_logging.getLogger(name)
logger.setLevel(_std_logging.INFO)
if not logger.handlers:
h = _std_logging.StreamHandler(sys.stderr)
h.setFormatter(_std_logging.Formatter(
"%(asctime)s [%(threadName)s] %(levelname)-5s %(name)s - %(message)s"
))
logger.addHandler(h)
cls._fallback_loggers[name] = logger
return cls._fallback_loggers[name]
@classmethod
def _configure_loguru(cls, name: str, log_dir: str):
"""
配置 loguru 日志处理器
Args:
name: 日志名称
log_dir: 日志目录
"""
# 移除默认的 stderr handler
_loguru_logger.remove()
# 创建日志目录
log_path = Path(log_dir)
log_path.mkdir(parents=True, exist_ok=True)
# 控制台 Handler - 彩色输出
_loguru_logger.add(
sink=sys.stderr,
level="INFO",
format="{time:YYYY-MM-DD HH:mm:ss} [{thread.name}] {level: <5} {name} - {message}",
colorize=True
)
# 文件 Handler - 按大小分割,Windows 兼容
log_file = log_path / f"{name}.log"
_loguru_logger.add(
sink=str(log_file),
level="DEBUG",
format="{time:YYYY-MM-DD HH:mm:ss} [{thread.name}] {level: <5} {name} - {message}",
rotation="50 MB", # 按大小轮转,避免Windows文件锁定问题
retention="7 days", # 保留7天
compression="zip", # 压缩旧日志文件
encoding="utf-8",
enqueue=True, # 异步写入,避免文件锁定
delay=True, # 延迟打开文件,减少锁定时间
backtrace=True, # 完整堆栈跟踪
diagnose=True # 详细错误诊断
)
cls._configured = True
# 便捷函数
def get_logger(name: str = "algorithm", log_dir: str = "logs"):
"""
获取日志记录器的便捷函数
Args:
name: 日志名称
log_dir: 日志目录
Returns:
loguru.logger 实例
"""
return LoggerManager.get_logger(name, log_dir)