mirror of
https://github.com/zhenxun-org/zhenxun_bot.git
synced 2025-12-15 14:22:55 +08:00
71 lines
1.4 KiB
Python
71 lines
1.4 KiB
Python
from strenum import StrEnum
|
|
from pydantic import BaseModel, validator
|
|
|
|
from zhenxun.utils.enum import PluginType
|
|
|
|
type2name: dict[str, str] = {
|
|
"NORMAL": "普通插件",
|
|
"ADMIN": "管理员插件",
|
|
"SUPERUSER": "超级用户插件",
|
|
"ADMIN_SUPERUSER": "管理员/超级用户插件",
|
|
"DEPENDANT": "依赖插件",
|
|
"HIDDEN": "其他插件",
|
|
}
|
|
|
|
|
|
class StorePluginInfo(BaseModel):
|
|
"""插件信息"""
|
|
|
|
module: str
|
|
module_path: str
|
|
description: str
|
|
usage: str
|
|
author: str
|
|
version: str
|
|
plugin_type: PluginType
|
|
is_dir: bool
|
|
github_url: str | None
|
|
|
|
@property
|
|
def plugin_type_name(self):
|
|
return type2name[self.plugin_type.value]
|
|
|
|
|
|
class RepoInfo(BaseModel):
|
|
"""仓库信息"""
|
|
|
|
owner: str
|
|
repo: str
|
|
branch: str | None
|
|
|
|
@validator("branch", pre=True, always=True)
|
|
def set_default_branch(cls, v):
|
|
return "main" if v is None else v
|
|
|
|
def get_download_url_with_path(self, path: str):
|
|
return f"https://raw.githubusercontent.com/{self.owner}/{self.repo}/{self.branch}/{path}"
|
|
|
|
|
|
class FileType(StrEnum):
|
|
"""文件类型"""
|
|
|
|
FILE = "file"
|
|
DIR = "directory"
|
|
|
|
|
|
class FileInfo(BaseModel):
|
|
"""文件信息"""
|
|
|
|
type: FileType
|
|
name: str
|
|
files: list["FileInfo"] | None
|
|
|
|
|
|
class JsdPackageInfo(BaseModel):
|
|
"""jsd包信息"""
|
|
|
|
type: str
|
|
name: str
|
|
version: str
|
|
files: list[FileInfo]
|