Diuse/zhenxun_bot
1
0
Fork 0
mirror of https://github.com/zhenxun-org/zhenxun_bot.git synced 2025-12-15 14:22:55 +08:00
Code Issues Actions 1 Packages Projects Releases Wiki Activity

Update set_admin_permissions.py

Browse Source
This commit is contained in:
HibiKier 2022-03-07 20:46:13 +08:00 committed by GitHub
parent c06248e46b
commit f1354308fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 41 additions and 30 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

71
basic_plugins/super_cmd/set_admin_permissions.py
View File

@ -1,7 +1,7 @@
from nonebot import on_command from nonebot import on_command
from nonebot.permission import SUPERUSER from nonebot.permission import SUPERUSER
from models.level_user import LevelUser from models.level_user import LevelUser
from nonebot.adapters.onebot.v11 import Bot, GroupMessageEvent, Message from nonebot.adapters.onebot.v11 import Bot, MessageEvent, Message, GroupMessageEvent
from utils.utils import get_message_at, is_number from utils.utils import get_message_at, is_number
from services.log import logger from services.log import logger
from utils.message_builder import at from utils.message_builder import at
@ -37,41 +37,49 @@ super_cmd = on_command(
@super_cmd.handle() @super_cmd.handle()
async def _(bot: Bot, event: GroupMessageEvent, cmd: Tuple[str, ...] = Command(), arg: Message = CommandArg()): async def _(
cmd = cmd[0] bot: Bot,
group_id = -1 event: MessageEvent,
level = 0 cmd: Tuple[str, ...] = Command(),
arg: Message = CommandArg(),
):
group_id = event.group_id if isinstance(event, GroupMessageEvent) else -1
level = None
args = arg.extract_plain_text().strip().split()
qq = get_message_at(event.json())
flag = 2
try: try:
args = arg.extract_plain_text().strip().split() if qq:
qq = get_message_at(event.json()) qq = qq[0]
flag = -1 if cmd[0][:2] == "添加" and args and is_number(args[0]):
if not qq: level = int(args[0])
if len(args) > 2: else:
if is_number(args[0]) and is_number(args[1]) and is_number(args[2]): if cmd[0][:2] == "添加":
if (
len(args) > 2
and is_number(args[0])
and is_number(args[1])
and is_number(args[2])
):
qq = int(args[0]) qq = int(args[0])
group_id = int(args[1]) group_id = int(args[1])
level = int(args[2]) level = int(args[2])
flag = 1
else:
await super_cmd.finish("所有参数必须是数字！", at_sender=True)
else: else:
await super_cmd.finish( if len(args) > 1 and is_number(args[0]) and is_number(args[1]):
"权限参数不完全\n\t格式：添加/删除权限 [at] [level]" qq = int(args[0])
"\n\t格式：添加/删除权限 [qq] [group_id] [level]", group_id = int(args[1])
at_sender=True, flag = 1
) level = -1 if cmd[0][:2] == "删除" else level
else: if group_id == -1 or not level or not qq:
if not is_number(args[0]): raise IndexError()
await super_cmd.finish("所有参数必须是数字！", at_sender=True) except IndexError:
level = int(args[0]) await super_cmd.finish(__plugin_usage__)
qq = qq[0] try:
group_id = event.group_id if cmd[0][:2] == "添加":
flag = 2
if cmd[:2] == "添加":
if await LevelUser.set_level(qq, group_id, level, 1): if await LevelUser.set_level(qq, group_id, level, 1):
result = "添加管理成功, 权限: " + str(level) result = f"添加管理成功, 权限: {level}"
else: else:
result = "管理已存在, 更新权限: " + str(level) result = f"管理已存在, 更新权限: {level}"
else: else:
if await LevelUser.delete_level(qq, event.group_id): if await LevelUser.delete_level(qq, event.group_id):
result = "删除管理成功!" result = "删除管理成功!"
@ -82,7 +90,10 @@ async def _(bot: Bot, event: GroupMessageEvent, cmd: Tuple[str, ...] = Command()
elif flag == 1: elif flag == 1:
await bot.send_group_msg( await bot.send_group_msg(
group_id=group_id, group_id=group_id,
message=Message(f"{at(qq)}管理员修改了你的权限" f"\n--------\n你当前的权限等级：{level}"), message=Message(
f"{at(qq)}管理员修改了你的权限"
f"\n--------\n你当前的权限等级：{level if level != -1 else 0}"
),
) )
await super_cmd.send("修改成功") await super_cmd.send("修改成功")
except Exception as e: except Exception as e: