From 9b72b5a996e486b23cc8f2c042fdf1d255fc569d Mon Sep 17 00:00:00 2001
From: AL76 <735756072@qq.com>
Date: Sat, 15 Mar 2025 04:19:20 +0800
Subject: [PATCH] =?UTF-8?q?refactor:=20=E4=B8=BB=E7=A8=8B=E5=BA=8Fbot.py?=
 =?UTF-8?q?=E6=97=A5=E5=BF=97=E4=BC=98=E5=8C=96?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 bot.py | 16 +++++-----------
 1 file changed, 5 insertions(+), 11 deletions(-)

diff --git a/bot.py b/bot.py
index 7a97f485e..50d04fd08 100644
--- a/bot.py
+++ b/bot.py
@@ -10,10 +10,11 @@ import uvicorn
 from dotenv import load_dotenv
 from nonebot.adapters.onebot.v11 import Adapter
 import platform
-from src.plugins.utils.logger_config import LogModule, LogClassification
+from src.common.logger import get_module_logger
 
 
-# 配置日志格式
+# 配置主程序日志格式
+logger = get_module_logger("主程序")
 
 # 获取没有加载env时的环境变量
 env_mask = {key: os.getenv(key) for key in os.environ}
@@ -76,11 +77,11 @@ def init_env():
 def load_env():
     # 使用闭包实现对加载器的横向扩展，避免大量重复判断
     def prod():
-        logger.success("加载生产环境变量配置")
+        logger.success("成功加载生产环境变量配置")
         load_dotenv(".env.prod", override=True)  # override=True 允许覆盖已存在的环境变量
 
     def dev():
-        logger.success("加载开发环境变量配置")
+        logger.success("成功加载开发环境变量配置")
         load_dotenv(".env.dev", override=True)  # override=True 允许覆盖已存在的环境变量
 
     fn_map = {"prod": prod, "dev": dev}
@@ -100,11 +101,6 @@ def load_env():
         RuntimeError(f"ENVIRONMENT 配置错误，请检查 .env 文件中的 ENVIRONMENT 变量及对应 .env.{env} 是否存在")
 
 
-def load_logger():
-    global logger # 使得bot.py中其他函数也能调用
-    log_module = LogModule()
-    logger = log_module.setup_logger(LogClassification.BASE)
-
 
 def scan_provider(env_config: dict):
     provider = {}
@@ -206,8 +202,6 @@ def raw_main():
 
 if __name__ == "__main__":
     try:
-        # 配置日志，使得主程序直接退出时候也能访问logger
-        load_logger()
         raw_main()
 
         app = nonebot.get_asgi()