perf👌: 广播与全局插件/被动管理

zhenxun/builtin_plugins/admin/plugin_switch/__init__.py

@@ -1,6 +1,6 @@
 from nonebot.adapters import Bot
 from nonebot.plugin import PluginMetadata
-from nonebot_plugin_alconna import Arparma, Match
+from nonebot_plugin_alconna import AlconnaQuery, Arparma, Match, Query
 from nonebot_plugin_saa import Image, Text
 from nonebot_plugin_session import EventSession
 
@@ -21,14 +21,16 @@ __plugin_meta__ = PluginMetadata(
     usage="""
     普通管理员
         格式:
-        开启/关闭[功能名称]    : 开关功能
-        群被动状态            : 查看被动技能开关状态
-        醒来                 : 结束休眠
-        休息吧                : 群组休眠, 不会再响应命令
+        开启/关闭[功能名称]         : 开关功能
+        开启/关闭群被动[被动名称]    : 群被动开关
+        群被动状态                 : 查看被动技能开关状态
+        醒来                      : 结束休眠
+        休息吧                    : 群组休眠, 不会再响应命令
 
         示例:
         开启签到              : 开启签到
         关闭签到              : 关闭签到
+        开启群被动早晚安       : 关闭被动任务早晚安
 
     超级管理员额外命令
         格式:
@@ -61,26 +63,21 @@ __plugin_meta__ = PluginMetadata(
 
 
 @_status_matcher.assign("$main")
-async def _(bot: Bot, session: EventSession, arparma: Arparma):
+async def _(
+    bot: Bot,
+    session: EventSession,
+    arparma: Arparma,
+    task: Query[bool] = AlconnaQuery("task.value", False),
+):
     image = None
-    if session.id1 in bot.config.superusers:
+    if task.result:
+        image = await build_task(session.id3 or session.id2)
+    elif session.id1 in bot.config.superusers:
         image = await build_plugin()
     if image:
         await Image(image.pic2bytes()).send(reply=True)
         logger.info(
-            f"查看功能列表",
-            arparma.header_result,
-            session=session,
-        )
-
-
-@_status_matcher.assign("task")
-async def _(bot: Bot, session: EventSession, arparma: Arparma):
-    image = None
-    if image := await build_task(session.id3 or session.id2):
-        await Image(image.pic2bytes()).send(reply=True)
-        logger.info(
-            f"查看被动列表",
+            f"查看{'功能' if arparma.find('task') else '被动'}列表",
             arparma.header_result,
             session=session,
         )
@@ -93,21 +90,35 @@ async def _(
     arparma: Arparma,
     name: str,
     group: Match[str],
+    task: Query[bool] = AlconnaQuery("task.value", False),
 ):
     if gid := session.id3 or session.id2:
-        result = await PluginManage.block_group_plugin(name, gid)
+        if task.result:
+            result = await PluginManage.unblock_group_task(name, gid)
+        else:
+            result = await PluginManage.block_group_plugin(name, gid)
         await Text(result).send(reply=True)
         logger.info(f"开启功能 {name}", arparma.header_result, session=session)
     elif session.id1 in bot.config.superusers:
         group_id = group.result if group.available else None
-        result = await PluginManage.superuser_block(name, None, group_id)
-        await Text(result).send(reply=True)
-        logger.info(
-            f"超级用户开启功能 {name}",
-            arparma.header_result,
-            session=session,
-            target=group_id,
-        )
+        if task.result:
+            result = await PluginManage.superuser_task_handle(name, group_id, True)
+            await Text(result).send(reply=True)
+            logger.info(
+                f"超级用户开启被动技能 {name}",
+                arparma.header_result,
+                session=session,
+                target=group_id,
+            )
+        else:
+            result = await PluginManage.superuser_block(name, None, group_id)
+            await Text(result).send(reply=True)
+            logger.info(
+                f"超级用户开启功能 {name}",
+                arparma.header_result,
+                session=session,
+                target=group_id,
+            )
 
 
 @_status_matcher.assign("close")
@@ -118,27 +129,41 @@ async def _(
     name: str,
     block_type: Match[str],
     group: Match[str],
+    task: Query[bool] = AlconnaQuery("task.value", False),
 ):
     if gid := session.id3 or session.id2:
-        result = await PluginManage.unblock_group_plugin(name, gid)
+        if task.result:
+            result = await PluginManage.block_group_task(name, gid)
+        else:
+            result = await PluginManage.unblock_group_plugin(name, gid)
         await Text(result).send(reply=True)
         logger.info(f"关闭功能 {name}", arparma.header_result, session=session)
     elif session.id1 in bot.config.superusers:
         group_id = group.result if group.available else None
-        _type = BlockType.ALL
-        if block_type.available:
-            if block_type.result in ["p", "private"]:
-                _type = BlockType.PRIVATE
-            elif block_type.result in ["g", "group"]:
-                _type = BlockType.GROUP
-        result = await PluginManage.superuser_block(name, _type, group_id)
-        await Text(result).send(reply=True)
-        logger.info(
-            f"超级用户关闭功能 {name}, 禁用类型: {_type}",
-            arparma.header_result,
-            session=session,
-            target=group_id,
-        )
+        if task.result:
+            result = await PluginManage.superuser_task_handle(name, group_id, False)
+            await Text(result).send(reply=True)
+            logger.info(
+                f"超级用户关闭被动技能 {name}",
+                arparma.header_result,
+                session=session,
+                target=group_id,
+            )
+        else:
+            _type = BlockType.ALL
+            if block_type.available:
+                if block_type.result in ["p", "private"]:
+                    _type = BlockType.PRIVATE
+                elif block_type.result in ["g", "group"]:
+                    _type = BlockType.GROUP
+            result = await PluginManage.superuser_block(name, _type, group_id)
+            await Text(result).send(reply=True)
+            logger.info(
+                f"超级用户关闭功能 {name}, 禁用类型: {_type}",
+                arparma.header_result,
+                session=session,
+                target=group_id,
+            )
 
 
 @_group_status_matcher.handle()

zhenxun/builtin_plugins/admin/plugin_switch/_data_source.py

@@ -104,7 +104,9 @@ async def build_task(group_id: str | None) -> BuildImage:
     column_name = ["ID", "模块", "名称", "群组状态", "全局状态", "运行时间"]
     group = None
     if group_id:
-        group = await GroupConsole.get_or_none(group_id=group_id)
+        group = await GroupConsole.get_or_none(
+            group_id=group_id, channel_id__isnull=True
+        )
         if not group:
             raise GroupInfoNotFound()
     else:
@@ -145,17 +147,23 @@ class PluginManage:
 
     @classmethod
     async def is_wake(cls, group_id: str) -> bool:
-        if c := await GroupConsole.get_or_none(group_id=group_id):
+        if c := await GroupConsole.get_or_none(
+            group_id=group_id, channel_id__isnull=True
+        ):
             return c.status
         return False
 
     @classmethod
     async def sleep(cls, group_id: str):
-        await GroupConsole.filter(group_id=group_id).update(status=False)
+        await GroupConsole.filter(group_id=group_id, channel_id__isnull=True).update(
+            status=False
+        )
 
     @classmethod
     async def wake(cls, group_id: str):
-        await GroupConsole.filter(group_id=group_id).update(status=True)
+        await GroupConsole.filter(group_id=group_id, channel_id__isnull=True).update(
+            status=True
+        )
 
     @classmethod
     async def block(cls, module: str):
@@ -178,6 +186,32 @@ class PluginManage:
         """
         return await cls._change_group_plugin(plugin_name, group_id, True)
 
+    @classmethod
+    async def unblock_group_task(cls, task_name: str, group_id: str) -> str:
+        """启用被动技能
+
+        参数:
+            task_name: 被动技能名称
+            group_id: 群组id
+
+        返回:
+            str: 返回信息
+        """
+        return await cls._change_group_task(task_name, group_id, False)
+
+    @classmethod
+    async def block_group_task(cls, task_name: str, group_id: str) -> str:
+        """禁用被动技能
+
+        参数:
+            task_name: 被动技能名称
+            group_id: 群组id
+
+        返回:
+            str: 返回信息
+        """
+        return await cls._change_group_task(task_name, group_id, True)
+
     @classmethod
     async def unblock_group_plugin(cls, plugin_name: str, group_id: str) -> str:
         """启用群组插件
@@ -191,6 +225,33 @@ class PluginManage:
         """
         return await cls._change_group_plugin(plugin_name, group_id, False)
 
+    @classmethod
+    async def _change_group_task(
+        cls, task_name: str, group_id: str, status: bool
+    ) -> str:
+        """改变群组被动技能状态
+
+        参数:
+            task_name: 被动技能名称
+            group_id: 群组Id
+            status: 状态
+
+        返回:
+            str: 返回信息
+        """
+        if task := await TaskInfo.get_or_none(name=task_name):
+            status_str = "关闭" if status else "开启"
+            group, _ = await GroupConsole.get_or_create(
+                group_id=group_id, channel_id__isnull=True
+            )
+            if status:
+                group.block_task += f"{task.module},"
+            else:
+                group.block_task = group.block_task.replace(f"{task.module},", "")
+            await group.save(update_fields=["block_task"])
+            return f"已成功{status_str} {task_name} 被动技能!"
+        return "没有找到这个被动技能喔..."
+
     @classmethod
     async def _change_group_plugin(
         cls, plugin_name: str, group_id: str, status: bool
@@ -212,7 +273,9 @@ class PluginManage:
             plugin = await PluginInfo.get_or_none(name=plugin_name)
         status_str = "开启" if status else "关闭"
         if plugin:
-            group, _ = await GroupConsole.get_or_create(group_id=group_id)
+            group, _ = await GroupConsole.get_or_create(
+                group_id=group_id, channel_id__isnull=True
+            )
             if status:
                 if plugin.module in group.block_plugin:
                     group.block_plugin = group.block_plugin.replace(
@@ -228,11 +291,44 @@ class PluginManage:
             return f"该功能已经{status_str}了喔，不要重复{status_str}..."
         return "没有找到这个功能喔..."
 
+    @classmethod
+    async def superuser_task_handle(
+        cls, task_name: str, group_id: str | None, status: bool
+    ) -> str:
+        """超级用户禁用被动技能
+
+        参数:
+            task_name: 被动技能名称
+            group_id: 群组id
+            status: 状态
+
+        返回:
+            str: 返回信息
+        """
+        if task := await TaskInfo.get_or_none(name=task_name):
+            status_str = "开启" if status else "关闭"
+            if group_id:
+                group, _ = await GroupConsole.get_or_create(
+                    group_id=group_id, channel_id__isnull=True
+                )
+                if status:
+                    group.block_task = group.block_task.replace(
+                        f"super:{task.module},", ""
+                    )
+                else:
+                    group.block_task += f"super:{task.module},"
+                await group.save(update_fields=["block_task"])
+            else:
+                task.status = status
+                await task.save(update_fields=["status"])
+            return f"已成功将被动技能 {task_name} 全局{status_str}!"
+        return "没有找到这个功能喔..."
+
     @classmethod
     async def superuser_block(
         cls, plugin_name: str, block_type: BlockType | None, group_id: str | None
     ) -> str:
-        """超级用户禁用
+        """超级用户禁用插件
 
         参数:
             plugin_name: 插件名称
@@ -248,7 +344,9 @@ class PluginManage:
             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):
+                if group := await GroupConsole.get_or_none(
+                    group_id=group_id, channel_id__isnull=True
+                ):
                     if f"super:{plugin.module}," not in group.block_plugin:
                         group.block_plugin += f"super:{plugin.module},"
                         await group.save(update_fields=["block_plugin"])

zhenxun/builtin_plugins/admin/plugin_switch/command.py

@@ -63,6 +63,15 @@ _status_matcher.shortcut(
     prefix=True,
 )
 
+
+_status_matcher.shortcut(
+    r"开启群被动(?P<name>.+)",
+    command="switch",
+    arguments=["open", "{name}", "--task"],
+    prefix=True,
+)
+
+
 _status_matcher.shortcut(
     r"开启(?P<name>.+)",
     command="switch",
@@ -70,6 +79,13 @@ _status_matcher.shortcut(
     prefix=True,
 )
 
+_status_matcher.shortcut(
+    r"关闭群被动(?P<name>.+)",
+    command="switch",
+    arguments=["close", "{name}", "--task"],
+    prefix=True,
+)
+
 _status_matcher.shortcut(
     r"关闭(?P<name>.+)",
     command="switch",
@@ -77,6 +93,7 @@ _status_matcher.shortcut(
     prefix=True,
 )
 
+
 _group_status_matcher.shortcut(
     r"醒来",
     command="group-status",