From cc43e53c4b4aa4c76a0ab57227cfc9d6d6c1fc22 Mon Sep 17 00:00:00 2001 From: snowy Date: Sun, 2 Jun 2024 02:22:16 +0800 Subject: [PATCH] =?UTF-8?q?:sparkles:=20npm=20=E4=BC=9A=E8=AF=9D=E5=BC=80?= =?UTF-8?q?=E5=85=B3=E5=8A=9F=E8=83=BD=E6=96=B0=E5=A2=9E=E7=BE=A4=E5=8F=B7?= =?UTF-8?q?=E5=8F=AF=E9=80=89=E9=A1=B9?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- docs/usage/basic_command.md | 4 +- liteyuki/plugins/liteyuki_pacman/common.py | 6 +-- liteyuki/plugins/liteyuki_pacman/npm.py | 49 ++++++++++++++++------ 3 files changed, 42 insertions(+), 17 deletions(-) diff --git a/docs/usage/basic_command.md b/docs/usage/basic_command.md index 2a75795..8fb7580 100644 --- a/docs/usage/basic_command.md +++ b/docs/usage/basic_command.md @@ -53,7 +53,7 @@ npm search # 通过关键词搜索插件 npm enable-global/disable-global # 全局启用/停用插件 # 群聊仅群主、管理员、超级用户可用，私聊所有人可用 -npm enable/disable # 当前会话启用/停用插件 +npm enable/disable [--group ] # 当前会话启用/停用插件 npm list [page] [num] # 列出所有插件 page为页数，num为每页显示数量 # 所有人 @@ -75,7 +75,7 @@ rpm reload # 重载所有资源包 ```shell npm 插件管理 | update 更新 | install 安装 | uninstall 卸载 | search 搜索 -enable 启用 | disable 停用 | enable-global 全局启用 | disable-global 全局停用 +enable 启用 | disable 停用 | enable-global 全局启用 | disable-global 全局停用 | --group -g rpm 资源包 | load 加载 | unload 卸载 | change 更改 | reload 重载 | list 列表 help 帮助 ``` diff --git a/liteyuki/plugins/liteyuki_pacman/common.py b/liteyuki/plugins/liteyuki_pacman/common.py index 203ecc5..b75fd76 100644 --- a/liteyuki/plugins/liteyuki_pacman/common.py +++ b/liteyuki/plugins/liteyuki_pacman/common.py @@ -141,11 +141,12 @@ def set_plugin_session_enable(event: T_MessageEvent, plugin_name: str, enable: b """ if event_utils.get_message_type(event) == "group": - session = group_db.where_one(Group(), "group_id = ?", str(event_utils.get_group_id(event)), + session: Group = group_db.where_one(Group(), "group_id = ?", str(event_utils.get_group_id(event)), default=Group(group_id=str(event_utils.get_group_id(event)))) else: - session = user_db.where_one(User(), "user_id = ?", str(event_utils.get_user_id(event)), + session: User = user_db.where_one(User(), "user_id = ?", str(event_utils.get_user_id(event)), default=User(user_id=str(event_utils.get_user_id(event)))) + print(session) default_enable = get_plugin_default_enable(plugin_name) if default_enable: if enable: @@ -160,7 +161,6 @@ def set_plugin_session_enable(event: T_MessageEvent, plugin_name: str, enable: b if event_utils.get_message_type(event) == "group": __group_data[str(event_utils.get_group_id(event))] = session - print(session) group_db.save(session) else: __user_data[str(event_utils.get_user_id(event))] = session diff --git a/liteyuki/plugins/liteyuki_pacman/npm.py b/liteyuki/plugins/liteyuki_pacman/npm.py index 7a94c5a..c611496 100644 --- a/liteyuki/plugins/liteyuki_pacman/npm.py +++ b/liteyuki/plugins/liteyuki_pacman/npm.py @@ -24,7 +24,7 @@ from liteyuki.utils.message.tools import clamp from .common import * require("nonebot_plugin_alconna") -from nonebot_plugin_alconna import on_alconna, Alconna, Args, Arparma, Subcommand +from nonebot_plugin_alconna import on_alconna, Alconna, Args, Arparma, Subcommand, Option, OptionResult, SubcommandResult # const enable_global = "enable-global" @@ -40,11 +40,21 @@ disable = "disable" Subcommand( "enable", Args["plugin_name", str], + Option( + "-g|--group", + Args["group_id", str, None], + help_text="群号", + ), alias=["e", "启用"], ), Subcommand( "disable", Args["plugin_name", str], + Option( + "-g|--group", + Args["group_id", str, None], + help_text="群号", + ), alias=["d", "停用"], ), Subcommand( @@ -93,13 +103,33 @@ async def _(result: Arparma, event: T_MessageEvent, bot: T_Bot, npm: Matcher): sc = result.subcommands # 获取子命令 perm_s = await SUPERUSER(bot, event) # 判断是否为超级用户 # 支持对自定义command_start的判断 - if sc.get("enable") or result.subcommands.get("disable"): + if sc.get("enable") or sc.get("disable"): toggle = result.subcommands.get("enable") is not None plugin_exist = get_plugin_exist(plugin_name) - session_enable = get_plugin_session_enable(event, plugin_name) # 获取插件当前状态 + # 判定会话类型 + # 输入群号 + if (group_id := (sc.get("enable", SubcommandResult()).options.get("group", OptionResult()).args.get("group_id") or + sc.get("disable", SubcommandResult()).options.get("group", OptionResult()).args.get("group_id"))) and await SUPERUSER(bot, event): + session_id = group_id + new_event = event.copy() + new_event.group_id = group_id + new_event.message_type = "group" + + elif event.message_type == "private": + session_id = event.user_id + new_event = event + else: + if await GROUP_ADMIN(bot, event) or await GROUP_OWNER(bot, event) or await SUPERUSER(bot, event): + session_id = event.group_id + new_event = event + else: + raise FinishedException(ulang.get("Permission Denied")) + + + session_enable = get_plugin_session_enable(new_event, plugin_name) # 获取插件当前状态 can_be_toggled = get_plugin_can_be_toggle(plugin_name) # 获取插件是否可以被启用/停用 @@ -113,15 +143,10 @@ async def _(result: Arparma, event: T_MessageEvent, bot: T_Bot, npm: Matcher): await npm.finish( ulang.get("npm.plugin_already", NAME=plugin_name, STATUS=ulang.get("npm.enable") if toggle else ulang.get("npm.disable"))) - if event.message_type == "private": - session = user_db.where_one(User(), "user_id = ?", event.user_id, default=User(user_id=event.user_id)) - else: - if await GROUP_ADMIN(bot, event) or await GROUP_OWNER(bot, event) or await SUPERUSER(bot, event): - session = group_db.where_one(Group(), "group_id = ?", event.group_id, default=Group(group_id=str(event.group_id))) - else: - raise FinishedException(ulang.get("Permission Denied")) + # 键入自定义群号的情况 + try: - set_plugin_session_enable(event, plugin_name, toggle) + set_plugin_session_enable(new_event, plugin_name, toggle) except Exception as e: nonebot.logger.error(e) await npm.finish( @@ -136,7 +161,7 @@ async def _(result: Arparma, event: T_MessageEvent, bot: T_Bot, npm: Matcher): ulang.get( "npm.toggle_success", NAME=plugin_name, - STATUS=ulang.get("npm.enable") if toggle else ulang.get("npm.disable")) + STATUS=(ulang.get("npm.enable") if toggle else ulang.get("npm.disable"))) + session_id ) elif sc.get(enable_global) or result.subcommands.get(disable_global) and await SUPERUSER(bot, event):