sansenhoshiandClaude
9371a28e35
Per docs/plugin-audit-report.md (plugins normalized to Trigger(handlers) → Service(services) → Model(repository/models) + utils): - Split monolithic __init__.py into handlers/services/utils across dailywife, deer_pipe, dice, galgame_card, helldivers_tools, huoziyinshua, learning_chat, makeaquote, mc_server_status, ncm_saying, picfinder_take, picstatus, random_jm_code, regif, steam_info, video_analysis, group_tools - Move static assets under res/: deer_pipe font/img, makeaquote font, helldivers img/templates, huoziyinshua HuoZiYinShua - Add config.py + register_config_items to ncm_saying, random_jm_code, group_tools; learning_chat unified config bridge - Remove deprecated: voice_trans plugin, bf_bot/test.py, dead code in dailywife/deer_pipe, empty dirs, debug scripts under helldivers temp - Disable brash_general_supercredits_tools (stub comment only) - bot.py: optional stdout/stderr redirect to log file for Web log viewer, force ANSI colorize on non-TTY sinks - Move runtime data (jm_code.json) out of plugin dir into hexi/data - Docs: plugin-audit-report.md; README reflects removed plugins Co-Authored-By: Claude <noreply@anthropic.com>
94 lines
2.6 KiB
Python
94 lines
2.6 KiB
Python
import asyncio
|
|
from dataclasses import dataclass
|
|
from datetime import datetime
|
|
from typing import TYPE_CHECKING
|
|
|
|
from nonebot import get_bots, logger
|
|
from nonebot.matcher import current_bot
|
|
|
|
from ...config import config
|
|
from ...misc_statistics import bot_connect_time, bot_info_cache, recv_num, send_num
|
|
from ...util import format_timedelta
|
|
from . import normal_collector
|
|
|
|
if TYPE_CHECKING:
|
|
from nonebot.adapters import Bot as BaseBot
|
|
|
|
try:
|
|
from nonebot.adapters.onebot.v11 import Bot as OBV11Bot
|
|
except ImportError:
|
|
OBV11Bot = None
|
|
|
|
|
|
@dataclass
|
|
class BotStatus:
|
|
self_id: str
|
|
adapter: str
|
|
nick: str
|
|
bot_connected: str
|
|
msg_rec: str
|
|
msg_sent: str
|
|
|
|
|
|
async def get_ob11_msg_num(bot: "BaseBot") -> tuple[int | None, int | None]:
|
|
if not (config.ps_ob_v11_use_get_status and OBV11Bot and isinstance(bot, OBV11Bot)):
|
|
return None, None
|
|
|
|
try:
|
|
bot_stat = (await bot.get_status()).get("stat")
|
|
except Exception as e:
|
|
logger.warning(
|
|
f"Error when getting bot status: {e.__class__.__name__}: {e}",
|
|
)
|
|
return None, None
|
|
if not bot_stat:
|
|
return None, None
|
|
|
|
msg_rec = bot_stat.get("message_received") or bot_stat.get(
|
|
"MessageReceived",
|
|
)
|
|
msg_sent = bot_stat.get("message_sent") or bot_stat.get("MessageSent")
|
|
return msg_rec, msg_sent
|
|
|
|
|
|
async def get_bot_status(bot: "BaseBot", now_time: datetime) -> BotStatus:
|
|
nick = (
|
|
((info := bot_info_cache[bot.self_id]).nick or info.name or info.id)
|
|
if (not config.ps_use_env_nick) and (bot.self_id in bot_info_cache)
|
|
else next(iter(config.nickname), None)
|
|
) or "Bot"
|
|
bot_connected = (
|
|
format_timedelta(now_time - t)
|
|
if (t := bot_connect_time.get(bot.self_id))
|
|
else "未知"
|
|
)
|
|
|
|
msg_rec, msg_sent = await get_ob11_msg_num(bot)
|
|
if msg_rec is None:
|
|
msg_rec = recv_num.get(bot.self_id)
|
|
if msg_sent is None:
|
|
msg_sent = send_num.get(bot.self_id)
|
|
msg_rec = "未知" if (msg_rec is None) else str(msg_rec)
|
|
msg_sent = "未知" if (msg_sent is None) else str(msg_sent)
|
|
|
|
return BotStatus(
|
|
self_id=bot.self_id,
|
|
adapter=bot.adapter.get_name(),
|
|
nick=nick,
|
|
bot_connected=bot_connected,
|
|
msg_rec=msg_rec,
|
|
msg_sent=msg_sent,
|
|
)
|
|
|
|
|
|
@normal_collector()
|
|
async def bots() -> list[BotStatus]:
|
|
now_time = datetime.now().astimezone()
|
|
return (
|
|
[await get_bot_status(current_bot.get(), now_time)]
|
|
if config.ps_show_current_bot_only
|
|
else await asyncio.gather(
|
|
*(get_bot_status(bot, now_time) for bot in get_bots().values()),
|
|
)
|
|
)
|