mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2026-09-28 16:20:56 +08:00
* fix(plugin-store-downlaod): 修复插件商店下载二进制文件错误 * fix(plugin_store): 修复插件商店中仓库类型判断逻辑错误 当处理外部插件时,修正了仓库类型的判断条件,确保正确的仓库类型被设置。 同时修复了目录文件列表获取时传递错误的仓库类型参数问题。 * feat(repo_utils): 更新文件管理器返回类型并改进仓库URL处理 - 将 RepoFileManager 的文件内容返回类型从 bytes 改为 str, 统一字符串类型的返回值格式 - 支持在 repo_url 中直接包含 /tree/<branch> 路径, 自动解析分支信息并正确处理仓库地址 - 改进 GitHub 和阿里云仓库的优先级顺序, 现在优先尝试阿里云然后是 GitHub - 修复 clone_url 处理逻辑,确保正确的 .git 后缀添加 - 移除未使用的 Sequence 类型导入,统一使用 list[str] 类型注解 * refactor(plugin_store): 移除StoreManager中不必要的repo_type参数 移除了data_source.py文件中StoreManager类的list_directory_files方法调用时 多余的repo_type=RepoType.GITHUB参数 * feat(plugin_store): 优化插件依赖文件下载和安装逻辑 - 移动依赖文件下载逻辑到下载阶段,避免重复下载 - 使用随机数生成唯一的临时依赖文件名以避免冲突 - 添加ignore_error参数来处理依赖文件下载失败的情况 - 统一依赖文件路径变量命名,提高代码可读性 - 确保requirement.txt和requirements.txt都能正确安装 * Add repo_type parameter to list_directory_files call
618 lines
22 KiB
Python
618 lines
22 KiB
Python
import os
|
||
from pathlib import Path
|
||
import random
|
||
import shutil
|
||
|
||
import ujson as json
|
||
|
||
from zhenxun.builtin_plugins.plugin_store.models import StorePluginInfo
|
||
from zhenxun.configs.path_config import TEMP_PATH
|
||
from zhenxun.models.plugin_info import PluginInfo
|
||
from zhenxun.services.cache.bounded_ttl import BoundedTTLCache
|
||
from zhenxun.services.log import logger
|
||
from zhenxun.services.plugin_init import PluginInitManager
|
||
from zhenxun.utils.enum import PluginType
|
||
from zhenxun.utils.image_utils import BuildImage, ImageTemplate, RowStyle
|
||
from zhenxun.utils.manager.virtual_env_package_manager import VirtualEnvPackageManager
|
||
from zhenxun.utils.repo_utils import RepoFileManager
|
||
from zhenxun.utils.repo_utils.models import RepoFileInfo, RepoType
|
||
from zhenxun.utils.utils import is_number, win_on_rm_error
|
||
|
||
from .config import (
|
||
BASE_PATH,
|
||
DEFAULT_GITHUB_URL,
|
||
EXTRA_GITHUB_URL,
|
||
LOG_COMMAND,
|
||
)
|
||
from .exceptions import PluginStoreException
|
||
|
||
_PLUGIN_STORE_DATA_CACHE = BoundedTTLCache[
|
||
str, tuple[list[StorePluginInfo], list[StorePluginInfo]]
|
||
](
|
||
"PLUGIN_STORE_DATA",
|
||
ttl_seconds=60,
|
||
max_items=1,
|
||
)
|
||
|
||
|
||
def row_style(column: str, text: str) -> RowStyle:
|
||
"""被动技能文本风格
|
||
|
||
参数:
|
||
column: 表头
|
||
text: 文本内容
|
||
|
||
返回:
|
||
RowStyle: RowStyle
|
||
"""
|
||
style = RowStyle()
|
||
if column == "-" and text == "已安装":
|
||
style.font_color = "#67C23A"
|
||
return style
|
||
|
||
|
||
class StoreManager:
|
||
@classmethod
|
||
def _resolve_local_plugin_path(
|
||
cls, plugin_info: StorePluginInfo, *, is_external: bool
|
||
) -> Path:
|
||
"""将商店插件信息映射到本地插件文件/目录路径。"""
|
||
plugin_name = plugin_info.module
|
||
|
||
if plugin_info.is_dir:
|
||
return BASE_PATH / "plugins" / plugin_name
|
||
|
||
return BASE_PATH / "plugins" / f"{plugin_name}.py"
|
||
|
||
@classmethod
|
||
async def get_data(cls) -> tuple[list[StorePluginInfo], list[StorePluginInfo]]:
|
||
"""获取插件信息数据
|
||
|
||
返回:
|
||
tuple[list[StorePluginInfo], list[StorePluginInfo]]:
|
||
原生插件信息数据,第三方插件信息数据
|
||
"""
|
||
cache_key = "plugins_json"
|
||
if cached_data := await _PLUGIN_STORE_DATA_CACHE.get(cache_key):
|
||
return cached_data
|
||
|
||
plugins = await RepoFileManager.get_text_content(
|
||
DEFAULT_GITHUB_URL, "plugins.json"
|
||
)
|
||
extra_plugins = await RepoFileManager.get_text_content(
|
||
EXTRA_GITHUB_URL, "plugins.json", "index"
|
||
)
|
||
result = (
|
||
[StorePluginInfo(**plugin) for plugin in json.loads(plugins)],
|
||
[StorePluginInfo(**plugin) for plugin in json.loads(extra_plugins)],
|
||
)
|
||
await _PLUGIN_STORE_DATA_CACHE.set(cache_key, result)
|
||
return result
|
||
|
||
@classmethod
|
||
def version_check(cls, plugin_info: StorePluginInfo, suc_plugin: dict[str, str]):
|
||
"""版本检查
|
||
|
||
参数:
|
||
plugin_info: StorePluginInfo
|
||
suc_plugin: 模块名: 版本号
|
||
|
||
返回:
|
||
str: 版本号
|
||
"""
|
||
module = plugin_info.module
|
||
if suc_plugin.get(module) and not cls.check_version_is_new(
|
||
plugin_info, suc_plugin
|
||
):
|
||
return f"{suc_plugin[module]} (有更新->{plugin_info.version})"
|
||
return plugin_info.version
|
||
|
||
@classmethod
|
||
def check_version_is_new(
|
||
cls, plugin_info: StorePluginInfo, suc_plugin: dict[str, str]
|
||
):
|
||
"""检查版本是否有更新
|
||
|
||
参数:
|
||
plugin_info: StorePluginInfo
|
||
suc_plugin: 模块名: 版本号
|
||
|
||
返回:
|
||
bool: 是否有更新
|
||
"""
|
||
module = plugin_info.module
|
||
return suc_plugin.get(module) and plugin_info.version == suc_plugin[module]
|
||
|
||
@classmethod
|
||
async def get_installed_plugins(cls) -> dict[str, str]:
|
||
"""获取已安装插件的模块与版本。
|
||
|
||
返回:
|
||
dict[str, str]: 模块 -> 版本
|
||
"""
|
||
db_plugin_list = await PluginInfo.get_plugins_values_list(
|
||
"module", "version", load_status=True, filter_parent=False
|
||
)
|
||
return {p[0]: (p[1] or "0.1") for p in db_plugin_list}
|
||
|
||
@classmethod
|
||
async def get_plugins_info(cls) -> list[BuildImage] | str:
|
||
"""插件列表
|
||
|
||
返回:
|
||
BuildImage | str: 返回消息
|
||
"""
|
||
plugin_list, extra_plugin_list = await cls.get_data()
|
||
column_name = ["-", "ID", "名称", "简介", "作者", "版本", "类型"]
|
||
suc_plugin = await cls.get_installed_plugins()
|
||
index = 0
|
||
data_list = []
|
||
extra_data_list = []
|
||
for plugin_info in plugin_list:
|
||
data_list.append(
|
||
[
|
||
"已安装" if plugin_info.module in suc_plugin else "",
|
||
index,
|
||
plugin_info.name,
|
||
plugin_info.description,
|
||
plugin_info.author,
|
||
cls.version_check(plugin_info, suc_plugin),
|
||
plugin_info.plugin_type_name,
|
||
]
|
||
)
|
||
index += 1
|
||
for plugin_info in extra_plugin_list:
|
||
extra_data_list.append(
|
||
[
|
||
"已安装" if plugin_info.module in suc_plugin else "",
|
||
index,
|
||
plugin_info.name,
|
||
plugin_info.description,
|
||
plugin_info.author,
|
||
cls.version_check(plugin_info, suc_plugin),
|
||
plugin_info.plugin_type_name,
|
||
]
|
||
)
|
||
index += 1
|
||
return [
|
||
await ImageTemplate.table_page(
|
||
"原生插件列表",
|
||
"通过添加/移除插件 ID 来管理插件",
|
||
column_name,
|
||
data_list,
|
||
text_style=row_style,
|
||
),
|
||
await ImageTemplate.table_page(
|
||
"第三方插件列表",
|
||
"通过添加/移除插件 ID 来管理插件",
|
||
column_name,
|
||
extra_data_list,
|
||
text_style=row_style,
|
||
),
|
||
]
|
||
|
||
@classmethod
|
||
async def get_plugin_by_value(
|
||
cls,
|
||
index_or_module: str,
|
||
is_update: bool = False,
|
||
is_remove: bool = False,
|
||
) -> tuple[StorePluginInfo, bool]:
|
||
"""获取插件信息
|
||
|
||
参数:
|
||
index_or_module: 插件索引或模块名
|
||
is_update: 是否是更新插件
|
||
is_remove: 是否是移除插件
|
||
|
||
异常:
|
||
PluginStoreException: 插件不存在
|
||
PluginStoreException: 插件已安装
|
||
|
||
返回:
|
||
StorePluginInfo: 插件信息
|
||
bool: 是否是外部插件
|
||
"""
|
||
plugin_list: list[StorePluginInfo]
|
||
extra_plugin_list: list[StorePluginInfo]
|
||
plugin_list, extra_plugin_list = await cls.get_data()
|
||
plugin_info = None
|
||
is_external = False
|
||
try:
|
||
plugin_key = await cls._resolve_plugin_key(index_or_module)
|
||
except PluginStoreException:
|
||
if not is_remove:
|
||
raise
|
||
# 移除时插件可能已不在商店列表,回退到数据库查找
|
||
plugin_key = None
|
||
|
||
if plugin_key is not None:
|
||
for p in plugin_list:
|
||
if p.module == plugin_key:
|
||
is_external = False
|
||
plugin_info = p
|
||
break
|
||
for p in extra_plugin_list:
|
||
if p.module == plugin_key:
|
||
is_external = True
|
||
plugin_info = p
|
||
break
|
||
|
||
installed_modules = set((await cls.get_installed_plugins()).keys())
|
||
|
||
if is_remove:
|
||
# 商店列表中找不到时,从数据库构建最小插件信息
|
||
if not plugin_info:
|
||
db_obj = await PluginInfo.get_plugin(
|
||
module=index_or_module, plugin_type=PluginType.PARENT
|
||
) or await PluginInfo.get_plugin(module=index_or_module)
|
||
if db_obj is None:
|
||
db_obj = await PluginInfo.get_or_none(name=index_or_module)
|
||
if db_obj is None:
|
||
raise PluginStoreException("插件 Module / 名称 不存在...")
|
||
_mp = db_obj.module_path
|
||
_path = BASE_PATH.parent / Path(_mp.replace(".", os.sep))
|
||
plugin_info = StorePluginInfo(
|
||
name=db_obj.name,
|
||
module=db_obj.module,
|
||
module_path=_mp,
|
||
description="",
|
||
usage="",
|
||
author=db_obj.author or "",
|
||
version=db_obj.version or "0.0.0",
|
||
plugin_type=db_obj.plugin_type or PluginType.NORMAL,
|
||
is_dir=_path.is_dir(),
|
||
)
|
||
is_external = True
|
||
if plugin_info.module not in installed_modules:
|
||
raise PluginStoreException(f"插件 {plugin_info.name} 未安装,无法移除")
|
||
if plugin_obj := await PluginInfo.get_plugin(
|
||
module=plugin_info.module,
|
||
plugin_type=PluginType.PARENT,
|
||
load_status=True,
|
||
):
|
||
plugin_info.module_path = plugin_obj.module_path
|
||
elif plugin_obj := await PluginInfo.get_plugin(
|
||
module=plugin_info.module, load_status=True
|
||
):
|
||
plugin_info.module_path = plugin_obj.module_path
|
||
return plugin_info, is_external
|
||
|
||
if not plugin_info:
|
||
raise PluginStoreException(f"插件不存在: {plugin_key}")
|
||
|
||
if is_update:
|
||
if plugin_info.module not in installed_modules:
|
||
raise PluginStoreException(f"插件 {plugin_info.name} 未安装,无法更新")
|
||
return plugin_info, is_external
|
||
|
||
if plugin_info.module in installed_modules:
|
||
raise PluginStoreException(f"插件 {plugin_info.name} 已安装,无需重复安装")
|
||
|
||
return plugin_info, is_external
|
||
|
||
@classmethod
|
||
async def add_plugin(cls, index_or_module: str, source: str | None = None) -> str:
|
||
"""添加插件
|
||
|
||
参数:
|
||
plugin_id: 插件id或模块名
|
||
|
||
返回:
|
||
str: 返回消息
|
||
"""
|
||
plugin_info, is_external = await cls.get_plugin_by_value(index_or_module)
|
||
if plugin_info.github_url is None:
|
||
plugin_info.github_url = DEFAULT_GITHUB_URL
|
||
version_split = plugin_info.version.split("-")
|
||
if len(version_split) > 1:
|
||
github_url_split = plugin_info.github_url.split("/tree/")
|
||
plugin_info.github_url = f"{github_url_split[0]}/tree/{version_split[1]}"
|
||
logger.info(f"正在安装插件 {plugin_info.name}...", LOG_COMMAND)
|
||
await cls.install_plugin_with_repo(
|
||
plugin_info,
|
||
is_external,
|
||
source,
|
||
)
|
||
return (
|
||
f"插件 {plugin_info.name} 安装完成\n"
|
||
"- 已下载插件文件\n"
|
||
"- 已处理依赖文件\n"
|
||
"- 重启后生效"
|
||
)
|
||
|
||
@classmethod
|
||
async def install_plugin_with_repo(
|
||
cls,
|
||
plugin_info: StorePluginInfo,
|
||
is_external: bool = False,
|
||
source: str | None = None,
|
||
branch: str = "main",
|
||
):
|
||
"""安装插件
|
||
|
||
参数:
|
||
plugin_info: 插件信息
|
||
is_external: 是否是外部仓库
|
||
source: 源
|
||
"""
|
||
repo_type = RepoType.GITHUB if is_external else None
|
||
if not is_external:
|
||
repo_type = RepoType.ALIYUN
|
||
elif (source is None and plugin_info.ali_url) or source == "ali":
|
||
repo_type = RepoType.ALIYUN
|
||
elif source == "git":
|
||
repo_type = RepoType.GITHUB
|
||
module_path = plugin_info.module_path
|
||
is_dir = plugin_info.is_dir
|
||
github_url = plugin_info.github_url
|
||
assert github_url
|
||
replace_module_path = module_path.replace(".", "/").lstrip("/")
|
||
plugin_module = plugin_info.module
|
||
if is_dir:
|
||
files = await RepoFileManager.list_directory_files(
|
||
github_url, replace_module_path, branch, repo_type=repo_type
|
||
)
|
||
else:
|
||
files = [RepoFileInfo(path=f"{replace_module_path}.py", is_dir=False)]
|
||
if is_dir:
|
||
target_dir = BASE_PATH / "plugins" / plugin_module
|
||
else:
|
||
target_dir = BASE_PATH / "plugins"
|
||
files = [file for file in files if not file.is_dir]
|
||
download_files: list[tuple[str, Path]] = []
|
||
|
||
for file in files:
|
||
src_path = file.path
|
||
if is_dir:
|
||
dst_path = target_dir / Path(src_path).relative_to(replace_module_path)
|
||
else:
|
||
dst_path = target_dir / f"{plugin_module}.py"
|
||
|
||
download_files.append((src_path, dst_path))
|
||
rand = random.randint(1, 10000)
|
||
requirement_path_ = TEMP_PATH / f"plugin_store_{rand}_req.txt"
|
||
requirements_path_ = TEMP_PATH / f"plugin_store_{rand}_reqs.txt"
|
||
download_files.append(("requirement.txt", requirement_path_))
|
||
download_files.append(("requirements.txt", requirements_path_))
|
||
result = await RepoFileManager.download_files(
|
||
github_url,
|
||
download_files,
|
||
branch,
|
||
repo_type=repo_type,
|
||
ignore_error=True,
|
||
)
|
||
if not result.success:
|
||
raise PluginStoreException(result.error_message)
|
||
|
||
requirement_paths = [
|
||
file
|
||
for file in files
|
||
if file.path.endswith("requirement.txt")
|
||
or file.path.endswith("requirements.txt")
|
||
]
|
||
|
||
is_install_req = False
|
||
for requirement_path in requirement_paths:
|
||
requirement_file = target_dir / Path(requirement_path.path).relative_to(
|
||
replace_module_path
|
||
)
|
||
if requirement_file.exists():
|
||
is_install_req = True
|
||
await VirtualEnvPackageManager.install_requirement(requirement_file)
|
||
|
||
if not is_install_req:
|
||
if requirement_path_.exists():
|
||
logger.info(
|
||
f"开始安装插件 {module_path} 依赖文件: {requirement_path_}",
|
||
LOG_COMMAND,
|
||
)
|
||
await VirtualEnvPackageManager.install_requirement(requirement_path_)
|
||
if requirements_path_.exists():
|
||
logger.info(
|
||
f"开始安装插件 {module_path} 依赖文件: {requirements_path_}",
|
||
LOG_COMMAND,
|
||
)
|
||
await VirtualEnvPackageManager.install_requirement(requirements_path_)
|
||
|
||
@classmethod
|
||
async def remove_plugin(cls, index_or_module: str) -> str:
|
||
"""移除插件
|
||
|
||
参数:
|
||
index_or_module: 插件id或模块名
|
||
|
||
返回:
|
||
str: 返回消息
|
||
"""
|
||
plugin_info, _ = await cls.get_plugin_by_value(index_or_module, is_remove=True)
|
||
is_external = not plugin_info.module_path.startswith("zhenxun.")
|
||
path = cls._resolve_local_plugin_path(plugin_info, is_external=is_external)
|
||
if not path.exists():
|
||
return f"插件 {plugin_info.name} 不存在..."
|
||
logger.debug(f"尝试移除插件 {plugin_info.name} 文件: {path}", LOG_COMMAND)
|
||
if plugin_info.is_dir:
|
||
# 处理 Windows 下 .git 等目录内只读文件导致的 WinError 5
|
||
shutil.rmtree(path, onerror=win_on_rm_error)
|
||
else:
|
||
path.unlink()
|
||
await PluginInitManager.remove(plugin_info.module_path)
|
||
plugin_records = await PluginInfo.get_plugins(
|
||
load_status=None,
|
||
filter_parent=False,
|
||
module_path=plugin_info.module_path,
|
||
)
|
||
for plugin_record in plugin_records:
|
||
await plugin_record.delete()
|
||
return f"插件 {plugin_info.name} 移除成功! 重启后生效"
|
||
|
||
@classmethod
|
||
async def search_plugin(cls, plugin_name_or_author: str) -> BuildImage | str:
|
||
"""搜索插件
|
||
|
||
参数:
|
||
plugin_name_or_author: 插件名称或作者
|
||
|
||
返回:
|
||
BuildImage | str: 返回消息
|
||
"""
|
||
plugin_list, extra_plugin_list = await cls.get_data()
|
||
all_plugin_list = plugin_list + extra_plugin_list
|
||
suc_plugin = await cls.get_installed_plugins()
|
||
filtered_data = [
|
||
(id, plugin_info)
|
||
for id, plugin_info in enumerate(all_plugin_list)
|
||
if plugin_name_or_author.lower() in plugin_info.name.lower()
|
||
or plugin_name_or_author.lower() in plugin_info.author.lower()
|
||
]
|
||
|
||
data_list = [
|
||
[
|
||
"已安装" if plugin_info.module in suc_plugin else "",
|
||
id,
|
||
plugin_info.name,
|
||
plugin_info.description,
|
||
plugin_info.author,
|
||
cls.version_check(plugin_info, suc_plugin),
|
||
plugin_info.plugin_type_name,
|
||
]
|
||
for id, plugin_info in filtered_data
|
||
]
|
||
if not data_list:
|
||
return "未找到相关插件..."
|
||
column_name = ["-", "ID", "名称", "简介", "作者", "版本", "类型"]
|
||
return await ImageTemplate.table_page(
|
||
"商店插件列表",
|
||
"通过添加/移除插件 ID 来管理插件",
|
||
column_name,
|
||
data_list,
|
||
text_style=row_style,
|
||
)
|
||
|
||
@classmethod
|
||
async def update_plugin(cls, index_or_module: str) -> str:
|
||
"""更新插件
|
||
|
||
参数:
|
||
index_or_module: 插件id
|
||
|
||
返回:
|
||
str: 返回消息
|
||
"""
|
||
plugin_info, is_external = await cls.get_plugin_by_value(index_or_module, True)
|
||
logger.info(f"尝试更新插件 {plugin_info.name}", LOG_COMMAND)
|
||
suc_plugin = await cls.get_installed_plugins()
|
||
logger.debug(f"当前插件列表: {suc_plugin}", LOG_COMMAND)
|
||
if cls.check_version_is_new(plugin_info, suc_plugin):
|
||
return f"插件 {plugin_info.name} 已是最新版本"
|
||
if plugin_info.github_url is None:
|
||
plugin_info.github_url = DEFAULT_GITHUB_URL
|
||
await cls.install_plugin_with_repo(
|
||
plugin_info,
|
||
is_external,
|
||
)
|
||
return f"插件 {plugin_info.name} 更新成功! 重启后生效"
|
||
|
||
@classmethod
|
||
async def update_all_plugin(cls) -> str:
|
||
"""更新插件
|
||
|
||
参数:
|
||
plugin_id: 插件id
|
||
|
||
返回:
|
||
str: 返回消息
|
||
"""
|
||
plugin_list, extra_plugin_list = await cls.get_data()
|
||
all_plugin_list = plugin_list + extra_plugin_list
|
||
plugin_name_list = [p.name for p in all_plugin_list]
|
||
update_failed_list = []
|
||
update_success_list = []
|
||
result = "--已更新{}个插件 {}个失败 {}个成功--"
|
||
logger.info(f"尝试更新全部插件 {plugin_name_list}", LOG_COMMAND)
|
||
suc_plugin = await cls.get_installed_plugins()
|
||
for plugin_info in all_plugin_list:
|
||
try:
|
||
if plugin_info.module not in suc_plugin:
|
||
logger.debug(
|
||
f"插件 {plugin_info.name}({plugin_info.module}) 未安装,跳过",
|
||
LOG_COMMAND,
|
||
)
|
||
continue
|
||
if cls.check_version_is_new(plugin_info, suc_plugin):
|
||
logger.debug(
|
||
f"插件 {plugin_info.name}({plugin_info.module}) "
|
||
"已是最新版本,跳过",
|
||
LOG_COMMAND,
|
||
)
|
||
continue
|
||
logger.info(
|
||
f"正在更新插件 {plugin_info.name}({plugin_info.module})",
|
||
LOG_COMMAND,
|
||
)
|
||
is_external = True
|
||
if plugin_info.github_url is None:
|
||
plugin_info.github_url = DEFAULT_GITHUB_URL
|
||
is_external = False
|
||
await cls.install_plugin_with_repo(
|
||
plugin_info,
|
||
is_external,
|
||
)
|
||
update_success_list.append(plugin_info.name)
|
||
except Exception as e:
|
||
logger.error(
|
||
f"更新插件 {plugin_info.name}({plugin_info.module}) 失败",
|
||
LOG_COMMAND,
|
||
e=e,
|
||
)
|
||
update_failed_list.append(plugin_info.name)
|
||
if not update_success_list and not update_failed_list:
|
||
return "全部插件已是最新版本"
|
||
if update_success_list:
|
||
result += "\n* 以下插件更新成功:\n\t- {}".format(
|
||
"\n\t- ".join(update_success_list)
|
||
)
|
||
if update_failed_list:
|
||
result += "\n* 以下插件更新失败:\n\t- {}".format(
|
||
"\n\t- ".join(update_failed_list)
|
||
)
|
||
return (
|
||
result.format(
|
||
len(update_success_list) + len(update_failed_list),
|
||
len(update_failed_list),
|
||
len(update_success_list),
|
||
)
|
||
+ "\n重启后生效"
|
||
)
|
||
|
||
@classmethod
|
||
async def _resolve_plugin_key(cls, plugin_id: str) -> str:
|
||
"""获取插件module
|
||
|
||
参数:
|
||
plugin_id: module,id或插件名称
|
||
|
||
异常:
|
||
PluginStoreException: 插件不存在
|
||
PluginStoreException: 插件不存在
|
||
|
||
返回:
|
||
str: 插件模块名
|
||
"""
|
||
plugin_list, extra_plugin_list = await cls.get_data()
|
||
all_plugin_list = plugin_list + extra_plugin_list
|
||
if is_number(plugin_id):
|
||
idx = int(plugin_id)
|
||
if idx < 0 or idx >= len(all_plugin_list):
|
||
raise PluginStoreException("插件ID不存在...")
|
||
return all_plugin_list[idx].module
|
||
elif isinstance(plugin_id, str):
|
||
if plugin_id in [v.module for v in all_plugin_list]:
|
||
return plugin_id
|
||
|
||
for plugin_info in all_plugin_list:
|
||
if plugin_info.name.lower() == plugin_id.lower():
|
||
return plugin_info.module
|
||
|
||
raise PluginStoreException("插件 Module / 名称 不存在...")
|