mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-09 07:25:25 +08:00
85 lines
3.1 KiB
Python
85 lines
3.1 KiB
Python
"""
|
|
根据 MoviePilot Wiki 清单生成发版快照中的插件市场默认值。
|
|
"""
|
|
|
|
import argparse
|
|
import ast
|
|
from pathlib import Path
|
|
from typing import Optional
|
|
|
|
from app.helper.market import extract_plugin_market_repos_from_wiki
|
|
|
|
|
|
OFFICIAL_PLUGIN_MARKET = "https://github.com/jxxghp/MoviePilot-Plugins"
|
|
|
|
|
|
def _parse_args(args: Optional[list[str]] = None) -> argparse.Namespace:
|
|
parser = argparse.ArgumentParser(description="生成插件市场发版默认值")
|
|
parser.add_argument("--wiki-file", type=Path, required=True)
|
|
parser.add_argument("--config-file", type=Path, required=True)
|
|
return parser.parse_args(args)
|
|
|
|
|
|
def _find_plugin_market_assignment(source: str) -> tuple[int, int, str]:
|
|
tree = ast.parse(source)
|
|
source_lines = source.splitlines(keepends=True)
|
|
for node in tree.body:
|
|
if not isinstance(node, ast.ClassDef) or node.name != "ConfigModel":
|
|
continue
|
|
for item in node.body:
|
|
if not isinstance(item, ast.AnnAssign):
|
|
continue
|
|
if not isinstance(item.target, ast.Name):
|
|
continue
|
|
if item.target.id != "PLUGIN_MARKET" or item.end_lineno is None:
|
|
continue
|
|
start = item.lineno - 1
|
|
indent_size = len(source_lines[start]) - len(source_lines[start].lstrip())
|
|
indent = source_lines[start][:indent_size]
|
|
return start, item.end_lineno, indent
|
|
raise ValueError("未在 ConfigModel 中找到 PLUGIN_MARKET 默认值")
|
|
|
|
|
|
def _format_plugin_market_assignment(repos: list[str], indent: str) -> str:
|
|
lines = [f"{indent}PLUGIN_MARKET: str = (\n"]
|
|
for index, repo in enumerate(repos):
|
|
suffix = "," if index < len(repos) - 1 else ""
|
|
lines.append(f'{indent} "{repo}{suffix}"\n')
|
|
lines.append(f"{indent})\n")
|
|
return "".join(lines)
|
|
|
|
|
|
def _generate_plugin_market_default(wiki_file: Path, config_file: Path) -> list[str]:
|
|
markdown = wiki_file.read_text(encoding="utf-8")
|
|
repos = extract_plugin_market_repos_from_wiki(markdown, require_markers=True)
|
|
if not repos:
|
|
raise ValueError("Wiki 插件仓库清单为空")
|
|
if OFFICIAL_PLUGIN_MARKET not in repos:
|
|
raise ValueError("Wiki 插件仓库清单缺少 MoviePilot 官方插件仓库")
|
|
|
|
source = config_file.read_text(encoding="utf-8")
|
|
start, end, indent = _find_plugin_market_assignment(source)
|
|
source_lines = source.splitlines(keepends=True)
|
|
replacement = _format_plugin_market_assignment(repos, indent)
|
|
updated_source = (
|
|
"".join(source_lines[:start])
|
|
+ replacement
|
|
+ "".join(source_lines[end:])
|
|
)
|
|
config_file.write_text(updated_source, encoding="utf-8")
|
|
return repos
|
|
|
|
|
|
def main(args: Optional[list[str]] = None) -> int:
|
|
"""
|
|
读取 Wiki 清单并更新指定配置文件中的插件市场默认值。
|
|
"""
|
|
options = _parse_args(args)
|
|
repos = _generate_plugin_market_default(options.wiki_file, options.config_file)
|
|
print(f"已生成 PLUGIN_MARKET 默认值，共 {len(repos)} 个仓库")
|
|
return 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
raise SystemExit(main())
|