mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-08-29 12:06:51 +08:00
502 lines
21 KiB
Python
502 lines
21 KiB
Python
import ast
|
|
import asyncio
|
|
import json
|
|
from pathlib import Path
|
|
from types import SimpleNamespace
|
|
|
|
from fastapi import HTTPException
|
|
|
|
from app.factory import create_app, localized_http_exception_handler
|
|
from app.runtime.localization import LocaleHelper
|
|
from app.runtime.progress import ProgressHelper
|
|
from app.schemas.dashboard import ScheduleInfo, ScheduleProgress
|
|
from app.schemas.response import Response
|
|
from version import APP_VERSION
|
|
|
|
|
|
def _has_chinese(text: str) -> bool:
|
|
"""判断文本是否包含中文字符。"""
|
|
return any("\u4e00" <= char <= "\u9fff" for char in text)
|
|
|
|
|
|
def _sample_message_expression(value: ast.AST) -> list[str]:
|
|
"""从接口 message 表达式中生成用于翻译校验的样本文本。"""
|
|
if isinstance(value, ast.Constant) and isinstance(value.value, str):
|
|
return [value.value]
|
|
if isinstance(value, ast.JoinedStr):
|
|
parts = []
|
|
has_chinese = False
|
|
index = 0
|
|
for item in value.values:
|
|
if isinstance(item, ast.Constant) and isinstance(item.value, str):
|
|
parts.append(item.value)
|
|
has_chinese = has_chinese or _has_chinese(item.value)
|
|
else:
|
|
index += 1
|
|
parts.append(f"样例{index}")
|
|
return ["".join(parts)] if has_chinese else []
|
|
if isinstance(value, ast.IfExp):
|
|
return _sample_message_expression(value.body) + _sample_message_expression(value.orelse)
|
|
if isinstance(value, ast.BoolOp):
|
|
samples = []
|
|
for item in value.values:
|
|
samples.extend(_sample_message_expression(item))
|
|
return samples
|
|
if isinstance(value, ast.BinOp) and isinstance(value.op, ast.Add):
|
|
left_samples = _sample_message_expression(value.left)
|
|
right_samples = _sample_message_expression(value.right)
|
|
if left_samples and right_samples:
|
|
return [left + right for left in left_samples for right in right_samples]
|
|
if left_samples:
|
|
return [f"{left}样例" for left in left_samples]
|
|
if right_samples:
|
|
return [f"样例{right}" for right in right_samples]
|
|
return []
|
|
|
|
|
|
def _sample_progress_text_expression(value: ast.AST) -> list[str]:
|
|
"""从进度 text 表达式中生成用于翻译校验的样本文本。"""
|
|
if isinstance(value, ast.Constant) and isinstance(value.value, str):
|
|
return [value.value]
|
|
if isinstance(value, ast.JoinedStr):
|
|
samples = [""]
|
|
placeholder_index = 0
|
|
for item in value.values:
|
|
if isinstance(item, ast.Constant) and isinstance(item.value, str):
|
|
choices = [item.value]
|
|
elif isinstance(item, ast.FormattedValue):
|
|
choices = [
|
|
sample
|
|
for sample in _sample_progress_text_expression(item.value)
|
|
if _has_chinese(sample)
|
|
]
|
|
if not choices:
|
|
placeholder_index += 1
|
|
choices = [f"样例{placeholder_index}"]
|
|
else:
|
|
placeholder_index += 1
|
|
choices = [f"样例{placeholder_index}"]
|
|
samples = [prefix + choice for prefix in samples for choice in choices]
|
|
return [sample for sample in samples if _has_chinese(sample)]
|
|
if isinstance(value, ast.IfExp):
|
|
return (
|
|
_sample_progress_text_expression(value.body)
|
|
+ _sample_progress_text_expression(value.orelse)
|
|
)
|
|
if isinstance(value, ast.BoolOp):
|
|
samples = []
|
|
for item in value.values:
|
|
samples.extend(_sample_progress_text_expression(item))
|
|
return samples
|
|
if isinstance(value, ast.BinOp) and isinstance(value.op, ast.Add):
|
|
left_samples = _sample_progress_text_expression(value.left)
|
|
right_samples = _sample_progress_text_expression(value.right)
|
|
if left_samples and right_samples:
|
|
return [left + right for left in left_samples for right in right_samples]
|
|
if left_samples:
|
|
return [f"{left}样例" for left in left_samples]
|
|
if right_samples:
|
|
return [f"样例{right}" for right in right_samples]
|
|
return []
|
|
|
|
|
|
def _is_progress_text_call(node: ast.Call) -> bool:
|
|
"""判断调用是否可能写入前端展示的进度文本。"""
|
|
func = ast.unparse(node.func)
|
|
return (
|
|
func.endswith("progress_callback")
|
|
or func.endswith(".update")
|
|
or func.endswith(".end")
|
|
)
|
|
|
|
|
|
def test_locale_helper_normalizes_supported_aliases():
|
|
"""语言别名应规范化为项目支持的语言标识。"""
|
|
assert LocaleHelper.normalize_locale("en") == "en-US"
|
|
assert LocaleHelper.normalize_locale("zh_Hant") == "zh-TW"
|
|
assert LocaleHelper.normalize_locale("unknown") == "zh-CN"
|
|
|
|
|
|
def test_locale_helper_reads_request_headers_by_priority():
|
|
"""请求语言应优先使用显式前端语言头，再回退 Accept-Language。"""
|
|
request = SimpleNamespace(
|
|
headers={
|
|
"x-moviepilot-locale": "en-US",
|
|
"accept-language": "zh-TW,zh;q=0.9",
|
|
}
|
|
)
|
|
|
|
assert LocaleHelper.get_locale_from_request(request) == "en-US"
|
|
|
|
|
|
def test_locale_helper_reads_query_locale_before_headers():
|
|
"""SSE 请求可通过查询参数显式指定前端语言。"""
|
|
request = SimpleNamespace(
|
|
query_params={"locale": "zh-TW"},
|
|
headers={
|
|
"x-moviepilot-locale": "en-US",
|
|
"accept-language": "en-US",
|
|
},
|
|
)
|
|
|
|
assert LocaleHelper.get_locale_from_request(request) == "zh-TW"
|
|
|
|
|
|
def test_locale_helper_parses_accept_language_quality():
|
|
"""Accept-Language 应按 q 权重选择支持的语言。"""
|
|
request = SimpleNamespace(
|
|
headers={
|
|
"accept-language": "fr-FR, en-US;q=0.8, zh-TW;q=0.9",
|
|
}
|
|
)
|
|
|
|
assert LocaleHelper.get_locale_from_request(request) == "zh-TW"
|
|
|
|
|
|
def test_locale_helper_translates_module_name_and_keeps_missing_text():
|
|
"""翻译存在时返回目标语言，缺失时返回默认文本。"""
|
|
assert LocaleHelper.translate(
|
|
"system.modules.DoubanModule.name",
|
|
locale="en-US",
|
|
default="豆瓣",
|
|
) == "Douban"
|
|
assert LocaleHelper.translate_text("模块不支持测试", locale="en-US") == "Module does not support testing"
|
|
assert LocaleHelper.translate_text(
|
|
"requirements.txt 文件下载失败", locale="en-US"
|
|
) == "Failed to download requirements.txt"
|
|
assert LocaleHelper.translate_text("未收录的中文错误", locale="en-US") == "未收录的中文错误"
|
|
|
|
|
|
def test_locale_helper_translates_dynamic_message_patterns():
|
|
"""动态消息应保留变量并翻译模板文本。"""
|
|
assert LocaleHelper.translate_text(
|
|
"无法连接Qbittorrent下载器：默认",
|
|
locale="en-US",
|
|
) == "Unable to connect to qBittorrent downloader: 默认"
|
|
assert LocaleHelper.translate_text(
|
|
"无法连接 api.themoviedb.org，错误码：403",
|
|
locale="en-US",
|
|
) == "Unable to connect to api.themoviedb.org, error code: 403"
|
|
assert LocaleHelper.translate_text(
|
|
"飞书 工作 未就绪",
|
|
locale="zh-TW",
|
|
) == "飛書 工作 尚未就緒"
|
|
|
|
|
|
def test_locale_helper_translates_common_backend_response_messages():
|
|
"""常见后端链路消息应能生成英文展示文本。"""
|
|
samples = {
|
|
"QQ Bot 默认 未就绪": "QQ Bot 默认 is not ready",
|
|
"微信 ClawBot 工作 未就绪：未登录": "WeChat ClawBot 工作 is not ready: 未登录",
|
|
"无法打开网站！": "Unable to open the site!",
|
|
"错误：403 Forbidden": "Error: 403 Forbidden",
|
|
"站点【https://example.com】不存在": "Site [https://example.com] does not exist",
|
|
"下载字幕文件失败，状态码：404 Not Found": "Failed to download subtitle file, status code: 404 Not Found",
|
|
"未获取到第 2 季的总集数": "Unable to get the total episode count for season 2",
|
|
"电影.mkv 已在整理队列中": "电影.mkv is already in the organization queue",
|
|
"未识别到媒体信息，类型：电视剧，id：123": "Unable to recognize media information, type: 电视剧, id: 123",
|
|
"默认 的下载目录 /downloads 不存在": "Download directory for 默认 does not exist: /downloads",
|
|
"不支持 Local 到 Alist 的文件整理": "File organization from Local to Alist is not supported",
|
|
"文件 /tmp/a.mkv 不存在": "File /tmp/a.mkv does not exist",
|
|
"添加种子任务失败：种子无效": "Failed to add torrent task: 种子无效",
|
|
"检查授权状态失败: timeout": "Failed to check authorization status: timeout",
|
|
"未找到名为 工作 的微信 ClawBot 通知配置": (
|
|
"No WeChat ClawBot notification configuration named 工作 was found"
|
|
),
|
|
"整理记录不存在: 1, 2": "Organization record does not exist: 1, 2",
|
|
"插件要求 MoviePilot 版本 >=2.14.0，当前版本 2.13.0 不满足，已拒绝安装": (
|
|
"The plugin requires MoviePilot version >=2.14.0, but current version 2.13.0 "
|
|
"does not satisfy it. Installation was rejected"
|
|
),
|
|
"已安排一次性 release 升级并重启": "Scheduled one-shot release upgrade and restart",
|
|
"已将微信 ClawBot 登录缓存从 old 迁移到 new": (
|
|
"Migrated WeChat ClawBot login cache from old to new"
|
|
),
|
|
"搜索完成，共 3 个资源": "Search completed, 3 resources",
|
|
"正在搜索关键字，已完成 1 / 6 个请求 ...": (
|
|
"Searching 关键字, completed 1/6 requests ..."
|
|
),
|
|
"正在搜索字幕关键字，已完成 1 / 6 个请求 ...": (
|
|
"Searching subtitles 关键字, completed 1/6 requests ..."
|
|
),
|
|
"未开启任何支持字幕搜索的有效站点，无法搜索字幕": (
|
|
"No valid subtitle-search site is enabled, unable to search subtitles"
|
|
),
|
|
"用户名或密码错误": "Incorrect username or password",
|
|
"工具 'query_schedulers' 未找到": "Tool 'query_schedulers' was not found",
|
|
"插件 test 不存在或未加载": "Plugin test does not exist or is not loaded",
|
|
"站点 1 不存在": "Site 1 does not exist",
|
|
"智能助手未启用，请先在系统设置中开启。": (
|
|
"The assistant is not enabled. Enable it in system settings first."
|
|
),
|
|
"智能助手执行失败: timeout": "Assistant execution failed: timeout",
|
|
}
|
|
|
|
for message, expected in samples.items():
|
|
assert LocaleHelper.translate_text(message, locale="en-US") == expected
|
|
|
|
|
|
def test_response_localizes_message_from_locale_context():
|
|
"""通用 Response 应根据请求语言上下文直接翻译消息。"""
|
|
token = LocaleHelper.set_current_locale("en-US")
|
|
try:
|
|
response = Response(success=False, message="模块不支持测试")
|
|
finally:
|
|
LocaleHelper.reset_current_locale(token)
|
|
|
|
assert response.message == "Module does not support testing"
|
|
assert not hasattr(response, "message_i18n")
|
|
|
|
|
|
def test_http_exception_handler_returns_localized_response_envelope():
|
|
"""HTTPException 响应应统一封装并直接翻译错误文本。"""
|
|
token = LocaleHelper.set_current_locale("en-US")
|
|
try:
|
|
response = asyncio.run(
|
|
localized_http_exception_handler(
|
|
SimpleNamespace(
|
|
query_params={},
|
|
headers={"accept-language": "en-US"},
|
|
),
|
|
HTTPException(status_code=401, detail="用户名或密码错误"),
|
|
)
|
|
)
|
|
finally:
|
|
LocaleHelper.reset_current_locale(token)
|
|
|
|
assert response.status_code == 401
|
|
payload = json.loads(response.body)
|
|
assert payload == {
|
|
"success": False,
|
|
"message": "Incorrect username or password",
|
|
"data": None,
|
|
}
|
|
|
|
|
|
def test_application_docs_use_v1_openapi_and_backend_version():
|
|
"""默认接口文档应展示 v1 地址并使用真实后端版本号。"""
|
|
app = create_app()
|
|
|
|
assert app.openapi_url == "/api/v1/openapi.json"
|
|
assert app.version == APP_VERSION
|
|
assert "/api/v1/openapi.json" in {route.path for route in app.routes}
|
|
assert "/api/v2/openapi.json" not in {route.path for route in app.routes}
|
|
|
|
|
|
def test_progress_helper_get_adds_i18n_fields_without_mutating_cache():
|
|
"""通用进度字典应返回展示字段翻译，同时保留缓存中的原始中文。"""
|
|
progress = ProgressHelper("__test_i18n_progress")
|
|
progress.start()
|
|
progress.update(
|
|
text="开始同步媒体服务器，共 2 个 ...",
|
|
data={"error": "后台服务不存在"},
|
|
)
|
|
|
|
detail = progress.get(locale="en-US")
|
|
assert detail is not None
|
|
assert detail["text"] == "开始同步媒体服务器，共 2 个 ..."
|
|
assert detail["text_i18n"] == "Starting media server sync, 2 servers ..."
|
|
assert detail["data"]["error"] == "后台服务不存在"
|
|
assert detail["data"]["error_i18n"] == "Background service does not exist"
|
|
|
|
detail["data"]["error_i18n"] = "mutated"
|
|
second_detail = progress.get(locale="en-US")
|
|
assert second_detail is not None
|
|
assert second_detail["data"]["error_i18n"] == "Background service does not exist"
|
|
|
|
|
|
def test_schedule_info_auto_fills_i18n_display_fields():
|
|
"""后台服务数据应补充前端展示所需的多语言字段。"""
|
|
token = LocaleHelper.set_current_locale("en-US")
|
|
try:
|
|
schedule = ScheduleInfo(
|
|
id="mediaserver_sync",
|
|
name="同步媒体服务器",
|
|
provider="[系统]",
|
|
status="等待",
|
|
next_run="2小时3分钟",
|
|
progress_text="开始同步媒体服务器，共 2 个 ...",
|
|
progress_detail=ScheduleProgress(
|
|
id="mediaserver_sync",
|
|
name="同步媒体服务器",
|
|
provider="[系统]",
|
|
text="媒体服务器 Emby 无可同步媒体库",
|
|
error="后台服务不存在",
|
|
),
|
|
)
|
|
finally:
|
|
LocaleHelper.reset_current_locale(token)
|
|
|
|
assert schedule.name_i18n == "Sync Media Servers"
|
|
assert schedule.provider_i18n == "[System]"
|
|
assert schedule.status_i18n == "Waiting"
|
|
assert schedule.next_run_i18n == "2h 3m"
|
|
assert schedule.progress_text_i18n == "Starting media server sync, 2 servers ..."
|
|
assert schedule.progress_detail.text_i18n == "Media server Emby has no libraries to sync"
|
|
assert schedule.progress_detail.error_i18n == "Background service does not exist"
|
|
|
|
token = LocaleHelper.set_current_locale("en-US")
|
|
try:
|
|
dynamic_schedule = ScheduleInfo(
|
|
id="mediaserver_sync_example",
|
|
name="同步媒体服务器 - Plex",
|
|
progress_text="同步媒体服务器 - Plex 开始执行 ...",
|
|
)
|
|
finally:
|
|
LocaleHelper.reset_current_locale(token)
|
|
assert dynamic_schedule.name_i18n == "Sync Media Server - Plex"
|
|
assert dynamic_schedule.progress_text_i18n == "Starting media server sync - Plex ..."
|
|
|
|
|
|
def test_scheduler_progress_patterns_translate_dynamic_texts():
|
|
"""定时任务进度动态模板应翻译固定词并保留业务变量。"""
|
|
assert (
|
|
LocaleHelper.translate_text("同步媒体服务器 开始执行 ...", locale="en-US")
|
|
== "Starting Sync Media Servers ..."
|
|
)
|
|
assert (
|
|
LocaleHelper.translate_text("同步媒体服务器 执行完成", locale="en-US")
|
|
== "Sync Media Servers completed"
|
|
)
|
|
assert (
|
|
LocaleHelper.translate_text("同步媒体服务器 执行失败", locale="zh-TW")
|
|
== "同步媒體伺服器 執行失敗"
|
|
)
|
|
assert (
|
|
LocaleHelper.translate_text("工作流动作（1/3）动作A 执行完成", locale="en-US")
|
|
== "Workflow action (1/3) 动作A completed"
|
|
)
|
|
|
|
|
|
def test_scheduler_progress_texts_have_english_translations():
|
|
"""定时任务相关进度文案应有英文翻译，避免前端切英文后回退中文。"""
|
|
untranslated = []
|
|
progress_paths = [
|
|
Path("app/scheduler.py"),
|
|
Path("app/chain/torrents.py"),
|
|
Path("app/chain/mediaserver.py"),
|
|
Path("app/chain/site.py"),
|
|
Path("app/chain/recommend.py"),
|
|
Path("app/chain/transfer.py"),
|
|
Path("app/chain/subscribe.py"),
|
|
Path("app/chain/workflow.py"),
|
|
]
|
|
for path in progress_paths:
|
|
tree = ast.parse(path.read_text())
|
|
for node in ast.walk(tree):
|
|
if not isinstance(node, ast.Call) or not _is_progress_text_call(node):
|
|
continue
|
|
for keyword in node.keywords:
|
|
if keyword.arg != "text":
|
|
continue
|
|
for message in _sample_progress_text_expression(keyword.value):
|
|
translated = LocaleHelper.translate_text(message, locale="en-US")
|
|
if translated == message:
|
|
untranslated.append(f"{path}:{keyword.value.lineno}:{message}")
|
|
|
|
assert untranslated == []
|
|
|
|
|
|
def test_api_endpoint_literal_messages_have_english_translations():
|
|
"""接口直接返回的中文 message 应有英文翻译，避免前端切英文后回退中文。"""
|
|
untranslated = []
|
|
for path in Path("app/api/endpoints").glob("*.py"):
|
|
tree = ast.parse(path.read_text())
|
|
for node in ast.walk(tree):
|
|
if not isinstance(node, ast.Call):
|
|
continue
|
|
for keyword in node.keywords:
|
|
if keyword.arg != "message":
|
|
continue
|
|
value = keyword.value
|
|
if not isinstance(value, ast.Constant) or not isinstance(value.value, str):
|
|
continue
|
|
message = value.value
|
|
if not any("\u4e00" <= char <= "\u9fff" for char in message):
|
|
continue
|
|
translated = LocaleHelper.translate_text(message, locale="en-US")
|
|
if translated == message:
|
|
untranslated.append(f"{path}:{value.lineno}:{message}")
|
|
|
|
assert untranslated == []
|
|
|
|
|
|
def test_api_endpoint_dynamic_messages_have_english_translations():
|
|
"""接口动态拼接的中文 message 应由模板翻译覆盖。"""
|
|
untranslated = []
|
|
for path in Path("app/api/endpoints").glob("*.py"):
|
|
tree = ast.parse(path.read_text())
|
|
for node in ast.walk(tree):
|
|
if not isinstance(node, ast.Call):
|
|
continue
|
|
for keyword in node.keywords:
|
|
if keyword.arg != "message":
|
|
continue
|
|
value = keyword.value
|
|
if not isinstance(value, ast.JoinedStr):
|
|
continue
|
|
parts = []
|
|
has_chinese = False
|
|
index = 0
|
|
for item in value.values:
|
|
if isinstance(item, ast.Constant) and isinstance(item.value, str):
|
|
parts.append(item.value)
|
|
has_chinese = has_chinese or _has_chinese(item.value)
|
|
else:
|
|
index += 1
|
|
parts.append(f"样例{index}")
|
|
if not has_chinese:
|
|
continue
|
|
message = "".join(parts)
|
|
translated = LocaleHelper.translate_text(message, locale="en-US")
|
|
if translated == message:
|
|
untranslated.append(f"{path}:{value.lineno}:{message}")
|
|
|
|
assert untranslated == []
|
|
|
|
|
|
def test_api_endpoint_message_expressions_have_english_translations():
|
|
"""接口 message 表达式中的中文分支应有英文翻译。"""
|
|
untranslated = []
|
|
for path in Path("app/api/endpoints").glob("*.py"):
|
|
tree = ast.parse(path.read_text())
|
|
for node in ast.walk(tree):
|
|
if not isinstance(node, ast.Call):
|
|
continue
|
|
if ast.unparse(node.func) not in {"schemas.Response", "Response"}:
|
|
continue
|
|
for keyword in node.keywords:
|
|
if keyword.arg != "message":
|
|
continue
|
|
for message in _sample_message_expression(keyword.value):
|
|
if not _has_chinese(message):
|
|
continue
|
|
translated = LocaleHelper.translate_text(message, locale="en-US")
|
|
if translated == message:
|
|
untranslated.append(f"{path}:{keyword.value.lineno}:{message}")
|
|
|
|
assert untranslated == []
|
|
|
|
|
|
def test_api_endpoint_http_exception_details_have_english_translations():
|
|
"""HTTPException 的中文 detail 应有英文翻译。"""
|
|
untranslated = []
|
|
for path in Path("app/api/endpoints").glob("*.py"):
|
|
tree = ast.parse(path.read_text())
|
|
for node in ast.walk(tree):
|
|
if not isinstance(node, ast.Call):
|
|
continue
|
|
if not ast.unparse(node.func).endswith("HTTPException"):
|
|
continue
|
|
for keyword in node.keywords:
|
|
if keyword.arg != "detail":
|
|
continue
|
|
for message in _sample_message_expression(keyword.value):
|
|
if not _has_chinese(message):
|
|
continue
|
|
translated = LocaleHelper.translate_text(message, locale="en-US")
|
|
if translated == message:
|
|
untranslated.append(f"{path}:{keyword.value.lineno}:{message}")
|
|
|
|
assert untranslated == []
|