xian_algorithm_new/app/utils/logger.py

"""
日志工具类
主进程使用 loguru，QGIS 子进程（无 loguru）回退到标准 logging。
"""
import sys
from pathlib import Path

try:
    from loguru import logger
    _HAS_LOGURU = True
except ImportError:
    _HAS_LOGURU = False
    import logging


class LoggerManager:
    """日志管理器"""

    _configured = False

    @classmethod
    def get_logger(cls, name: str = "algorithm", log_dir: str = "logs"):
        if not _HAS_LOGURU:
            return cls._get_stdlib_logger(name)
        if not cls._configured:
            cls._configure_loguru(name, log_dir)
        return logger

    @classmethod
    def _get_stdlib_logger(cls, name: str):
        lg = logging.getLogger(name)
        if not lg.handlers:
            lg.setLevel(logging.INFO)
            h = logging.StreamHandler(sys.stderr)
            h.setFormatter(logging.Formatter(
                "%(asctime)s [%(threadName)s] %(levelname)-5s %(name)s - %(message)s"
            ))
            lg.addHandler(h)
        return lg

    @classmethod
    def _configure_loguru(cls, name: str, log_dir: str):
        """
        配置 loguru 日志处理器
        
        Args:
            name: 日志名称
            log_dir: 日志目录
        """
        # 移除默认的 stderr handler
        logger.remove()

        # 创建日志目录
        log_path = Path(log_dir)
        log_path.mkdir(parents=True, exist_ok=True)

        # 控制台 Handler - 彩色输出
        logger.add(
            sink=sys.stderr,
            level="INFO",
            format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> [<cyan>{thread.name}</cyan>] <level>{level: <5}</level> <blue>{name}</blue> - <level>{message}</level>",
            colorize=True
        )

        # 文件 Handler - 按大小分割，Windows 兼容
        log_file = log_path / f"{name}.log"
        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)