refactor: logger in src\plugins\chat\__init__.py

src/plugins/chat/__init__.py

@@ -2,9 +2,8 @@ import asyncio
 import time
 
 from loguru import logger
-from nonebot import get_driver, on_command, on_message, require
+from nonebot import get_driver, on_message, require
 from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message, MessageSegment
-from nonebot.rule import to_me
 from nonebot.typing import T_State
 
 from ...common.database import Database
@@ -16,6 +15,10 @@ from .config import global_config
 from .emoji_manager import emoji_manager
 from .relationship_manager import relationship_manager
 from .willing_manager import willing_manager
+from ..memory_system.memory import hippocampus, memory_graph
+from .bot import ChatBot
+from .message_sender import message_manager, message_sender
+
 
 # 创建LLM统计实例
 llm_stats = LLMStatistics("llm_statistics.txt")
@@ -35,19 +38,13 @@ Database.initialize(
     password=config.MONGODB_PASSWORD,
     auth_source=config.MONGODB_AUTH_SOURCE
 )
-print("\033[1;32m[初始化数据库完成]\033[0m")
+logger.success("初始化数据库成功")
 
-# 导入其他模块
-from ..memory_system.memory import hippocampus, memory_graph
-from .bot import ChatBot
-
-# from .message_send_control import message_sender
-from .message_sender import message_manager, message_sender
 
 # 初始化表情管理器
 emoji_manager.initialize()
 
-print(f"\033[1;32m正在唤醒{global_config.BOT_NICKNAME}......\033[0m")
+logger.debug(f"正在唤醒{global_config.BOT_NICKNAME}......")
 # 创建机器人实例
 chat_bot = ChatBot()
 # 注册群消息处理器
@@ -61,7 +58,7 @@ async def start_background_tasks():
     """启动后台任务"""
     # 启动LLM统计
     llm_stats.start()
-    logger.success("[初始化]LLM统计功能已启动")
+    logger.success("LLM统计功能启动成功")
 
     # 初始化并启动情绪管理器
     mood_manager = MoodManager.get_instance()
@@ -77,7 +74,7 @@ async def start_background_tasks():
 @driver.on_startup
 async def init_relationships():
     """在 NoneBot2 启动时初始化关系管理器"""
-    print("\033[1;32m[初始化]\033[0m 正在加载用户关系数据...")
+    logger.debug("正在加载用户关系数据...")
     await relationship_manager.load_all_relationships()
     asyncio.create_task(relationship_manager._start_relationship_manager())
 
@@ -86,19 +83,19 @@ async def init_relationships():
 async def _(bot: Bot):
     """Bot连接成功时的处理"""
     global _message_manager_started
-    print(f"\033[1;38;5;208m-----------{global_config.BOT_NICKNAME}成功连接！-----------\033[0m")
+    logger.debug(f"-----------{global_config.BOT_NICKNAME}成功连接！-----------")
     await willing_manager.ensure_started()
 
     message_sender.set_bot(bot)
-    print("\033[1;38;5;208m-----------消息发送器已启动！-----------\033[0m")
+    logger.success("-----------消息发送器已启动！-----------")
 
     if not _message_manager_started:
         asyncio.create_task(message_manager.start_processor())
         _message_manager_started = True
-        print("\033[1;38;5;208m-----------消息处理器已启动！-----------\033[0m")
+        logger.success("-----------消息处理器已启动！-----------")
 
     asyncio.create_task(emoji_manager._periodic_scan(interval_MINS=global_config.EMOJI_REGISTER_INTERVAL))
-    print("\033[1;38;5;208m-----------开始偷表情包！-----------\033[0m")
+    logger.success("-----------开始偷表情包！-----------")
 
 
 @group_msg.handle()
@@ -110,13 +107,15 @@ async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
 @scheduler.scheduled_job("interval", seconds=global_config.build_memory_interval, id="build_memory")
 async def build_memory_task():
     """每build_memory_interval秒执行一次记忆构建"""
-    print(
-        "\033[1;32m[记忆构建]\033[0m -------------------------------------------开始构建记忆-------------------------------------------")
+    logger.debug(
+        "[记忆构建]"
+        "------------------------------------开始构建记忆--------------------------------------")
     start_time = time.time()
     await hippocampus.operation_build_memory(chat_size=20)
     end_time = time.time()
-    print(
-        f"\033[1;32m[记忆构建]\033[0m -------------------------------------------记忆构建完成：耗时: {end_time - start_time:.2f} 秒-------------------------------------------")
+    logger.success(
+        f"[记忆构建]--------------------------记忆构建完成：耗时: {end_time - start_time:.2f} "
+        "秒-------------------------------------------")
 
 
 @scheduler.scheduled_job("interval", seconds=global_config.forget_memory_interval, id="forget_memory")