From 2db63999107f5246882052500d74393a3829a892 Mon Sep 17 00:00:00 2001
From: Ronifue <lllstarlll@outlook.com>
Date: Sat, 29 Nov 2025 20:40:07 +0800
Subject: [PATCH] =?UTF-8?q?fix:=20=E4=BF=AE=E6=AD=A3logger?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bot.py                      | 67 ++++++++++++++++++++-----------------
 src/webui/routers/system.py |  6 ++--
 2 files changed, 41 insertions(+), 32 deletions(-)

diff --git a/bot.py b/bot.py
index 8a47fa48..34dbbda4 100644
--- a/bot.py
+++ b/bot.py
@@ -10,6 +10,33 @@ import subprocess
 from dotenv import load_dotenv
 from pathlib import Path
 from rich.traceback import install
+from src.common.logger import initialize_logging, get_logger, shutdown_logging
+
+# 设置工作目录为脚本所在目录
+script_dir = os.path.dirname(os.path.abspath(__file__))
+os.chdir(script_dir)
+
+env_path = Path(__file__).parent / ".env"
+template_env_path = Path(__file__).parent / "template" / "template.env"
+
+if env_path.exists():
+    load_dotenv(str(env_path), override=True)
+else:
+    try:
+        if template_env_path.exists():
+            shutil.copyfile(template_env_path, env_path)
+            print("未找到.env，已从 template/template.env 自动创建")
+            load_dotenv(str(env_path), override=True)
+        else:
+            print("未找到.env文件，也未找到模板 template/template.env")
+            raise FileNotFoundError(".env 文件不存在，请创建并配置所需的环境变量")
+    except Exception as e:
+        print(f"自动创建 .env 失败: {e}")
+        raise
+
+initialize_logging()
+install(extra_lines=3)
+logger = get_logger("main")
 
 # 定义重启退出码
 RESTART_EXIT_CODE = 42
@@ -27,7 +54,7 @@ def run_runner_process():
     env["MAIBOT_WORKER_PROCESS"] = "1"
 
     while True:
-        print(f"正在启动 {script_file}...")
+        logger.info(f"正在启动 {script_file}...")
         
         # 启动子进程 (Worker)
         # 使用 sys.executable 确保使用相同的 Python 解释器
@@ -40,11 +67,11 @@ def run_runner_process():
             return_code = process.wait()
             
             if return_code == RESTART_EXIT_CODE:
-                print("检测到重启请求 (退出码 42)，正在重启...")
+                logger.info("检测到重启请求 (退出码 42)，正在重启...")
                 time.sleep(1) # 稍作等待
                 continue
             else:
-                print(f"程序已退出 (退出码 {return_code})")
+                logger.info(f"程序已退出 (退出码 {return_code})")
                 sys.exit(return_code)
                 
         except KeyboardInterrupt:
@@ -56,7 +83,7 @@ def run_runner_process():
                     process.terminate()
                     process.wait(timeout=5)
                 except subprocess.TimeoutExpired:
-                    print("子进程未响应，强制关闭...")
+                    logger.warning("子进程未响应，强制关闭...")
                     process.kill()
             sys.exit(0)
 
@@ -71,42 +98,22 @@ if os.environ.get("MAIBOT_WORKER_PROCESS") != "1":
 
 # 以下是 Worker 进程的逻辑
 
-env_path = Path(__file__).parent / ".env"
-template_env_path = Path(__file__).parent / "template" / "template.env"
-
-if env_path.exists():
-    load_dotenv(str(env_path), override=True)
-    print("成功加载环境变量配置")
-else:
-    try:
-        if template_env_path.exists():
-            shutil.copyfile(template_env_path, env_path)
-            print("未找到.env，已从 template/template.env 自动创建")
-            load_dotenv(str(env_path), override=True)
-        else:
-            print("未找到.env文件，也未找到模板 template/template.env")
-            raise FileNotFoundError(".env 文件不存在，请创建并配置所需的环境变量")
-    except Exception as e:
-        print(f"自动创建 .env 失败: {e}")
-        raise
-
 # 最早期初始化日志系统，确保所有后续模块都使用正确的日志格式
-from src.common.logger import initialize_logging, get_logger, shutdown_logging  # noqa
-
-initialize_logging()
+# from src.common.logger import initialize_logging, get_logger, shutdown_logging  # noqa
+# initialize_logging()
 
 from src.main import MainSystem  # noqa
 from src.manager.async_task_manager import async_task_manager  # noqa
 
 
-logger = get_logger("main")
+# logger = get_logger("main")
 
 
-install(extra_lines=3)
+# install(extra_lines=3)
 
 # 设置工作目录为脚本所在目录
-script_dir = os.path.dirname(os.path.abspath(__file__))
-os.chdir(script_dir)
+# script_dir = os.path.dirname(os.path.abspath(__file__))
+# os.chdir(script_dir)
 logger.info(f"已设置工作目录为: {script_dir}")
 
 
diff --git a/src/webui/routers/system.py b/src/webui/routers/system.py
index dde63ced..7499c06d 100644
--- a/src/webui/routers/system.py
+++ b/src/webui/routers/system.py
@@ -11,8 +11,10 @@ from datetime import datetime
 from fastapi import APIRouter, HTTPException
 from pydantic import BaseModel
 from src.config.config import MMC_VERSION
+from src.common.logger import get_logger
 
 router = APIRouter(prefix="/system", tags=["system"])
+logger = get_logger("webui_system")
 
 # 记录启动时间
 _start_time = time.time()
@@ -46,14 +48,14 @@ async def restart_maibot():
 
     try:
         # 记录重启操作
-        print(f"[{datetime.now()}] WebUI 触发重启操作")
+        logger.info("WebUI 触发重启操作")
 
         # 定义延迟重启的异步任务
         async def delayed_restart():
             await asyncio.sleep(0.5)  # 延迟0.5秒，确保响应已发送
             # 使用 os._exit(42) 退出当前进程，配合外部 runner 脚本进行重启
             # 42 是约定的重启状态码
-            print(f"[{datetime.now()}] WebUI 请求重启，退出代码 42")
+            logger.info("WebUI 请求重启，退出代码 42")
             os._exit(42)
 
         # 创建后台任务执行重启