2021-11-23 21:44:59 +08:00
|
|
|
|
from nonebot import on_command
|
|
|
|
|
|
from nonebot.permission import SUPERUSER
|
|
|
|
|
|
from models.level_user import LevelUser
|
|
|
|
|
|
from nonebot.typing import T_State
|
|
|
|
|
|
from nonebot.adapters.cqhttp import Bot, GroupMessageEvent, Message
|
|
|
|
|
|
from utils.utils import get_message_at, get_message_text, is_number
|
|
|
|
|
|
from services.log import logger
|
|
|
|
|
|
from utils.message_builder import at
|
|
|
|
|
|
|
|
|
|
|
|
__zx_plugin_name__ = "用户权限管理 [Superuser]"
|
|
|
|
|
|
__plugin_usage__ = """
|
|
|
|
|
|
usage：
|
|
|
|
|
|
增删改用户的权限
|
|
|
|
|
|
指令：
|
|
|
|
|
|
添加权限 [at] [权限]
|
|
|
|
|
|
添加权限 [qq] [group_id] [权限]
|
|
|
|
|
|
删除权限 [at]
|
|
|
|
|
|
""".strip()
|
|
|
|
|
|
__plugin_des__ = "增删改用户的权限"
|
|
|
|
|
|
__plugin_cmd__ = [
|
|
|
|
|
|
"添加权限 [at] [权限]",
|
|
|
|
|
|
"添加权限 [qq] [group_id] [权限]",
|
|
|
|
|
|
"删除权限 [at]",
|
|
|
|
|
|
]
|
|
|
|
|
|
__plugin_version__ = 0.1
|
|
|
|
|
|
__plugin_author__ = "HibiKier"
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
super_cmd = on_command(
|
|
|
|
|
|
"添加管理",
|
|
|
|
|
|
aliases={"删除管理", "添加权限", "删除权限"},
|
|
|
|
|
|
priority=1,
|
|
|
|
|
|
permission=SUPERUSER,
|
|
|
|
|
|
block=True,
|
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@super_cmd.handle()
|
|
|
|
|
|
async def _(bot: Bot, event: GroupMessageEvent, state: T_State):
|
|
|
|
|
|
group_id = -1
|
|
|
|
|
|
level = 0
|
|
|
|
|
|
try:
|
|
|
|
|
|
args = get_message_text(event.json()).strip().split()
|
|
|
|
|
|
qq = get_message_at(event.json())
|
|
|
|
|
|
flag = -1
|
|
|
|
|
|
if not qq:
|
|
|
|
|
|
if len(args) > 2:
|
|
|
|
|
|
if is_number(args[0]) and is_number(args[1]) and is_number(args[2]):
|
|
|
|
|
|
qq = int(args[0])
|
|
|
|
|
|
group_id = int(args[1])
|
|
|
|
|
|
level = int(args[2])
|
|
|
|
|
|
flag = 1
|
|
|
|
|
|
else:
|
|
|
|
|
|
await super_cmd.finish("所有参数必须是数字！", at_sender=True)
|
|
|
|
|
|
else:
|
|
|
|
|
|
await super_cmd.finish(
|
|
|
|
|
|
"权限参数不完全\n\t格式：添加/删除权限 [at] [level]"
|
|
|
|
|
|
"\n\t格式：添加/删除权限 [qq] [group] [level]",
|
|
|
|
|
|
at_sender=True,
|
|
|
|
|
|
)
|
|
|
|
|
|
else:
|
|
|
|
|
|
qq = qq[0]
|
|
|
|
|
|
group_id = event.group_id
|
|
|
|
|
|
flag = 2
|
|
|
|
|
|
if state["_prefix"]["raw_command"][:2] == "添加":
|
|
|
|
|
|
if is_number(args[0]):
|
|
|
|
|
|
level = int(args[0])
|
|
|
|
|
|
else:
|
|
|
|
|
|
await super_cmd.finish("权限等级必须是数字！", at_sender=True)
|
|
|
|
|
|
if await LevelUser.set_level(qq, group_id, level, 1):
|
|
|
|
|
|
result = "添加管理成功, 权限: " + str(level)
|
|
|
|
|
|
else:
|
|
|
|
|
|
result = "管理已存在, 更新权限: " + str(level)
|
|
|
|
|
|
else:
|
|
|
|
|
|
if await LevelUser.delete_level(qq, event.group_id):
|
|
|
|
|
|
result = "删除管理成功!"
|
|
|
|
|
|
else:
|
|
|
|
|
|
result = "该账号无管理权限!"
|
|
|
|
|
|
if flag == 2:
|
|
|
|
|
|
await super_cmd.send(result)
|
|
|
|
|
|
elif flag == 1:
|
|
|
|
|
|
await bot.send_group_msg(
|
|
|
|
|
|
group_id=group_id,
|
|
|
|
|
|
message=Message(f"{at(qq)}管理员修改了你的权限" f"\n--------\n你当前的权限等级：{level}"),
|
|
|
|
|
|
)
|
|
|
|
|
|
await super_cmd.send("修改成功")
|
|
|
|
|
|
except Exception as e:
|
|
|
|
|
|
await super_cmd.send("执行指令失败!")
|
|
|
|
|
|
logger.error(f"执行指令失败 e：{e}")
|
