补充提交

utils/logger.py

@@ -0,0 +1,88 @@
+"""
+日志配置模块
+"""
+
+import logging
+import sys
+from typing import Optional
+from logging.handlers import RotatingFileHandler
+
+from config import settings
+
+
+def setup_logger(
+        name: str,
+        level: Optional[str] = None,
+        log_file: Optional[str] = None
+) -> logging.Logger:
+    """
+    配置和获取logger
+
+    Args:
+        name: logger名称
+        level: 日志级别
+        log_file: 日志文件路径
+
+    Returns:
+        配置好的logger实例
+    """
+    # 获取日志级别
+    if level is None:
+        level = settings.LOG_LEVEL
+
+    log_level = getattr(logging, level.upper(), logging.INFO)
+
+    # 创建logger
+    logger = logging.getLogger(name)
+    logger.setLevel(log_level)
+
+    # 避免重复添加handler
+    if logger.handlers:
+        return logger
+
+    # 创建formatter
+    formatter = logging.Formatter(
+        '%(asctime)s - %(name)s - %(levelname)s - %(message)s',
+        datefmt='%Y-%m-%d %H:%M:%S'
+    )
+
+    # 控制台handler
+    console_handler = logging.StreamHandler(sys.stdout)
+    console_handler.setLevel(log_level)
+    console_handler.setFormatter(formatter)
+    logger.addHandler(console_handler)
+
+    # 文件handler（如果配置了日志文件）
+    if log_file or settings.LOG_FILE:
+        file_path = log_file or settings.LOG_FILE
+        try:
+            file_handler = RotatingFileHandler(
+                file_path,
+                maxBytes=10 * 1024 * 1024,  # 10MB
+                backupCount=5,
+                encoding='utf-8'
+            )
+            file_handler.setLevel(log_level)
+            file_handler.setFormatter(formatter)
+            logger.addHandler(file_handler)
+        except Exception as e:
+            logger.warning(f"Failed to create log file handler: {e}")
+
+    return logger
+
+
+# 创建根logger
+root_logger = setup_logger("sur_face_feature")
+
+
+def get_logger(name: str) -> logging.Logger:
+    """
+    获取指定名称的logger
+
+    Args:
+        name: logger名称
+
+    Returns:
+        logger实例
+    """
+    return setup_logger(name)