From e01b1ed882a01d03b763e9b06b8b5d0608fc5f21 Mon Sep 17 00:00:00 2001 From: Kyomotoi <1172294279@qq.com> Date: Sat, 24 Apr 2021 16:20:19 +0800 Subject: =?UTF-8?q?=E2=9C=A8=F0=9F=94=A5=F0=9F=8E=A8=F0=9F=90=9B=E2=99=BB?= =?UTF-8?q?=EF=B8=8F=F0=9F=93=9D=20=E6=9B=B4=E6=96=B0?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit 新增：开发工具中ub paste，再也不怕长消息了。 新增：错误处理两项`ReadFileError`、`FormatError` 修复：bot退群显示被自己踢出群 删除：Service部分内容 删除：hitokoto本地库，改用请求 优化：b站小程序抓取 重构：bot管理部分，对命令进行简化 --- ATRI/plugins/manage/modules/service.py | 205 +++++++++++++++++++++++++++++++++ 1 file changed, 205 insertions(+) create mode 100644 ATRI/plugins/manage/modules/service.py (limited to 'ATRI/plugins/manage/modules/service.py') diff --git a/ATRI/plugins/manage/modules/service.py b/ATRI/plugins/manage/modules/service.py new file mode 100644 index 0000000..8bb7e9c --- /dev/null +++ b/ATRI/plugins/manage/modules/service.py @@ -0,0 +1,205 @@ +from nonebot.typing import T_State +from nonebot.permission import SUPERUSER +from nonebot.adapters.cqhttp.permission import GROUP_ADMIN, GROUP_OWNER +from nonebot.adapters.cqhttp import ( + Bot, + MessageEvent, + GroupMessageEvent, + PrivateMessageEvent +) + +from ATRI.service import Service as sv + + +__doc__ = """ +启用功能，针对单群 +权限组：维护者，群管理 +用法： + 启用 目标命令 +示例： + 启用 以图搜图 +""" + +cur_service_ena = sv.on_command( + cmd="启用功能", + docs=__doc__, + permission=SUPERUSER|GROUP_ADMIN|GROUP_OWNER +) + +@cur_service_ena.args_parser # type: ignore +async def _cur_ena_load(bot: Bot, + event: GroupMessageEvent, + state: T_State) -> None: + msg = str(event.message).strip() + quit_list = ['算了', '罢了', '取消'] + if msg in quit_list: + await cur_service_ena.finish('好吧...') + if not msg: + await cur_service_ena.reject('请告诉咱目标命令！') + else: + state['service_e'] = msg + +@cur_service_ena.handle() +async def _cur_ena(bot: Bot, + event: GroupMessageEvent, + state: T_State) -> None: + msg = str(event.message).strip() + if msg: + state['service_e'] = msg + +@cur_service_ena.got('service_e', prompt='请告诉咱目标命令！') +async def _deal_cur_ena(bot: Bot, + event: GroupMessageEvent, + state: T_State) -> None: + cmd = state['service_e'] + group = str(event.group_id) + sv.control_service(cmd, False, True, group=group) + await cur_service_ena.finish(f'成功！本群已启用：{cmd}') + +@cur_service_ena.handle() +async def _refuse_cur_ena(bot: Bot, + event: PrivateMessageEvent, + state: T_State) -> None: + await cur_service_ena.finish('只能在群聊中决定哦...') + + +__doc__ = """ +禁用功能，针对单群 +权限组：维护者，群管理 +用法： + 禁用 目标命令 +示例： + 禁用 以图搜图 +""" + +cur_service_dis = sv.on_command( + cmd="禁用功能", + docs=__doc__, + permission=SUPERUSER|GROUP_ADMIN|GROUP_OWNER +) + +@cur_service_dis.args_parser # type: ignore +async def _cur_dis_load(bot: Bot, + event: GroupMessageEvent, + state: T_State) -> None: + msg = str(event.message).strip() + quit_list = ['算了', '罢了', '取消'] + if msg in quit_list: + await cur_service_dis.finish('好吧...') + if not msg: + await cur_service_dis.reject('请告诉咱目标命令！') + else: + state['service_d'] = msg + +@cur_service_dis.handle() +async def _cur_dis(bot: Bot, + event: GroupMessageEvent, + state: T_State) -> None: + msg = str(event.message).strip() + if msg: + state['service_d'] = msg + +@cur_service_dis.got('service_d', prompt='请告诉咱目标命令！') +async def _deal_cur_dis(bot: Bot, + event: GroupMessageEvent, + state: T_State) -> None: + cmd = state['service_d'] + group = str(event.group_id) + sv.control_service(cmd, False, False, group=group) + await cur_service_dis.finish(f'成功！本群已禁用：{cmd}') + +@cur_service_dis.handle() +async def _refuse_cur_dis(bot: Bot, + event: PrivateMessageEvent, + state: T_State) -> None: + await cur_service_dis.finish('只能在群聊中决定哦...') + + +__doc__ = """ +全局启用功能 +权限组：维护者 +用法： + 全局启用 目标命令 +示例： + 全局启用 以图搜图 +""" + +glo_service_ena = sv.on_command( + cmd="全局启用", + docs=__doc__, + permission=SUPERUSER +) + +@glo_service_ena.args_parser # type: ignore +async def _glo_ena_load(bot: Bot, + event: MessageEvent, + state: T_State) -> None: + msg = str(event.message).strip() + quit_list = ['算了', '罢了', '取消'] + if msg in quit_list: + await glo_service_ena.finish('好吧...') + if not msg: + await glo_service_ena.reject('请告诉咱目标命令！') + else: + state['service_e_g'] = msg + +@glo_service_ena.handle() +async def _glo_ena(bot: Bot, + event: MessageEvent, + state: T_State) -> None: + msg = str(event.message).strip() + if msg: + state['service_e_g'] = msg + +@glo_service_ena.got('service_e_g', prompt='请告诉咱目标命令！') +async def _deal_glo_ena(bot: Bot, + event: MessageEvent, + state: T_State) -> None: + cmd = state['service_e_g'] + sv.control_service(cmd, True, True) + await glo_service_ena.finish(f'成功！已全局启用：{cmd}') + + +__doc__ = """ +全局禁用功能 +权限组：维护者 +用法： + 禁用启用 目标命令 +示例： + 禁用启用 以图搜图 +""" + +glo_service_dis = sv.on_command( + cmd="全局禁用", + docs=__doc__, + permission=SUPERUSER +) + +@glo_service_dis.args_parser # type: ignore +async def _glo_dis_load(bot: Bot, + event: MessageEvent, + state: T_State) -> None: + msg = str(event.message).strip() + quit_list = ['算了', '罢了', '取消'] + if msg in quit_list: + await glo_service_dis.finish('好吧...') + if not msg: + await glo_service_dis.reject('请告诉咱目标命令！') + else: + state['service_d_g'] = msg + +@glo_service_dis.handle() +async def _glo_dis(bot: Bot, + event: MessageEvent, + state: T_State) -> None: + msg = str(event.message).strip() + if msg: + state['service_d_g'] = msg + +@glo_service_dis.got('service_d_g', prompt='请告诉咱目标命令！') +async def _deal_glo_dis(bot: Bot, + event: MessageEvent, + state: T_State) -> None: + cmd = state['service_d_g'] + sv.control_service(cmd, True, False) + await glo_service_dis.finish(f'成功！已全局禁用：{cmd}') -- cgit v1.2.3