mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-29 08:39:59 +08:00
重构webui主页与好友/群组api
This commit is contained in:
+105
-105
@@ -1,115 +1,115 @@
|
||||
from typing import Optional
|
||||
# from typing import Optional
|
||||
|
||||
import cattrs
|
||||
from fastapi import APIRouter
|
||||
# import cattrs
|
||||
# from fastapi import APIRouter
|
||||
|
||||
from configs.config import Config
|
||||
from services.log import logger
|
||||
from utils.manager import plugin_data_manager, plugins2settings_manager, plugins_manager
|
||||
from utils.manager.models import PluginData, PluginType
|
||||
# from configs.config import Config
|
||||
# from services.log import logger
|
||||
# from utils.manager import plugin_data_manager, plugins2settings_manager, plugins_manager
|
||||
# from utils.manager.models import PluginData, PluginType
|
||||
|
||||
from ..config import *
|
||||
from ..models.model import Plugin, PluginConfig, Result
|
||||
from ..models.params import UpdateConfig, UpdatePlugin
|
||||
from ..utils import authentication
|
||||
# from ..config import *
|
||||
# from ..base_model import Plugin, PluginConfig, Result
|
||||
# from ..models.params import UpdateConfig, UpdatePlugin
|
||||
# from ..utils import authentication
|
||||
|
||||
router = APIRouter()
|
||||
# router = APIRouter()
|
||||
|
||||
|
||||
@router.get("/get_plugins", dependencies=[authentication()])
|
||||
def _(
|
||||
plugin_type: PluginType,
|
||||
) -> Result:
|
||||
"""
|
||||
获取插件列表
|
||||
:param plugin_type: 类型 normal, superuser, hidden, admin
|
||||
"""
|
||||
try:
|
||||
plugin_list = []
|
||||
for module in plugin_data_manager.keys():
|
||||
plugin_data: Optional[PluginData] = plugin_data_manager[module]
|
||||
if plugin_data and plugin_data.plugin_type == plugin_type:
|
||||
plugin_config = None
|
||||
if plugin_data.plugin_configs:
|
||||
plugin_config = {}
|
||||
for key in plugin_data.plugin_configs:
|
||||
plugin_config[key] = PluginConfig(
|
||||
key=key,
|
||||
module=module,
|
||||
has_type=bool(plugin_data.plugin_configs[key].type),
|
||||
**plugin_data.plugin_configs[key].dict(),
|
||||
)
|
||||
plugin_list.append(
|
||||
Plugin(
|
||||
model=module,
|
||||
plugin_settings=plugin_data.plugin_setting,
|
||||
plugin_manager=plugin_data.plugin_status,
|
||||
plugin_config=plugin_config,
|
||||
cd_limit=plugin_data.plugin_cd,
|
||||
block_limit=plugin_data.plugin_block,
|
||||
count_limit=plugin_data.plugin_count,
|
||||
)
|
||||
)
|
||||
except Exception as e:
|
||||
logger.error("调用API错误", "/get_plugins", e=e)
|
||||
return Result.fail(f"{type(e)}: {e}")
|
||||
return Result.ok(plugin_list, "拿到了新鲜出炉的数据!")
|
||||
# @router.get("/get_plugins", dependencies=[authentication()])
|
||||
# def _(
|
||||
# plugin_type: PluginType,
|
||||
# ) -> Result:
|
||||
# """
|
||||
# 获取插件列表
|
||||
# :param plugin_type: 类型 normal, superuser, hidden, admin
|
||||
# """
|
||||
# try:
|
||||
# plugin_list = []
|
||||
# for module in plugin_data_manager.keys():
|
||||
# plugin_data: Optional[PluginData] = plugin_data_manager[module]
|
||||
# if plugin_data and plugin_data.plugin_type == plugin_type:
|
||||
# plugin_config = None
|
||||
# if plugin_data.plugin_configs:
|
||||
# plugin_config = {}
|
||||
# for key in plugin_data.plugin_configs:
|
||||
# plugin_config[key] = PluginConfig(
|
||||
# key=key,
|
||||
# module=module,
|
||||
# has_type=bool(plugin_data.plugin_configs[key].type),
|
||||
# **plugin_data.plugin_configs[key].dict(),
|
||||
# )
|
||||
# plugin_list.append(
|
||||
# Plugin(
|
||||
# model=module,
|
||||
# plugin_settings=plugin_data.plugin_setting,
|
||||
# plugin_manager=plugin_data.plugin_status,
|
||||
# plugin_config=plugin_config,
|
||||
# cd_limit=plugin_data.plugin_cd,
|
||||
# block_limit=plugin_data.plugin_block,
|
||||
# count_limit=plugin_data.plugin_count,
|
||||
# )
|
||||
# )
|
||||
# except Exception as e:
|
||||
# logger.error("调用API错误", "/get_plugins", e=e)
|
||||
# return Result.fail(f"{type(e)}: {e}")
|
||||
# return Result.ok(plugin_list, "拿到了新鲜出炉的数据!")
|
||||
|
||||
|
||||
@router.post("/update_plugins", dependencies=[authentication()])
|
||||
def _(plugin: UpdatePlugin) -> Result:
|
||||
"""
|
||||
修改插件信息
|
||||
:param plugin: 插件内容
|
||||
"""
|
||||
try:
|
||||
module = plugin.module
|
||||
if p2s := plugins2settings_manager.get(module):
|
||||
p2s.default_status = plugin.default_status
|
||||
p2s.limit_superuser = plugin.limit_superuser
|
||||
p2s.cost_gold = plugin.cost_gold
|
||||
p2s.cmd = plugin.cmd
|
||||
p2s.level = plugin.group_level
|
||||
if pd := plugin_data_manager.get(module):
|
||||
menu_lin = None
|
||||
if len(pd.menu_type) > 1:
|
||||
menu_lin = pd.menu_type[1]
|
||||
if menu_lin is not None:
|
||||
pd.menu_type = (plugin.menu_type, menu_lin)
|
||||
else:
|
||||
pd.menu_type = (plugin.menu_type,)
|
||||
if pm := plugins_manager.get(module):
|
||||
if plugin.block_type:
|
||||
pm.block_type = plugin.block_type
|
||||
pm.status = False
|
||||
else:
|
||||
pm.block_type = None
|
||||
pm.status = True
|
||||
plugins2settings_manager.save()
|
||||
plugins_manager.save()
|
||||
except Exception as e:
|
||||
logger.error("调用API错误", "/update_plugins", e=e)
|
||||
return Result.fail(f"{type(e)}: {e}")
|
||||
return Result.ok(info="已经帮你写好啦!")
|
||||
# @router.post("/update_plugins", dependencies=[authentication()])
|
||||
# def _(plugin: UpdatePlugin) -> Result:
|
||||
# """
|
||||
# 修改插件信息
|
||||
# :param plugin: 插件内容
|
||||
# """
|
||||
# try:
|
||||
# module = plugin.module
|
||||
# if p2s := plugins2settings_manager.get(module):
|
||||
# p2s.default_status = plugin.default_status
|
||||
# p2s.limit_superuser = plugin.limit_superuser
|
||||
# p2s.cost_gold = plugin.cost_gold
|
||||
# p2s.cmd = plugin.cmd
|
||||
# p2s.level = plugin.group_level
|
||||
# if pd := plugin_data_manager.get(module):
|
||||
# menu_lin = None
|
||||
# if len(pd.menu_type) > 1:
|
||||
# menu_lin = pd.menu_type[1]
|
||||
# if menu_lin is not None:
|
||||
# pd.menu_type = (plugin.menu_type, menu_lin)
|
||||
# else:
|
||||
# pd.menu_type = (plugin.menu_type,)
|
||||
# if pm := plugins_manager.get(module):
|
||||
# if plugin.block_type:
|
||||
# pm.block_type = plugin.block_type
|
||||
# pm.status = False
|
||||
# else:
|
||||
# pm.block_type = None
|
||||
# pm.status = True
|
||||
# plugins2settings_manager.save()
|
||||
# plugins_manager.save()
|
||||
# except Exception as e:
|
||||
# logger.error("调用API错误", "/update_plugins", e=e)
|
||||
# return Result.fail(f"{type(e)}: {e}")
|
||||
# return Result.ok(info="已经帮你写好啦!")
|
||||
|
||||
|
||||
@router.post("/update_config", dependencies=[authentication()])
|
||||
def _(config_list: List[UpdateConfig]) -> Result:
|
||||
try:
|
||||
for config in config_list:
|
||||
if cg := Config.get(config.module):
|
||||
if c := cg.configs.get(config.key):
|
||||
if isinstance(c.value, (list, tuple)) or isinstance(
|
||||
c.default_value, (list, tuple)
|
||||
):
|
||||
value = config.value.split(",")
|
||||
else:
|
||||
value = config.value
|
||||
if c.type and value is not None:
|
||||
value = cattrs.structure(value, c.type)
|
||||
Config.set_config(config.module, config.key, value)
|
||||
except Exception as e:
|
||||
logger.error("调用API错误", "/update_config", e=e)
|
||||
return Result.fail(f"{type(e)}: {e}")
|
||||
Config.save(save_simple_data=True)
|
||||
return Result.ok(info="写入配置项了哦!")
|
||||
# @router.post("/update_config", dependencies=[authentication()])
|
||||
# def _(config_list: List[UpdateConfig]) -> Result:
|
||||
# try:
|
||||
# for config in config_list:
|
||||
# if cg := Config.get(config.module):
|
||||
# if c := cg.configs.get(config.key):
|
||||
# if isinstance(c.value, (list, tuple)) or isinstance(
|
||||
# c.default_value, (list, tuple)
|
||||
# ):
|
||||
# value = config.value.split(",")
|
||||
# else:
|
||||
# value = config.value
|
||||
# if c.type and value is not None:
|
||||
# value = cattrs.structure(value, c.type)
|
||||
# Config.set_config(config.module, config.key, value)
|
||||
# except Exception as e:
|
||||
# logger.error("调用API错误", "/update_config", e=e)
|
||||
# return Result.fail(f"{type(e)}: {e}")
|
||||
# Config.save(save_simple_data=True)
|
||||
# return Result.ok(info="写入配置项了哦!")
|
||||
|
||||
Reference in New Issue
Block a user