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

🐛 修复超级用户对群组功能开关

Browse Source
This commit is contained in:
HibiKier 2024-09-02 11:43:01 +08:00
parent 133d4e193a
commit a56d1f2bd8
2 changed files with 51 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
zhenxun/builtin_plugins/admin/plugin_switch/__init__.py
View File

@ -94,7 +94,7 @@ async def _(
if session.id1 in bot.config.superusers:
image = await build_plugin()
logger.info(
f"查看功能列表",
"查看功能列表",
arparma.header_result,
session=session,
)
@ -123,7 +123,7 @@ async def _(
if task.result:
if all.result:
result = await PluginManage.unblock_group_all_task(gid)
logger.info(f"开启所有群组被动", arparma.header_result, session=session)
logger.info("开启所有群组被动", arparma.header_result, session=session)
else:
result = await PluginManage.unblock_group_task(name, gid)
logger.info(
@ -145,7 +145,7 @@ async def _(
True, default_status.result, gid
)
logger.info(
f"开启群组中全部功能",
"开启群组中全部功能",
arparma.header_result,
session=session,
)
@ -206,7 +206,7 @@ async def _(
)
await MessageUtils.build_message(result).finish(reply_to=True)
else:
result = await PluginManage.superuser_block(name, None, group_id)
result = await PluginManage.superuser_unblock(name, None, group_id)
logger.info(
f"超级用户开启功能 {name}",
arparma.header_result,

47
zhenxun/builtin_plugins/admin/plugin_switch/_data_source.py
View File

@ -546,3 +546,50 @@ class PluginManage:
if block_type == BlockType.PRIVATE:
return f"已成功将 {plugin.name} 全局私聊关闭!"
return "没有找到这个功能喔..."
@classmethod
async def superuser_unblock(
cls, plugin_name: str, block_type: BlockType | None, group_id: str | None
) -> str:
"""超级用户开启插件
参数:
plugin_name: 插件名称
block_type: 禁用类型
group_id: 群组id
返回:
str: 返回信息
"""
if plugin_name.isdigit():
plugin = await PluginInfo.get_or_none(id=int(plugin_name))
else:
plugin = await PluginInfo.get_or_none(name=plugin_name)
if plugin:
if group_id:
if group := await GroupConsole.get_or_none(
group_id=group_id, channel_id__isnull=True
):
if f"super:{plugin.module}," in group.block_plugin:
group.block_plugin = group.block_plugin.replace(
f"super:{plugin.module},", ""
)
await group.save(update_fields=["block_plugin"])
return (
f"已成功开启群组 {group.group_name} 的 {plugin_name} 功能!"
)
return "此群组该功能已被超级用户开启，不要重复开启..."
return "群组信息未更新，请先更新群组信息..."
plugin.block_type = block_type
plugin.status = not bool(block_type)
await plugin.save(update_fields=["status", "block_type"])
if not block_type:
return f"已成功将 {plugin.name} 全局启用!"
else:
if block_type == BlockType.ALL:
return f"已成功将 {plugin.name} 全局开启!"
if block_type == BlockType.GROUP:
return f"已成功将 {plugin.name} 全局群组开启!"
if block_type == BlockType.PRIVATE:
return f"已成功将 {plugin.name} 全局私聊开启!"
return "没有找到这个功能喔..."