feat: unify media source identity flow (#6129)

This commit is contained in:
jxxghp
2026-07-21 20:20:09 +08:00
parent 6f8b6cfbc9
commit 3b709b7f2e
78 changed files with 2519 additions and 1121 deletions
+41
View File
@@ -0,0 +1,41 @@
from app.core.context import MediaInfo
from app.helper.message import TemplateContextBuilder
from app.schemas.types import MediaType
def test_message_context_contains_all_media_identity_fields() -> None:
"""消息模板上下文应向插件暴露全部媒体源 ID 和当前主身份。"""
context = {}
media = MediaInfo(
source="anilist",
type=MediaType.TV,
title="测试动画",
tmdb_id=24680,
douban_id="35000000",
bangumi_id=499390,
anilist_id=170942,
)
TemplateContextBuilder._add_media_info(context, media)
assert context["tmdbid"] == 24680
assert context["doubanid"] == "35000000"
assert context["bangumiid"] == 499390
assert context["anilistid"] == 170942
assert context["media_source"] == "anilist"
assert context["media_id"] == "170942"
assert media.to_dict()["mediaid_prefix"] == "anilist"
assert media.to_dict()["media_id"] == "170942"
def test_media_info_preserves_plugin_source_identity() -> None:
"""核心媒体对象应原样保留插件自定义数据源的原生 ID。"""
media = MediaInfo(
source="plugin_source",
media_id="custom-100",
type=MediaType.MOVIE,
title="插件电影",
)
assert media.media_id == "custom-100"
assert media.to_dict()["media_id"] == "custom-100"