From 02c04f7773b561134fbc6fb3b6e54be62bd523d6 Mon Sep 17 00:00:00 2001
From: BalconyJH <balconyjh@gmail.com>
Date: Thu, 31 Oct 2024 20:22:26 +0800
Subject: [PATCH] :sparkles: Add get_plugins and get_tasks methods.

---
 zhenxun/models/bot_console.py | 86 ++++++++++++++++++++++++++++++++---
 1 file changed, 80 insertions(+), 6 deletions(-)

diff --git a/zhenxun/models/bot_console.py b/zhenxun/models/bot_console.py
index 40328b94..a39d6cad 100644
--- a/zhenxun/models/bot_console.py
+++ b/zhenxun/models/bot_console.py
@@ -1,4 +1,4 @@
-from typing import NoReturn, overload
+from typing import overload
 
 from tortoise import fields
 
@@ -61,6 +61,80 @@ class BotConsole(Model):
         result = await cls.get_or_none(bot_id=bot_id)
         return result.status if result else False
 
+    @overload
+    @classmethod
+    async def get_tasks(cls) -> list[tuple[str, str]]: ...
+
+    @overload
+    @classmethod
+    async def get_tasks(cls, bot_id: str) -> str: ...
+
+    @overload
+    @classmethod
+    async def get_tasks(cls, *, status: bool) -> list[tuple[str, str]]: ...
+
+    @overload
+    @classmethod
+    async def get_tasks(cls, bot_id: str, status: bool = True) -> str: ...
+
+    @classmethod
+    async def get_tasks(cls, bot_id: str | None = None, status: bool | None = True):
+        """
+        获取bot被动技能
+
+        Args:
+            bot_id (str | None, optional): bot_id. Defaults to None.
+            status (bool | None, optional): 被动状态. Defaults to True.
+
+        Returns:
+            list[tuple[str, str]] | str: 被动技能
+        """
+        if not bot_id:
+            task_field = "available_tasks" if status else "block_task"
+            return await cls.all().values_list("bot_id", task_field)
+
+        result = await cls.get_or_none(bot_id=bot_id)
+        if result:
+            return result.available_tasks if status else result.block_task
+        return ""
+
+    @overload
+    @classmethod
+    async def get_plugins(cls) -> list[tuple[str, str]]: ...
+
+    @overload
+    @classmethod
+    async def get_plugins(cls, bot_id: str) -> str: ...
+
+    @overload
+    @classmethod
+    async def get_plugins(cls, *, status: bool) -> list[tuple[str, str]]: ...
+
+    @overload
+    @classmethod
+    async def get_plugins(cls, bot_id: str, status: bool = True) -> str: ...
+
+    @classmethod
+    async def get_plugins(cls, bot_id: str | None = None, status: bool = True):
+        """
+        获取bot插件
+
+        Args:
+            bot_id (str | None, optional): bot_id. Defaults to None.
+            status (bool, optional): 插件状态. Defaults to True.
+
+        Returns:
+            list[tuple[str, str]] | str: 插件
+        """
+        if not bot_id:
+            plugin_field = "available_plugins" if status else "block_plugin"
+            return await cls.all().values_list("bot_id", plugin_field)
+
+        result = await cls.get_or_none(bot_id=bot_id)
+        if result:
+            return result.available_plugins if status else result.block_plugin
+        return ""
+
     @classmethod
     async def set_bot_status(cls, status: bool, bot_id: str | None = None) -> None:
         """
@@ -104,7 +178,7 @@ class BotConsole(Model):
         block_field: str,
         data: str,
         to_block: bool,
-    ) -> NoReturn:
+    ) -> None:
         """
         在 available_field 和 block_field 之间移动指定的 data
 
@@ -148,7 +222,7 @@ class BotConsole(Model):
         await bot_data.save(update_fields=[available_field, block_field])
 
     @classmethod
-    async def set_block_plugin(cls, bot_id: str | None, module: str) -> NoReturn:
+    async def set_block_plugin(cls, bot_id: str | None, module: str) -> None:
         """
         禁用插件
 
@@ -172,7 +246,7 @@ class BotConsole(Model):
                 )
 
     @classmethod
-    async def set_unblock_plugin(cls, bot_id: str | None, module: str) -> NoReturn:
+    async def set_unblock_plugin(cls, bot_id: str | None, module: str) -> None:
         """
         启用插件
 
@@ -196,7 +270,7 @@ class BotConsole(Model):
                 )
 
     @classmethod
-    async def set_block_task(cls, bot_id: str | None, module: str) -> NoReturn:
+    async def set_block_task(cls, bot_id: str | None, module: str) -> None:
         """
         禁用被动技能
 
@@ -216,7 +290,7 @@ class BotConsole(Model):
                 )
 
     @classmethod
-    async def set_unblock_task(cls, bot_id: str | None, module: str) -> NoReturn:
+    async def set_unblock_task(cls, bot_id: str | None, module: str) -> None:
         """
         启用被动技能