重构DBN模型

app/utils/logger.py

@@ -1,81 +1,78 @@
 """
 日志工具类
-支持按天分割、自动清理过期日志
+使用 loguru 提供增强的日志功能，支持按天分割、自动清理过期日志
 """
-import logging
+import sys
 from pathlib import Path
-from logging.handlers import TimedRotatingFileHandler
-from datetime import datetime, timedelta
+from loguru import logger
 
 
 class LoggerManager:
-    """日志管理器"""
+    """日志管理器 - 基于 loguru"""
 
-    _loggers = {}
+    _configured = False
 
     @classmethod
-    def get_logger(cls, name: str = "algorithm", log_dir: str = "logs") -> logging.Logger:
+    def get_logger(cls, name: str = "algorithm", log_dir: str = "logs"):
         """
-        获取日志记录器
+        获取日志记录器（loguru 不需要传统意义上的 logger 实例）
+        
+        Args:
+            name: 日志名称（用于文件命名）
+            log_dir: 日志目录
+            
+        Returns:
+            loguru.logger 实例
+        """
+        if not cls._configured:
+            cls._configure_logger(name, log_dir)
+        return logger
+
+    @classmethod
+    def _configure_logger(cls, name: str, log_dir: str):
+        """
+        配置 loguru 日志处理器
         
         Args:
             name: 日志名称
             log_dir: 日志目录
-            
-        Returns:
-            logging.Logger 实例
         """
-        if name in cls._loggers:
-            return cls._loggers[name]
+        # 移除默认的 stderr handler
+        logger.remove()
 
         # 创建日志目录
         log_path = Path(log_dir)
         log_path.mkdir(parents=True, exist_ok=True)
 
-        # 创建 logger
-        logger = logging.getLogger(name)
-        logger.setLevel(logging.DEBUG)
-
-        # 避免重复添加 handler
-        if logger.handlers:
-            cls._loggers[name] = logger
-            return logger
-
-        # 日志格式
-        formatter = logging.Formatter(
-            '%(asctime)s [%(threadName)s] %(levelname)-5s %(name)s - %(message)s',
-            datefmt='%Y-%m-%d %H:%M:%S'
+        # 控制台 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
-        console_handler = logging.StreamHandler()
-        console_handler.setLevel(logging.INFO)
-        console_handler.setFormatter(formatter)
-        logger.addHandler(console_handler)
-
-        # 文件 Handler - 按天分割
+        # 文件 Handler - 按大小分割，Windows 兼容
         log_file = log_path / f"{name}.log"
-        file_handler = TimedRotatingFileHandler(
-            filename=str(log_file),
-            when='midnight',
-            interval=1,
-            backupCount=7,
-            encoding='utf-8'
+        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  # 详细错误诊断
         )
-        file_handler.setLevel(logging.DEBUG)
-        file_handler.setFormatter(formatter)
 
-        # 设置日志文件命名格式
-        file_handler.suffix = "%Y-%m-%d.log"
-
-        logger.addHandler(file_handler)
-
-        cls._loggers[name] = logger
-        return logger
+        cls._configured = True
 
 
 # 便捷函数
-def get_logger(name: str = "algorithm", log_dir: str = "logs") -> logging.Logger:
+def get_logger(name: str = "algorithm", log_dir: str = "logs"):
     """
     获取日志记录器的便捷函数
     
@@ -84,6 +81,6 @@ def get_logger(name: str = "algorithm", log_dir: str = "logs") -> logging.Logger
         log_dir: 日志目录
         
     Returns:
-        logging.Logger 实例
+        loguru.logger 实例
     """
     return LoggerManager.get_logger(name, log_dir)