fix(transfer): 修复订阅自定义识别词在整理时失效 (#6018)

tests/test_download_chain.py

@@ -158,6 +158,61 @@ def test_download_single_submits_download_added_to_background(monkeypatch):
     )
 
 
+def test_download_single_persists_custom_words_snapshot(monkeypatch):
+    """下载成功登记历史时，应把传入的订阅识别词原样存入快照，供整理时原样复现识别。"""
+    captured = {}
+
+    class _CapturingDownloadHistoryOper:
+        """捕获写入下载历史的字段，验证识别词快照确实落库。"""
+
+        def add(self, **kwargs):
+            captured.update(kwargs)
+
+        def add_files(self, _files):
+            pass
+
+    _FakeThreadHelper.submitted = []
+    monkeypatch.setattr(download_module, "ThreadHelper", _FakeThreadHelper)
+    monkeypatch.setattr(download_module, "DownloadHistoryOper", _CapturingDownloadHistoryOper)
+    monkeypatch.setattr(download_module, "TorrentHelper", _FakeTorrentHelper)
+
+    chain = DownloadChain.__new__(DownloadChain)
+    chain.download = MagicMock(return_value=("qb", "hash123", "Original", "添加下载成功"))
+    chain.download_added = MagicMock()
+    chain.eventmanager = MagicMock()
+    chain.eventmanager.send_event.return_value = None
+    chain.post_message = MagicMock()
+
+    context = Context(
+        meta_info=MetaInfo("Demo Show 2024"),
+        media_info=MediaInfo(
+            type=MediaType.TV,
+            title="Demo Show",
+            year="2024",
+            tmdb_id=1,
+            genre_ids=[18],
+        ),
+        torrent_info=TorrentInfo(
+            title="Demo Show 2024",
+            enclosure="https://example.com/demo.torrent",
+            site_cookie="uid=1",
+            site_name="TestSite",
+        ),
+    )
+
+    custom_words = "S04 => S01\n第 <> 集 >> EP+66"
+    result = chain.download_single(
+        context=context,
+        torrent_content=b"torrent-content",
+        save_path="/downloads",
+        username="tester",
+        custom_words=custom_words,
+    )
+
+    assert result == "hash123"
+    assert captured["custom_words"] == custom_words
+
+
 def test_save_subtitle_response_creates_missing_temp_directory(monkeypatch, tmp_path):
     """
     下载字幕 API 保存响应前应自动创建缺失的临时目录。
@@ -468,6 +523,29 @@ def test_batch_download_does_not_download_duplicate_movie_after_success(monkeypa
     assert chain.download_single.call_args.args[0] is first_context
 
 
+def test_batch_download_threads_custom_words_to_download_single(monkeypatch):
+    """订阅识别词须经 batch_download 透传到 download_single，作为整理快照随下载存档。"""
+    _FakeBatchTorrentHelper.episodes = []
+    monkeypatch.setattr(download_module, "TorrentHelper", _FakeBatchTorrentHelper)
+    monkeypatch.setattr(download_module.eventmanager, "send_event", lambda *args, **kwargs: None)
+
+    chain = DownloadChain.__new__(DownloadChain)
+    chain.download_single = MagicMock(return_value="hash")
+
+    context = SimpleNamespace(
+        media_info=SimpleNamespace(type=MediaType.MOVIE, title_year="Demo Movie (2026)"),
+        meta_info=SimpleNamespace(season_episode=""),
+        torrent_info=SimpleNamespace(title="Demo Movie"),
+    )
+
+    custom_words = "S04 => S01\n第 <> 集 >> EP+66"
+    downloads, _lefts = chain.batch_download(contexts=[context], custom_words=custom_words)
+
+    assert downloads == [context]
+    chain.download_single.assert_called_once()
+    assert chain.download_single.call_args.kwargs["custom_words"] == custom_words
+
+
 def test_batch_download_accepts_complete_coverage_when_files_cover_target_range(monkeypatch):
     """
     自定义起始集场景按目标范围覆盖判断，100-143 可满足 start=100、total=143。

tests/test_subscribe_chain.py

@@ -385,6 +385,7 @@ class SubscribeChainTest(TestCase):
             "description": None,
             "last_update": None,
             "username": None,
+            "custom_words": None,
             "to_dict": lambda: {},
         }
         data.update(overrides)
@@ -1247,7 +1248,11 @@ class SubscribeChainTest(TestCase):
         )
 
     def test_episode_best_version_downloads_full_pack_before_episode_fallback(self):
-        subscribe = self._build_subscribe(best_version_full=0, total_episode=3)
+        subscribe = self._build_subscribe(
+            best_version_full=0,
+            total_episode=3,
+            custom_words="S04 => S01\n第 <> 集 >> EP+66",
+        )
         full_pack_context = SimpleNamespace(
             torrent_info=SimpleNamespace(pri_order=90),
             media_info=SimpleNamespace(type=MediaType.TV),
@@ -1295,6 +1300,8 @@ class SubscribeChainTest(TestCase):
         self.assertEqual(len(calls), 1)
         self.assertEqual(calls[0]["contexts"], [full_pack_context])
         self.assertEqual(calls[0]["no_exists"]["media-key"][1].episodes, [])
+        # 订阅识别词须作为入参随下载下传，供整理时复现识别（避免下载模块反查订阅的循环依赖）
+        self.assertEqual(calls[0]["custom_words"], "S04 => S01\n第 <> 集 >> EP+66")
 
     def test_episode_best_version_falls_back_when_full_pack_not_downloaded(self):
         subscribe = self._build_subscribe(best_version_full=0, total_episode=3)

tests/test_transfer_custom_words.py

@@ -0,0 +1,76 @@
+# -*- coding: utf-8 -*-
+"""订阅自定义识别词快照用例：下载时保存完整识别词，整理时快照优先、实时反查兜底。
+
+回归场景：订阅做季+集组合偏移（如 S04E05→S01E71），下载阶段生效但整理阶段因实时反查订阅
+返回空而静默回退全局识别词、丢失偏移。修复后由订阅链在发起下载时将完整识别词作为入参传入
+下载模块并存档（避免下载模块反查订阅的同级循环依赖），整理时优先复用该快照。
+"""
+from types import SimpleNamespace
+
+import app.chain.transfer as transfer_module
+from app.chain.transfer import TransferChain
+
+
+def _fake_history(custom_words=None, note=None):
+    """构造仅含测试所需字段的下载历史替身。"""
+    return SimpleNamespace(custom_words=custom_words, note=note)
+
+
+def test_transfer_prefers_snapshot_over_live_lookup(monkeypatch):
+    """整理时存在下载快照，应直接使用快照且不触发实时反查订阅。"""
+    called = {"lookup": False}
+
+    class _GuardSubscribeChain:
+        def get_subscribe_by_source(self, source):
+            # 一旦走到实时反查即视为失败：快照存在时不应触发
+            called["lookup"] = True
+            return SimpleNamespace(custom_words="不应使用\n实时反查")
+
+    monkeypatch.setattr(transfer_module, "SubscribeChain", _GuardSubscribeChain)
+
+    history = _fake_history(
+        custom_words="S04 => S01\n第 <> 集 >> EP+66",
+        note={"source": "Subscribe|{...}"},
+    )
+    result = TransferChain._get_subscribe_custom_words(history)
+
+    assert result == ["S04 => S01", "第 <> 集 >> EP+66"]
+    assert called["lookup"] is False
+
+
+def test_transfer_falls_back_to_live_lookup_without_snapshot(monkeypatch):
+    """整理时无快照（历史旧记录），应按下载来源实时反查订阅取识别词。"""
+
+    class _FakeSubscribeChain:
+        def get_subscribe_by_source(self, source):
+            assert source == "Subscribe|{...}"
+            return SimpleNamespace(custom_words="A => B")
+
+    monkeypatch.setattr(transfer_module, "SubscribeChain", _FakeSubscribeChain)
+
+    history = _fake_history(custom_words=None, note={"source": "Subscribe|{...}"})
+    result = TransferChain._get_subscribe_custom_words(history)
+
+    assert result == ["A => B"]
+
+
+def test_transfer_returns_none_when_unavailable(monkeypatch):
+    """无下载记录、note 非字典、或来源反查不到订阅时返回 None（回退全局识别词）。"""
+
+    class _NoneSubscribeChain:
+        def get_subscribe_by_source(self, source):
+            return None
+
+    monkeypatch.setattr(transfer_module, "SubscribeChain", _NoneSubscribeChain)
+
+    # 无下载记录
+    assert TransferChain._get_subscribe_custom_words(None) is None
+    # 无快照且 note 非字典：不应触发实时反查
+    assert TransferChain._get_subscribe_custom_words(_fake_history(note="不是字典")) is None
+    # 无快照、来源可解析但反查不到订阅
+    assert (
+        TransferChain._get_subscribe_custom_words(
+            _fake_history(note={"source": "Subscribe|{}"})
+        )
+        is None
+    )