Merge branch 'main' of https://github.com/Fairy-Oracle-Sanctuary/NeoBot into dev

plugins/bot_status.py

@@ -6,6 +6,7 @@ Bot 状态查询插件
 import os
 import psutil
 import time
+import asyncio
 from datetime import datetime, timedelta
 
 from core.bot import Bot
@@ -32,6 +33,7 @@ def _get_system_info():
     """
     同步函数：使用 psutil 获取系统信息，避免阻塞事件循环。
     """
+    try:
         # interval=1 会阻塞1秒，必须在线程池中运行
         cpu_percent = psutil.cpu_percent(interval=1)
         mem_info = psutil.virtual_memory()
@@ -41,6 +43,13 @@ def _get_system_info():
             "mem_percent": f"{mem_info.percent:.1f}",
             "bot_mem_mb": f"{bot_mem_mb:.2f}",
         }
+    except Exception as e:
+        logger.error(f"获取系统信息失败: {e}")
+        return {
+            "cpu_percent": "N/A",
+            "mem_percent": "N/A",
+            "bot_mem_mb": "N/A",
+        }
 
 @matcher.command("status", "状态")
 async def handle_status(bot: Bot, event: MessageEvent, args: list[str]):
@@ -93,6 +102,7 @@ async def handle_status(bot: Bot, event: MessageEvent, args: list[str]):
         }
 
         # 3. 获取统计数据
+        try:
             msgs_recv = await redis_manager.get("neobot:stats:messages_received") or 0
             msgs_sent = await redis_manager.get("neobot:stats:messages_sent") or 0
             command_stats_raw = await redis_manager.redis.hgetall("neobot:command_stats")
@@ -110,9 +120,33 @@ async def handle_status(bot: Bot, event: MessageEvent, args: list[str]):
                 key=lambda x: x["count"],
                 reverse=True
             )
+        except Exception as e:
+            logger.error(f"获取Redis统计数据失败: {e}")
+            stats_data = {
+                "messages_received": 0,
+                "messages_sent": 0,
+                "total_commands": 0,
+            }
+            command_stats_data = []
 
         # 4. 异步获取系统信息
-        system_data = await run_in_thread_pool(_get_system_info)
+        # 设置超时，防止 psutil 阻塞过久
+        try:
+            system_data = await asyncio.wait_for(run_in_thread_pool(_get_system_info), timeout=5.0)
+        except asyncio.TimeoutError:
+            logger.error("获取系统信息超时")
+            system_data = {
+                "cpu_percent": "Timeout",
+                "mem_percent": "Timeout",
+                "bot_mem_mb": "Timeout",
+            }
+        except Exception as e:
+            logger.error(f"获取系统信息异常: {e}")
+            system_data = {
+                "cpu_percent": "Error",
+                "mem_percent": "Error",
+                "bot_mem_mb": "Error",
+            }
 
         # 5. 准备模板所需的所有数据
         template_data = {
@@ -125,6 +159,7 @@ async def handle_status(bot: Bot, event: MessageEvent, args: list[str]):
         }
 
         # 6. 渲染图片
+        try:
             base64_str = await image_manager.render_template_to_base64(
                 template_name="status.html",
                 data=template_data,
@@ -137,6 +172,9 @@ async def handle_status(bot: Bot, event: MessageEvent, args: list[str]):
             else:
                 # 如果渲染失败，image_manager 内部会记录错误，这里给用户一个通用提示
                 await event.reply("状态图片生成失败，可能是渲染服务出现问题，请联系管理员。")
+        except Exception as e:
+            logger.error(f"渲染图片失败: {e}")
+            await event.reply("状态图片渲染过程中发生错误。")
 
     except Exception as e:
         logger.exception(f"生成状态图时发生意外错误, 用户: {event.user_id}")