generated from A-kirami/nonebot-plugin-template
-
-
Notifications
You must be signed in to change notification settings - Fork 10
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
Showing
11 changed files
with
309 additions
and
305 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -1,110 +1,108 @@ | ||
| from typing import Optional | ||
|
|
||
| from nonebot import logger, on_command | ||
| from nonebot.adapters import Bot, Event, Message | ||
| from nonebot.matcher import Matcher | ||
| from nonebot.params import CommandArg | ||
| from nonebot.permission import SUPERUSER | ||
| from nonebot.rule import Rule, to_me | ||
| from nonebot_plugin_saa import Image, MessageFactory, extract_target | ||
|
|
||
| from .config import config | ||
| from .draw import get_stat_pic | ||
| from .util import async_request, download_telegram_file | ||
|
|
||
| try: | ||
| from nonebot.adapters.onebot.v11 import MessageEvent as OBV11MessageEvent | ||
| except ImportError: | ||
| OBV11MessageEvent = None | ||
|
|
||
| try: | ||
| from nonebot.adapters.telegram.event import MessageEvent as TGMessageEvent | ||
| except ImportError: | ||
| TGMessageEvent = None | ||
|
|
||
|
|
||
| def check_empty_arg_rule(arg: Message = CommandArg()): | ||
| return not arg.extract_plain_text() | ||
|
|
||
|
|
||
| def trigger_rule(): | ||
| rule = Rule(check_empty_arg_rule) | ||
| if config.ps_need_at: | ||
| rule = rule & to_me() | ||
| return rule | ||
|
|
||
|
|
||
| async def extract_msg_pic(bot: Bot, event: Event) -> Optional[bytes]: | ||
| if OBV11MessageEvent and isinstance(event, OBV11MessageEvent): | ||
| if (event.reply and (img := event.reply.message["image"])) or ( | ||
| img := event.message["image"] | ||
| ): | ||
| url = img[0].data["url"] | ||
| return await async_request(url) | ||
|
|
||
| elif TGMessageEvent and isinstance(event, TGMessageEvent): | ||
| msg = event.message | ||
| if event.reply_to_message and event.reply_to_message.message: | ||
| msg += event.reply_to_message.message | ||
|
|
||
| file_id = None | ||
| if photos := msg["photo"]: | ||
| file_id = photos[0].data["file"] | ||
|
|
||
| elif documents := msg["document"]: | ||
| doc = next( | ||
| ( | ||
| doc | ||
| for doc in documents | ||
| if doc.data["mime_type"].startswith("image/") | ||
| ), | ||
| None, | ||
| ) | ||
| if doc: | ||
| data = doc.data | ||
| if data["file_size"] > (config.ps_tg_max_file_size * 1024 * 1024): | ||
| logger.warning("附带图片文件过大，回退到默认行为") | ||
| await MessageFactory("附带图片文件过大，将使用默认图片").send( | ||
| reply=config.ps_reply_target, | ||
| ) | ||
| else: | ||
| file_id = data["file_id"] | ||
|
|
||
| if file_id: | ||
| return await download_telegram_file(bot, file_id) | ||
|
|
||
| return None | ||
|
|
||
|
|
||
| _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, | ||
| ) | ||
|
|
||
|
|
||
| @stat_matcher.handle() | ||
| async def _(bot: Bot, event: Event, matcher: Matcher): | ||
| try: | ||
| extract_target(event) | ||
| except RuntimeError: | ||
| logger.warning("SAA 不支持的平台，取消响应") | ||
| return | ||
|
|
||
| pic = None | ||
| try: | ||
| pic = await extract_msg_pic(bot, event) | ||
| except Exception: | ||
| logger.exception("获取消息中附带图片失败，回退到默认行为") | ||
| await MessageFactory("获取消息中附带图片失败，将使用默认图片").send(reply=config.ps_reply_target) | ||
|
|
||
| try: | ||
| ret = await get_stat_pic(bot, pic) | ||
| except Exception: | ||
| logger.exception("获取运行状态图失败") | ||
| await MessageFactory("获取运行状态图片失败，请检查后台输出").send(reply=config.ps_reply_target) | ||
| await matcher.finish() | ||
|
|
||
| await MessageFactory(Image(ret)).finish(reply=config.ps_reply_target) | ||
| from typing import Optional | ||
|
|
||
| from nonebot import logger, on_command | ||
| from nonebot.adapters import Bot, Event, Message | ||
| from nonebot.params import CommandArg | ||
| from nonebot.permission import SUPERUSER | ||
| from nonebot.rule import Rule, to_me | ||
| from nonebot_plugin_saa import Image, MessageFactory, extract_target | ||
|
|
||
| from .config import config | ||
| from .draw import get_stat_pic | ||
| from .util import async_request, download_telegram_file | ||
|
|
||
| try: | ||
| from nonebot.adapters.onebot.v11 import MessageEvent as OBV11MessageEvent | ||
| except ImportError: | ||
| OBV11MessageEvent = None | ||
|
|
||
| try: | ||
| from nonebot.adapters.telegram.event import MessageEvent as TGMessageEvent | ||
| except ImportError: | ||
| TGMessageEvent = None | ||
|
|
||
|
|
||
| def check_empty_arg_rule(arg: Message = CommandArg()): | ||
| return not arg.extract_plain_text() | ||
|
|
||
|
|
||
| def trigger_rule(): | ||
| rule = Rule(check_empty_arg_rule) | ||
| if config.ps_need_at: | ||
| rule = rule & to_me() | ||
| return rule | ||
|
|
||
|
|
||
| async def extract_msg_pic(bot: Bot, event: Event) -> Optional[bytes]: | ||
| if OBV11MessageEvent and isinstance(event, OBV11MessageEvent): | ||
| if (event.reply and (img := event.reply.message["image"])) or ( | ||
| img := event.message["image"] | ||
| ): | ||
| url = img[0].data["url"] | ||
| return await async_request(url) | ||
|
|
||
| elif TGMessageEvent and isinstance(event, TGMessageEvent): | ||
| msg = event.message | ||
| if event.reply_to_message and event.reply_to_message.message: | ||
| msg += event.reply_to_message.message | ||
|
|
||
| file_id = None | ||
| if photos := msg["photo"]: | ||
| file_id = photos[0].data["file"] | ||
|
|
||
| elif documents := msg["document"]: | ||
| doc = next( | ||
| ( | ||
| doc | ||
| for doc in documents | ||
| if doc.data["mime_type"].startswith("image/") | ||
| ), | ||
| None, | ||
| ) | ||
| if doc: | ||
| data = doc.data | ||
| if data["file_size"] > (config.ps_tg_max_file_size * 1024 * 1024): | ||
| logger.warning("附带图片文件过大，回退到默认行为") | ||
| await MessageFactory("附带图片文件过大，将使用默认图片").send( | ||
| reply=config.ps_reply_target, | ||
| ) | ||
| else: | ||
| file_id = data["file_id"] | ||
|
|
||
| if file_id: | ||
| return await download_telegram_file(bot, file_id) | ||
|
|
||
| return None | ||
|
|
||
|
|
||
| _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, | ||
| ) | ||
|
|
||
|
|
||
| @stat_matcher.handle() | ||
| async def _(bot: Bot, event: Event): | ||
| try: | ||
| extract_target(event) | ||
| except RuntimeError: | ||
| logger.warning("SAA 不支持的平台，取消响应") | ||
| return | ||
|
|
||
| pic = None | ||
| try: | ||
| pic = await extract_msg_pic(bot, event) | ||
| except Exception: | ||
| logger.exception("获取消息中附带图片失败，回退到默认行为") | ||
| await MessageFactory("获取消息中附带图片失败，将使用默认图片").send(reply=config.ps_reply_target) | ||
|
|
||
| try: | ||
| ret = await get_stat_pic(bot, pic) | ||
| except Exception: | ||
| logger.exception("获取运行状态图失败") | ||
| await MessageFactory("获取运行状态图片失败，请检查后台输出").finish(reply=config.ps_reply_target) | ||
|
|
||
| await MessageFactory(Image(ret)).finish(reply=config.ps_reply_target) |
Oops, something went wrong.