refactor: use config and logger in service

src/utils/logger.py

@@ -0,0 +1,91 @@
+import logging
+import os
+from datetime import datetime
+from .config import get_instance as get_config
+
+# 定义颜色代码
+class Colors:
+    RED = '\033[31m'
+    GREEN = '\033[32m'
+    YELLOW = '\033[33m'
+    BLUE = '\033[34m'
+    MAGENTA = '\033[35m'
+    CYAN = '\033[36m'
+    WHITE = '\033[37m'
+    RESET = '\033[0m'  # 重置颜色
+
+# 自定义控制台处理器，为不同日志级别添加颜色
+class ColoredConsoleHandler(logging.StreamHandler):
+    def emit(self, record):
+        # 为不同日志级别设置颜色
+        colors = {
+            logging.DEBUG: Colors.CYAN,
+            logging.INFO: Colors.GREEN,
+            logging.WARNING: Colors.YELLOW,
+            logging.ERROR: Colors.RED,
+            logging.CRITICAL: Colors.MAGENTA
+        }
+        
+        # 获取对应级别的颜色，默认为白色
+        color = colors.get(record.levelno, Colors.WHITE)
+        
+        # 获取原始消息
+        message = self.format(record)
+        
+        # 添加颜色并输出
+        self.stream.write(f"{color}{message}{Colors.RESET}\n")
+        self.flush()
+
+def init():
+    config = get_config()
+    log_dir = config.get("log", "log_dir", fallback="logs")
+    log_file = config.get("log", "log_file", fallback="if.u.service")
+    log_level = config.get("log", "log_level", fallback=logging.INFO)
+    console_log_level = config.get("log", "console_log_level", fallback=logging.DEBUG)
+
+    # 创建logs目录（如果不存在）
+    if not os.path.exists(log_dir):
+        os.makedirs(log_dir)
+    
+    # 设置日志格式
+    log_format = "[%(asctime)s.%(msecs)03d][%(filename)s:%(lineno)d][%(levelname)s] %(message)s"
+    date_format = "%Y-%m-%d %H:%M:%S"
+    
+    # 创建格式化器
+    formatter = logging.Formatter(log_format, datefmt=date_format)
+    
+    # 获取根日志记录器
+    root_logger = logging.getLogger()
+    root_logger.setLevel(logging.NOTSET)
+    
+    # 清除现有的处理器
+    root_logger.handlers.clear()
+    
+    # 创建控制台处理器并设置颜色
+    console_handler = ColoredConsoleHandler()
+    console_handler.setFormatter(formatter)
+    console_handler.setLevel(console_log_level)
+    root_logger.addHandler(console_handler)
+    
+    # 创建文件处理器
+    log_filename = os.path.join(log_dir, f"{log_file}_{datetime.now().strftime('%Y%m%d')}.log")
+    file_handler = logging.FileHandler(log_filename, encoding='utf-8')
+    file_handler.setFormatter(formatter)
+    file_handler.setLevel(log_level)
+    root_logger.addHandler(file_handler)
+    
+    # 确保日志消息被正确处理
+    logging.addLevelName(logging.DEBUG, "D")
+    logging.addLevelName(logging.INFO, "I")
+    logging.addLevelName(logging.WARNING, "W")
+    logging.addLevelName(logging.ERROR, "E")
+    logging.addLevelName(logging.CRITICAL, "C")
+
+
+if __name__ == "__main__":
+    init(log_dir="logs", log_file="test", log_level=logging.INFO, console_log_level=logging.DEBUG)
+    logging.debug("debug log")
+    logging.info("info log")
+    logging.warning("warning log")
+    logging.error("error log")
+    logging.critical("critical log")