mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-14 13:42:56 +08:00
99f1388e23
* ✨ 添加全局优先级hook * ✨ 添加基础配置api * ✨ 添加数据库连接测试 * 💬 提示重启 * 🩹 填充过配置时友好提示 * 🐛 首次生成简易配置后自动加载 * ✨ 添加配置后重启接口 * ✨ 添加重启标志文件 * ✨ 添加重启脚本命令 * ✨ 添加重启系统限制 * ✨ 首次配置判断是否为win系统 * 🔥 移除bat * ✨ 添加关于菜单 * ✨ 支持整合包插件安装和添加整合包文档 * 🩹 检测数据库路径 * 🩹 修改数据库路径检测 * 🩹 修改数据库路径检测 * 🩹 修复路径注入 * 🎨 显示添加优先级 * 🐛 修改PriorityLifecycle字典类名称 * ⚡ 修复路径问题 * ⚡ 修复路径检测 * ✨ 新增路径验证功能，确保用户输入的路径安全并在项目根目录内 * ✨ 优化路径验证功能，增加对非法字符和路径长度的检查，确保用户输入的路径更加安全 * 🚨 auto fix by pre-commit hooks * ✨ 优化获取文件列表的代码格式 * 📝 修改README中webui示例图 * ✨ 更新PriorityLifecycle.on_startup装饰器 * ✨ 简化安装依赖的命令构建逻辑 * 🚨 auto fix by pre-commit hooks --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
45 lines
1.6 KiB
Python
45 lines
1.6 KiB
Python
from pathlib import Path
|
|
import shutil
|
|
import zipfile
|
|
|
|
from nonebot.utils import run_sync
|
|
|
|
from zhenxun.services.log import logger
|
|
from zhenxun.utils.github_utils import GithubUtils
|
|
from zhenxun.utils.http_utils import AsyncHttpx
|
|
|
|
from ..config import PUBLIC_PATH, TMP_PATH, WEBUI_DIST_GITHUB_URL
|
|
|
|
COMMAND_NAME = "WebUI资源管理"
|
|
|
|
|
|
async def update_webui_assets():
|
|
webui_assets_path = TMP_PATH / "webui_assets.zip"
|
|
download_url = await GithubUtils.parse_github_url(
|
|
WEBUI_DIST_GITHUB_URL
|
|
).get_archive_download_urls()
|
|
logger.info("开始下载 webui_assets 资源...", COMMAND_NAME)
|
|
if await AsyncHttpx.download_file(
|
|
download_url, webui_assets_path, follow_redirects=True
|
|
):
|
|
logger.info("下载 webui_assets 成功...", COMMAND_NAME)
|
|
return await _file_handle(webui_assets_path)
|
|
raise Exception("下载 webui_assets 失败", COMMAND_NAME)
|
|
|
|
|
|
@run_sync
|
|
def _file_handle(webui_assets_path: Path):
|
|
logger.debug("开始解压 webui_assets...", COMMAND_NAME)
|
|
if webui_assets_path.exists():
|
|
tf = zipfile.ZipFile(webui_assets_path)
|
|
tf.extractall(TMP_PATH)
|
|
logger.debug("解压 webui_assets 成功...", COMMAND_NAME)
|
|
else:
|
|
raise Exception("解压 webui_assets 失败，文件不存在...", COMMAND_NAME)
|
|
download_file_path = next(f for f in TMP_PATH.iterdir() if f.is_dir())
|
|
shutil.rmtree(PUBLIC_PATH, ignore_errors=True)
|
|
shutil.copytree(download_file_path / "dist", PUBLIC_PATH, dirs_exist_ok=True)
|
|
logger.debug("复制 webui_assets 成功...", COMMAND_NAME)
|
|
shutil.rmtree(TMP_PATH, ignore_errors=True)
|
|
return [x.name for x in PUBLIC_PATH.iterdir() if x.is_dir()]
|