feat: support TMDB episode group (g=) in explicit media tags and custom identifiers

- Add episode_group (g=) parameter parsing to explicit media tags in both Python and Rust metainfo parsers
- Propagate episode_group through MetaInfo, MetaBase, MediaInfo, and context models
- Update SKILL.md and update_custom_identifiers.py docs to describe episode group usage
- Add tests for episode_group recognition in metainfo and chain recognition logic
This commit is contained in:
jxxghp
2026-05-24 23:32:27 +08:00
parent ea52537423
commit ac3432c54f
10 changed files with 203 additions and 63 deletions

View File

@@ -0,0 +1,37 @@
import sys
from types import ModuleType
from unittest.mock import patch
sys.modules.setdefault("qbittorrentapi", ModuleType("qbittorrentapi"))
setattr(sys.modules["qbittorrentapi"], "TorrentFilesList", list)
sys.modules.setdefault("transmission_rpc", ModuleType("transmission_rpc"))
setattr(sys.modules["transmission_rpc"], "File", object)
sys.modules.setdefault("psutil", ModuleType("psutil"))
from app.chain import ChainBase
from app.core.context import MediaInfo
from app.core.meta import MetaBase
from app.schemas.types import MediaType
def test_recognize_media_uses_meta_episode_group():
"""
识别链未显式传 episode_group 时,应沿用元数据中识别出的剧集组。
"""
group_id = "5ad0ec240e0a26303f00d84d"
chain = ChainBase()
meta = MetaBase("测试剧集")
meta.name = "测试剧集"
meta.type = MediaType.TV
meta.episode_group = group_id
mediainfo = MediaInfo(title="测试剧集", year="2024", tmdb_id=100, type=MediaType.TV)
with patch.object(chain, "run_module", return_value=mediainfo) as run_module, patch(
"app.chain.MediaRecognizeShareHelper.report",
return_value=True,
), patch("app.chain.MediaRecognizeShareHelper.query") as query_mock:
result = chain.recognize_media(meta=meta, cache=False)
assert result is mediainfo
assert run_module.call_args.kwargs["episode_group"] == group_id
query_mock.assert_not_called()

View File

@@ -1,6 +1,7 @@
# -*- coding: utf-8 -*-
from pathlib import Path
from unittest import TestCase
from unittest.mock import patch
from app.core.metainfo import MetaInfo, MetaInfoPath, find_metainfo
from tests.cases.meta import meta_cases
@@ -132,6 +133,33 @@ class MetaInfoTest(TestCase):
self.assertEqual(meta.episode, "E04")
self.assertEqual(meta.apply_words, custom_words)
def test_custom_words_support_episode_group_parameter(self):
"""测试自定义识别词替换结果中的 g 参数会写入剧集组"""
group_id = "5ad0ec240e0a26303f00d84d"
custom_words = [
f"Bakemonogatari => 物语系列 {{[tmdbid=46195;type=tv;g={group_id};s=1]}}"
]
meta = MetaInfo(title="Bakemonogatari 01", custom_words=custom_words)
self.assertEqual(meta.tmdbid, 46195)
self.assertEqual(meta.type.value, "电视剧")
self.assertEqual(meta.begin_season, 1)
self.assertEqual(meta.episode_group, group_id)
self.assertEqual(meta.apply_words, custom_words)
def test_find_metainfo_supports_episode_group_parameter(self):
"""测试显式媒体标签支持 g 剧集组参数"""
group_id = "5ad0ec240e0a26303f00d84d"
title, metainfo = find_metainfo(f"物语系列 {{[tmdbid=46195;type=tv;g={group_id};s=1]}}")
self.assertEqual(metainfo["episode_group"], group_id)
self.assertNotIn("g=", title)
def test_find_metainfo_does_not_support_episode_group_alias(self):
"""测试 e_group 不会被当作剧集组参数识别"""
group_id = "5ad0ec240e0a26303f00d84d"
with patch("app.core.metainfo.rust_accel.find_metainfo", return_value=None):
_, metainfo = find_metainfo(f"物语系列 {{[tmdbid=46195;type=tv;e_group={group_id};s=1]}}")
self.assertIsNone(metainfo["episode_group"])
def test_video_bit_extracted_for_video_title(self):
"""测试普通影视标题中的视频位深可单独识别"""
meta = MetaInfo(title="The 355 2022 BluRay 1080p DTS-HD MA5.1 X265.10bit-BeiTai")

View File

@@ -223,6 +223,22 @@ def test_rust_metainfo_parser_handles_anime_from_entry():
assert result["audio_encode"] == "AAC"
def test_rust_metainfo_parser_handles_episode_group():
"""
Rust MetaInfo 入口应识别显式媒体标签中的 g 剧集组参数。
"""
group_id = "5ad0ec240e0a26303f00d84d"
result = rust_accel.parse_metainfo(
f"物语系列 {{[tmdbid=46195;type=tv;g={group_id};s=1]}} 01",
options=_metainfo_options(),
)
assert result["tmdbid"] == 46195
assert result["type"] == MediaType.TV.value
assert result["episode_group"] == group_id
assert result["begin_season"] == 1
def test_rust_metainfo_path_parser_merges_parent_title():
"""
Rust MetaInfoPath 入口应在 Rust 内完成父目录标题合并。