add plugin agent management tools

app/db/models/plugindata.py

@@ -1,7 +1,14 @@
-from sqlalchemy import Column, String, JSON
+from sqlalchemy import Column, String, JSON, select
+from sqlalchemy.ext.asyncio import AsyncSession
 from sqlalchemy.orm import Session
 
-from app.db import db_query, db_update, get_id_column, Base
+from app.db import (
+    db_query,
+    db_update,
+    async_db_query,
+    get_id_column,
+    Base,
+)
 
 
 class PluginData(Base):
@@ -18,11 +25,27 @@ class PluginData(Base):
     def get_plugin_data(cls, db: Session, plugin_id: str):
         return db.query(cls).filter(cls.plugin_id == plugin_id).all()
 
+    @classmethod
+    @async_db_query
+    async def async_get_plugin_data(cls, db: AsyncSession, plugin_id: str):
+        result = await db.execute(select(cls).where(cls.plugin_id == plugin_id))
+        return result.scalars().all()
+
     @classmethod
     @db_query
     def get_plugin_data_by_key(cls, db: Session, plugin_id: str, key: str):
         return db.query(cls).filter(cls.plugin_id == plugin_id, cls.key == key).first()
 
+    @classmethod
+    @async_db_query
+    async def async_get_plugin_data_by_key(
+        cls, db: AsyncSession, plugin_id: str, key: str
+    ):
+        result = await db.execute(
+            select(cls).where(cls.plugin_id == plugin_id, cls.key == key)
+        )
+        return result.scalar_one_or_none()
+
     @classmethod
     @db_update
     def del_plugin_data_by_key(cls, db: Session, plugin_id: str, key: str):
@@ -37,3 +60,11 @@ class PluginData(Base):
     @db_query
     def get_plugin_data_by_plugin_id(cls, db: Session, plugin_id: str):
         return db.query(cls).filter(cls.plugin_id == plugin_id).all()
+
+    @classmethod
+    @async_db_query
+    async def async_get_plugin_data_by_plugin_id(
+        cls, db: AsyncSession, plugin_id: str
+    ):
+        result = await db.execute(select(cls).where(cls.plugin_id == plugin_id))
+        return result.scalars().all()

app/db/plugindata_oper.py

@@ -38,6 +38,21 @@ class PluginDataOper(DbOper):
         else:
             return PluginData.get_plugin_data(self._db, plugin_id)
 
+    async def async_get_data(self, plugin_id: str, key: Optional[str] = None) -> Any:
+        """
+        异步获取插件数据。
+        :param plugin_id: 插件id
+        :param key: 数据key
+        """
+        if key:
+            data = await PluginData.async_get_plugin_data_by_key(
+                self._db, plugin_id, key
+            )
+            if not data:
+                return None
+            return data.value
+        return await PluginData.async_get_plugin_data(self._db, plugin_id)
+
     def del_data(self, plugin_id: str, key: Optional[str] = None) -> Any:
         """
         删除插件数据
@@ -61,3 +76,10 @@ class PluginDataOper(DbOper):
         :param plugin_id: 插件id
         """
         return PluginData.get_plugin_data_by_plugin_id(self._db, plugin_id)
+
+    async def async_get_data_all(self, plugin_id: str) -> Any:
+        """
+        异步获取插件所有数据。
+        :param plugin_id: 插件id
+        """
+        return await PluginData.async_get_plugin_data_by_plugin_id(self._db, plugin_id)