版本更新

zhenxun/models/group_console.py

@@ -9,7 +9,6 @@ from zhenxun.services.db_context import Model
 
 
 class GroupConsole(Model):
-
     id = fields.IntField(pk=True, generated=True, auto_increment=True)
     """自增id"""
     group_id = fields.CharField(255, description="群组id")
@@ -34,8 +33,14 @@ class GroupConsole(Model):
     """群认证标记"""
     block_plugin = fields.TextField(default="", description="禁用插件")
     """禁用插件"""
+    superuser_block_plugin = fields.TextField(
+        default="", description="超级用户禁用插件"
+    )
+    """超级用户禁用插件"""
     block_task = fields.TextField(default="", description="禁用插件")
     """禁用插件"""
+    superuser_block_task = fields.TextField(default="", description="超级用户禁用被动")
+    """超级用户禁用被动"""
     platform = fields.CharField(255, default="qq", description="所属平台")
     """所属平台"""
 
@@ -53,6 +58,7 @@ class GroupConsole(Model):
         if modules := await TaskInfo.filter(default_status=False).values_list(
             "module", flat=True
         ):
+            modules = [f"<{module}" for module in modules]
             group.block_task = ",".join(modules) + ","  # type: ignore
             await group.save(using_db=using_db, update_fields=["block_task"])
         return group
@@ -73,6 +79,7 @@ class GroupConsole(Model):
                 "module", flat=True
             )
         ):
+            modules = [f"<{module}" for module in modules]
             group.block_task = ",".join(modules) + ","  # type: ignore
             await group.save(using_db=using_db, update_fields=["block_task"])
         return group, is_create
@@ -93,6 +100,7 @@ class GroupConsole(Model):
                 "module", flat=True
             )
         ):
+            modules = [f"<{module}" for module in modules]
             group.block_task = ",".join(modules) + ","  # type: ignore
             await group.save(using_db=using_db, update_fields=["block_task"])
         return group, is_create
@@ -127,27 +135,94 @@ class GroupConsole(Model):
         return group.is_super if (group := await cls.get_group(group_id)) else False
 
     @classmethod
-    async def is_super_block_plugin(
-        cls, group_id: str, module: str, channel_id: str | None = None
-    ) -> bool:
+    async def is_superuser_block_plugin(cls, group_id: str, module: str) -> bool:
         """查看群组是否超级用户禁用功能
 
         参数:
             group_id: 群组id
             module: 模块名称
-            channel_id: 频道id
 
         返回:
             bool: 是否禁用被动
         """
         return await cls.exists(
             group_id=group_id,
-            channel_id=channel_id,
-            block_plugin__contains=f"super:{module},",
+            superuser_block_plugin__contains=f"<{module},",
         )
 
     @classmethod
-    async def is_block_plugin(
+    async def is_block_plugin(cls, group_id: str, module: str) -> bool:
+        """查看群组是否禁用插件
+
+        参数:
+            group_id: 群组id
+            plugin: 插件名称
+
+        返回:
+            bool: 是否禁用插件
+        """
+        return await cls.exists(
+            group_id=group_id, block_plugin__contains=f"<{module},"
+        ) or await cls.exists(
+            group_id=group_id, superuser_block_plugin__contains=f"<{module},"
+        )
+
+    @classmethod
+    async def set_block_plugin(
+        cls,
+        group_id: str,
+        module: str,
+        is_superuser: bool = False,
+        platform: str | None = None,
+    ):
+        """禁用群组插件
+
+        参数:
+            group_id: 群组id
+            task: 任务模块
+            is_superuser: 是否为超级用户
+            platform: 平台
+        """
+        group, _ = await cls.get_or_create(
+            group_id=group_id, defaults={"platform": platform}
+        )
+        if is_superuser:
+            if f"<{module}," not in group.superuser_block_plugin:
+                group.superuser_block_plugin += f"<{module},"
+        elif f"<{module}," not in group.block_plugin:
+            group.block_plugin += f"<{module},"
+        await group.save(update_fields=["block_plugin", "superuser_block_plugin"])
+
+    @classmethod
+    async def set_unblock_plugin(
+        cls,
+        group_id: str,
+        module: str,
+        is_superuser: bool = False,
+        platform: str | None = None,
+    ):
+        """禁用群组插件
+
+        参数:
+            group_id: 群组id
+            task: 任务模块
+            is_superuser: 是否为超级用户
+            platform: 平台
+        """
+        group, _ = await cls.get_or_create(
+            group_id=group_id, defaults={"platform": platform}
+        )
+        if is_superuser:
+            if f"<{module}," in group.superuser_block_plugin:
+                group.superuser_block_plugin = group.superuser_block_plugin.replace(
+                    f"<{module},", ""
+                )
+        elif f"<{module}," in group.block_plugin:
+            group.block_plugin = group.block_plugin.replace(f"<{module},", "")
+        await group.save(update_fields=["block_plugin", "superuser_block_plugin"])
+
+    @classmethod
+    async def is_normal_block_plugin(
         cls, group_id: str, module: str, channel_id: str | None = None
     ) -> bool:
         """查看群组是否禁用功能
@@ -163,7 +238,23 @@ class GroupConsole(Model):
         return await cls.exists(
             group_id=group_id,
             channel_id=channel_id,
-            block_plugin__contains=f"{module},",
+            block_plugin__contains=f"<{module},",
+        )
+
+    @classmethod
+    async def is_superuser_block_task(cls, group_id: str, task: str) -> bool:
+        """查看群组是否超级用户禁用被动
+
+        参数:
+            group_id: 群组id
+            task: 模块名称
+
+        返回:
+            bool: 是否禁用被动
+        """
+        return await cls.exists(
+            group_id=group_id,
+            superuser_block_task__contains=f"<{task},",
         )
 
     @classmethod
@@ -184,12 +275,79 @@ class GroupConsole(Model):
             return await cls.exists(
                 group_id=group_id,
                 channel_id__isnull=True,
-                block_task__contains=f"{task},",
+                block_task__contains=f"<{task},",
+            ) or await cls.exists(
+                group_id=group_id,
+                channel_id__isnull=True,
+                superuser_block_task__contains=f"<{task},",
             )
         return await cls.exists(
-            group_id=group_id, channel_id=channel_id, block_task__contains=f"{task},"
+            group_id=group_id, channel_id=channel_id, block_task__contains=f"<{task},"
+        ) or await cls.exists(
+            group_id=group_id,
+            channel_id__isnull=True,
+            superuser_block_task__contains=f"<{task},",
         )
 
+    @classmethod
+    async def set_block_task(
+        cls,
+        group_id: str,
+        task: str,
+        is_superuser: bool = False,
+        platform: str | None = None,
+    ):
+        """禁用群组插件
+
+        参数:
+            group_id: 群组id
+            task: 任务模块
+            is_superuser: 是否为超级用户
+            platform: 平台
+        """
+        group, _ = await cls.get_or_create(
+            group_id=group_id, defaults={"platform": platform}
+        )
+        if is_superuser:
+            if f"<{task}," not in group.superuser_block_task:
+                group.superuser_block_task += f"<{task},"
+        elif f"<{task}," not in group.block_task:
+            group.block_task += f"<{task},"
+        await group.save(update_fields=["block_task", "superuser_block_task"])
+
+    @classmethod
+    async def set_unblock_task(
+        cls,
+        group_id: str,
+        task: str,
+        is_superuser: bool = False,
+        platform: str | None = None,
+    ):
+        """禁用群组插件
+
+        参数:
+            group_id: 群组id
+            task: 任务模块
+            is_superuser: 是否为超级用户
+            platform: 平台
+        """
+        group, _ = await cls.get_or_create(
+            group_id=group_id, defaults={"platform": platform}
+        )
+        if is_superuser:
+            if f"<{task}," in group.superuser_block_task:
+                group.superuser_block_task = group.superuser_block_task.replace(
+                    f"<{task},", ""
+                )
+        elif f"<{task}," in group.block_task:
+            group.block_task = group.block_task.replace(f"<{task},", "")
+        await group.save(update_fields=["block_task", "superuser_block_task"])
+
     @classmethod
     def _run_script(cls):
-        return []
+        return [
+            "ALTER TABLE group_console ADD superuser_block_plugin"
+            " character varying(255) NOT NULL DEFAULT '';",
+            "ALTER TABLE group_console ADD superuser_block_task"
+            " character varying(255) NOT NULL DEFAULT '';",
+        ]