From 7ddbc3ba49d9a8aca17e2c746dc750e878aa9f25 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?=E5=A2=A8=E6=A2=93=E6=9F=92?= <1787882683@qq.com>
Date: Fri, 2 May 2025 01:49:59 +0800
Subject: [PATCH] =?UTF-8?q?=E9=87=8D=E6=9E=84=E6=97=A5=E5=BF=97=E8=AE=B0?=
 =?UTF-8?q?=E5=BD=95=EF=BC=8C=E7=BB=9F=E4=B8=80=E4=BD=BF=E7=94=A8get=5Fmod?=
 =?UTF-8?q?ule=5Flogger=E5=88=9D=E5=A7=8B=E5=8C=96=E6=97=A5=E5=BF=97?=
 =?UTF-8?q?=E8=AE=B0=E5=BD=95=E5=99=A8?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 .gitignore           |   3 +-
 main.py              |   5 +-
 src/config.py        |   3 +-
 src/logger.py        | 149 +++++++++++++++++++++++++++++++++++++------
 src/message_queue.py |   4 +-
 src/mmc_com_layer.py |   4 +-
 src/recv_handler.py  |   4 +-
 src/send_handler.py  |   3 +-
 src/utils.py         |   3 +-
 9 files changed, 152 insertions(+), 26 deletions(-)

diff --git a/.gitignore b/.gitignore
index a8b468a..ee46f8e 100644
--- a/.gitignore
+++ b/.gitignore
@@ -2,4 +2,5 @@
 .vscode
 test
 .ruff_cache
-config.toml
\ No newline at end of file
+config.toml
+logs/
\ No newline at end of file
diff --git a/main.py b/main.py
index c3d96a9..9beca8a 100644
--- a/main.py
+++ b/main.py
@@ -2,7 +2,7 @@ import asyncio
 import sys
 import json
 import websockets as Server
-from src.logger import logger
+from src.logger import get_module_logger
 from src.recv_handler import recv_handler
 from src.send_handler import send_handler
 from src.config import global_config
@@ -10,6 +10,9 @@ from src.mmc_com_layer import mmc_start_com, mmc_stop_com, router
 from src.message_queue import message_queue, put_response, check_timeout_response
 
 
+logger = get_module_logger("适配器主程序")
+
+
 async def message_recv(server_connection: Server.ServerConnection):
     recv_handler.server_connection = server_connection
     send_handler.server_connection = server_connection
diff --git a/src/config.py b/src/config.py
index 920b6d6..f2abe29 100644
--- a/src/config.py
+++ b/src/config.py
@@ -2,10 +2,11 @@ import os
 import sys
 import tomli
 import shutil
-from .logger import logger
+from loguru import logger
 from typing import Optional
 
 
+
 class Config:
     platform: str = "qq"
     nickname: Optional[str] = None
diff --git a/src/logger.py b/src/logger.py
index c097f14..4fa2560 100644
--- a/src/logger.py
+++ b/src/logger.py
@@ -1,25 +1,138 @@
 from loguru import logger
-from .config import global_config
+from typing import Optional, Union, List, Tuple
 import sys
-# import builtins
+import os
+from types import ModuleType
+from pathlib import Path
+from .config import global_config
 
-logger.remove()
-logger.add(
-    sys.stderr,
-    level=global_config.debug_level,
-    format="<green>{time:YYYY-MM-DD HH:mm:ss}</green> | <level>{level: <8}</level> | <cyan>{name}</cyan>:<cyan>{function}</cyan>:<cyan>{line}</cyan> - <level>{message}</level>",
-)
+# 保存原生处理器ID
+default_handler_id = None
+for handler_id in logger._core.handlers:
+    default_handler_id = handler_id
+    break
+
+# 移除默认处理器
+if default_handler_id is not None:
+    logger.remove(default_handler_id)
+
+# 类型别名
+LoguruLogger = logger.__class__
+
+# 全局注册表：记录模块与处理器ID的映射
+_handler_registry: dict[str, List[int]] = {}
+_custom_style_handlers: dict[Tuple[str, str], List[int]] = {}  # 记录自定义样式处理器ID
+
+# 获取日志存储根地址
+current_file_path = Path(__file__).resolve()
+LOG_ROOT = "logs"
+
+# LOG_LEVEL = global_config.get("Debug", {}).get("level", "INFO").upper()
+# print(global_config.debug_level)
+
+DEFAULT_CONFIG = {
+        # 日志级别配置
+        "console_level": global_config.debug_level,
+        "file_level": "DEBUG",
+        # 格式配置
+        "console_format": (
+            "<white>{time:YYYY-MM-DD HH:mm:ss}</white> | <level>{level: <8}</level> | <cyan>{extra[module]: <15}</cyan> | <level>{message}</level>"
+        ),
+        "file_format": "{time:YYYY-MM-DD HH:mm:ss} | {level: <8} | {extra[module]: <15} | {message}",
+        "log_dir": LOG_ROOT,
+        "rotation": "00:00",
+        "retention": "3 days",
+        "compression": "zip",
+    }
+
+def is_registered_module(record: dict) -> bool:
+    """检查是否为已注册的模块"""
+    return record["extra"].get("module") in _handler_registry
 
 
-# def handle_output(message: str):
-#     if "连接失败" in message:
-#         logger.error(message)
-#     elif "收到无效的" in message:
-#         logger.warning(message)
-#     elif "检测到平台" in message:
-#         logger.warning(message)
-#     else:
-#         logger.info(message)
+def is_unregistered_module(record: dict) -> bool:
+    """检查是否为未注册的模块"""
+    return not is_registered_module(record)
 
 
-# builtins.print = handle_output
+def log_patcher(record: dict) -> None:
+    """自动填充未设置模块名的日志记录，保留原生模块名称"""
+    if "module" not in record["extra"]:
+        # 尝试从name中提取模块名
+        module_name = record.get("name", "")
+        if module_name == "":
+            module_name = "root"
+        record["extra"]["module"] = module_name
+class LogConfig:
+    """日志配置类"""
+
+    def __init__(self, **kwargs):
+        self.config = DEFAULT_CONFIG.copy()
+        self.config.update(kwargs)
+
+    def to_dict(self) -> dict:
+        return self.config.copy()
+
+    def update(self, **kwargs):
+        self.config.update(kwargs)
+
+
+def get_module_logger(
+    module: Union[str, ModuleType],
+    *,
+    console_level: Optional[str] = None,
+    file_level: Optional[str] = None,
+    extra_handlers: Optional[List[dict]] = None,
+    config: Optional[LogConfig] = None,
+) -> LoguruLogger:
+    module_name = module if isinstance(module, str) else module.__name__
+    current_config = config.config if config else DEFAULT_CONFIG
+
+    # 清理旧处理器
+    if module_name in _handler_registry:
+        for handler_id in _handler_registry[module_name]:
+            logger.remove(handler_id)
+        del _handler_registry[module_name]
+
+    handler_ids = []
+
+    # 控制台处理器
+    console_id = logger.add(
+        sink=sys.stderr,
+        level=os.getenv("CONSOLE_LOG_LEVEL", console_level or current_config["console_level"]),
+        format=current_config["console_format"],
+        filter=lambda record: record["extra"].get("module") == module_name and "custom_style" not in record["extra"],
+        enqueue=True,
+    )
+    handler_ids.append(console_id)
+
+    # 文件处理器
+    log_dir = Path("logs")
+    log_dir.mkdir(parents=True, exist_ok=True)
+    log_file = log_dir / "{time:YYYY-MM-DD}.log"
+    log_file.parent.mkdir(parents=True, exist_ok=True)
+
+    file_id = logger.add(
+        sink=str(log_file),
+        level=os.getenv("FILE_LOG_LEVEL", file_level or current_config["file_level"]),
+        format=current_config["file_format"],
+        rotation=current_config["rotation"],
+        retention=current_config["retention"],
+        compression=current_config["compression"],
+        encoding="utf-8",
+        filter=lambda record: record["extra"].get("module") == module_name and "custom_style" not in record["extra"],
+        enqueue=True,
+    )
+    handler_ids.append(file_id)
+
+    # 额外处理器
+    if extra_handlers:
+        for handler in extra_handlers:
+            handler_id = logger.add(**handler)
+            handler_ids.append(handler_id)
+
+    # 更新注册表
+    _handler_registry[module_name] = handler_ids
+
+    return logger.bind(module=module_name)
+
diff --git a/src/message_queue.py b/src/message_queue.py
index 95d646c..ba01518 100644
--- a/src/message_queue.py
+++ b/src/message_queue.py
@@ -2,7 +2,9 @@ import asyncio
 import time
 from typing import Dict
 from .config import global_config
-from .logger import logger
+from .logger import get_module_logger
+
+logger = get_module_logger("消息队列")
 
 response_dict: Dict = {}
 response_time_dict: Dict = {}
diff --git a/src/mmc_com_layer.py b/src/mmc_com_layer.py
index 76a549f..e46514b 100644
--- a/src/mmc_com_layer.py
+++ b/src/mmc_com_layer.py
@@ -1,8 +1,10 @@
 from maim_message import Router, RouteConfig, TargetConfig
 from .config import global_config
-from .logger import logger
+from .logger import get_module_logger
 from .send_handler import send_handler
 
+logger = get_module_logger("麦麦核心链接")
+
 route_config = RouteConfig(
     route_config={
         global_config.platform: TargetConfig(
diff --git a/src/recv_handler.py b/src/recv_handler.py
index 33f98b9..e638fbb 100644
--- a/src/recv_handler.py
+++ b/src/recv_handler.py
@@ -1,4 +1,4 @@
-from .logger import logger
+from .logger import get_module_logger
 from .config import global_config
 from .qq_emoji_list import qq_face
 import time
@@ -31,6 +31,8 @@ from .utils import (
 from .message_queue import get_response
 
 
+logger = get_module_logger("收到消息处理")
+
 class RecvHandler:
     maibot_router: Router = None
 
diff --git a/src/send_handler.py b/src/send_handler.py
index 247b512..ab7b4c1 100644
--- a/src/send_handler.py
+++ b/src/send_handler.py
@@ -5,7 +5,7 @@ import uuid
 # from .config import global_config
 # 白名单机制不启用
 from .message_queue import get_response
-from .logger import logger
+from .logger import get_module_logger
 
 from maim_message import (
     UserInfo,
@@ -17,6 +17,7 @@ from maim_message import (
 
 from .utils import get_image_format, convert_image_to_gif
 
+logger = get_module_logger("发送处理器")
 
 class SendHandler:
     def __init__(self):
diff --git a/src/utils.py b/src/utils.py
index c699d8d..c7e3adc 100644
--- a/src/utils.py
+++ b/src/utils.py
@@ -2,7 +2,7 @@ import websockets as Server
 import json
 import base64
 import uuid
-from .logger import logger
+from .logger import get_module_logger
 from .message_queue import get_response
 
 import urllib3
@@ -11,6 +11,7 @@ import ssl
 from PIL import Image
 import io
 
+logger = get_module_logger("utils")
 
 class SSLAdapter(urllib3.PoolManager):
     def __init__(self, *args, **kwargs):