From 668f82bab90cb6772aad84349bbbb83221ee2f1b Mon Sep 17 00:00:00 2001
From: ljh938527 <2689819155@qq.com>
Date: Sat, 13 Dec 2025 23:12:47 +0800
Subject: [PATCH] =?UTF-8?q?=E6=B7=BB=E5=8A=A0=E6=97=A5=E5=BF=97=E6=A0=BC?=
 =?UTF-8?q?=E5=BC=8F=E9=A2=9C=E8=89=B2?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 models/logger.py | 15 +++++++++------
 1 file changed, 9 insertions(+), 6 deletions(-)

diff --git a/models/logger.py b/models/logger.py
index 832c393..46df5e4 100644
--- a/models/logger.py
+++ b/models/logger.py
@@ -2,6 +2,7 @@ import logging
 from logging.handlers import RotatingFileHandler
 from datetime import datetime
 from pathlib import Path
+import colorlog
 
 def setup_logger(log_file=None, log_level=logging.INFO, max_bytes=10485760, backup_count=5):
     """
@@ -34,10 +35,12 @@ def setup_logger(log_file=None, log_level=logging.INFO, max_bytes=10485760, back
         today = datetime.now().strftime('%Y-%m-%d')
         log_file = log_dir / f'{today}.log'
 
-    # 创建格式化器
-    formatter = logging.Formatter(
-        '[%(asctime)s] [%(levelname)s] [%(filename)s:%(lineno)d] - %(message)s'
-    )
+    # 创建格式
+    log_format = '[%(asctime)s] [%(levelname)s] [%(filename)s:%(lineno)d] - %(message)s'
+    color_format = '%(log_color)s' + log_format
+
+    # 有颜色的日志格式化器
+    color_formatter = colorlog.ColoredFormatter(color_format, reset=True)
 
     # 创建文件处理器（带轮转）
     file_handler = RotatingFileHandler(
@@ -47,12 +50,12 @@ def setup_logger(log_file=None, log_level=logging.INFO, max_bytes=10485760, back
         encoding='utf-8'
     )
     file_handler.setLevel(log_level)
-    file_handler.setFormatter(formatter)
+    file_handler.setFormatter(color_formatter)
 
     # 创建控制台处理器
     console_handler = logging.StreamHandler()
     console_handler.setLevel(log_level)
-    console_handler.setFormatter(formatter)
+    console_handler.setFormatter(color_formatter)
 
     # 添加处理器到logger
     logger.addHandler(file_handler)