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>
82 lines
2.6 KiB
Python
82 lines
2.6 KiB
Python
import asyncio
|
|
|
|
from cookit.loguru import warning_suppress
|
|
from cookit.nonebot.alconna import extract_reply_msg
|
|
from nonebot import logger, on_command
|
|
from nonebot.adapters import Bot as BaseBot, Event as BaseEvent, Message as BaseMessage
|
|
from nonebot.matcher import current_bot, current_event, current_matcher
|
|
from nonebot.params import CommandArg
|
|
from nonebot.permission import SUPERUSER
|
|
from nonebot.rule import Rule, to_me
|
|
from nonebot.typing import T_State
|
|
from nonebot_plugin_alconna.uniseg import Image, OriginalUniMsg, UniMessage, image_fetch
|
|
|
|
from .bg_provider import BgBytesData, bg_preloader
|
|
from .services.collectors import collect_all
|
|
from .config import config
|
|
from .misc_statistics import bot_avatar_cache, bot_info_cache, cache_bot_avatar
|
|
from .templates import render_current_template
|
|
|
|
|
|
def check_empty_arg_rule(arg: BaseMessage = CommandArg()):
|
|
return not arg.extract_plain_text()
|
|
|
|
|
|
def trigger_rule():
|
|
rule = Rule(check_empty_arg_rule)
|
|
if config.ps_need_at:
|
|
rule &= to_me()
|
|
return rule
|
|
|
|
|
|
_cmd, *_alias = config.ps_command
|
|
stat_matcher = on_command(
|
|
_cmd,
|
|
aliases=set(_alias),
|
|
rule=trigger_rule(),
|
|
permission=SUPERUSER if config.ps_only_su else None,
|
|
)
|
|
|
|
|
|
async def get_pic_from_msg(msg: UniMessage) -> BgBytesData | None:
|
|
msg = r if ((r := extract_reply_msg(msg)) and Image in r) else msg
|
|
if Image not in msg:
|
|
return None
|
|
img = msg[Image, 0]
|
|
data = await image_fetch(
|
|
current_event.get(),
|
|
current_bot.get(),
|
|
current_matcher.get().state,
|
|
img,
|
|
)
|
|
if not data:
|
|
return None
|
|
return BgBytesData(data=data, mime=img.mimetype or "image")
|
|
|
|
|
|
@stat_matcher.handle()
|
|
async def _(bot: BaseBot, event: BaseEvent, state: T_State, msg: OriginalUniMsg):
|
|
if (
|
|
(bot.self_id not in bot_avatar_cache)
|
|
and (info := bot_info_cache.get(bot.self_id))
|
|
and info.avatar
|
|
):
|
|
await cache_bot_avatar(info.avatar, bot, event, state)
|
|
|
|
async def get_bg():
|
|
with warning_suppress("Failed to fetch image from user message"):
|
|
if bg := await get_pic_from_msg(msg):
|
|
return bg
|
|
return await bg_preloader.get()
|
|
|
|
try:
|
|
bg, collected = await asyncio.gather(get_bg(), collect_all())
|
|
ret = await render_current_template(collected=collected, bg=bg)
|
|
except Exception:
|
|
logger.exception("获取运行状态图失败")
|
|
await UniMessage("获取运行状态图片失败，请检查后台输出").send(
|
|
reply_to=config.ps_reply_target,
|
|
)
|
|
else:
|
|
await UniMessage.image(raw=ret).send(reply_to=config.ps_reply_target)
|