52 lines
1.4 KiB
Python
52 lines
1.4 KiB
Python
import json
|
|
from typing import Any
|
|
|
|
from app.db import DbOper
|
|
from app.db.models import Base
|
|
from app.db.models.plugin import PluginData
|
|
from app.utils.object import ObjectUtils
|
|
|
|
|
|
class PluginDataOper(DbOper):
|
|
"""
|
|
插件数据管理
|
|
"""
|
|
|
|
def save(self, plugin_id: str, key: str, value: Any) -> Base:
|
|
"""
|
|
保存插件数据
|
|
:param plugin_id: 插件id
|
|
:param key: 数据key
|
|
:param value: 数据值
|
|
"""
|
|
if ObjectUtils.is_obj(value):
|
|
value = json.dumps(value)
|
|
plugin = PluginData.get_plugin_data_by_key(self._db, plugin_id, key)
|
|
if plugin:
|
|
plugin.update(self._db, {
|
|
"value": value
|
|
})
|
|
return plugin
|
|
else:
|
|
plugin = PluginData(plugin_id=plugin_id, key=key, value=value)
|
|
return plugin.create(self._db)
|
|
|
|
def get_data(self, plugin_id: str, key: str) -> Any:
|
|
"""
|
|
获取插件数据
|
|
:param plugin_id: 插件id
|
|
:param key: 数据key
|
|
"""
|
|
data = PluginData.get_plugin_data_by_key(self._db, plugin_id, key)
|
|
if not data:
|
|
return None
|
|
if ObjectUtils.is_obj(data.value):
|
|
return json.loads(data.value)
|
|
return data.value
|
|
|
|
def truncate(self):
|
|
"""
|
|
清空转移记录
|
|
"""
|
|
PluginData.truncate(self._db)
|