mirror of
https://github.com/jxxghp/MoviePilot.git
synced 2026-02-02 18:22:39 +08:00
41381a920c
问题：订阅中添加的自定义识别词（特别是集数偏移）在下载时正常生效， 但在下载完成整理时没有生效。 根因：下载历史中未保存识别词，整理时 MetaInfoPath 未接收 custom_words 参数。 修复： - 在 DownloadHistory 模型中添加 custom_words 字段 - 下载时从 meta.apply_words 获取并保存识别词到下载历史 - MetaInfoPath 函数添加 custom_words 参数支持 - 整理时从下载历史获取 custom_words 并传递给 MetaInfoPath - 添加 Alembic 迁移脚本 (2.2.3) - 添加相关单元测试
99 lines
4.0 KiB
Python
99 lines
4.0 KiB
Python
# -*- coding: utf-8 -*-
|
||
from pathlib import Path
|
||
from unittest import TestCase
|
||
|
||
from app.core.metainfo import MetaInfo, MetaInfoPath
|
||
from tests.cases.meta import meta_cases
|
||
|
||
|
||
class MetaInfoTest(TestCase):
|
||
def setUp(self) -> None:
|
||
pass
|
||
|
||
def tearDown(self) -> None:
|
||
pass
|
||
|
||
def test_metainfo(self):
|
||
for info in meta_cases:
|
||
if info.get("path"):
|
||
meta_info = MetaInfoPath(path=Path(info.get("path")))
|
||
else:
|
||
meta_info = MetaInfo(title=info.get("title"), subtitle=info.get("subtitle"), custom_words=["#"])
|
||
target = {
|
||
"type": meta_info.type.value,
|
||
"cn_name": meta_info.cn_name or "",
|
||
"en_name": meta_info.en_name or "",
|
||
"year": meta_info.year or "",
|
||
"part": meta_info.part or "",
|
||
"season": meta_info.season,
|
||
"episode": meta_info.episode,
|
||
"restype": meta_info.edition,
|
||
"pix": meta_info.resource_pix or "",
|
||
"video_codec": meta_info.video_encode or "",
|
||
"audio_codec": meta_info.audio_encode or ""
|
||
}
|
||
|
||
# 检查tmdbid
|
||
if info.get("target").get("tmdbid"):
|
||
target["tmdbid"] = meta_info.tmdbid
|
||
|
||
self.assertEqual(target, info.get("target"))
|
||
|
||
def test_emby_format_ids(self):
|
||
"""
|
||
测试Emby格式ID识别
|
||
"""
|
||
# 测试文件路径
|
||
test_paths = [
|
||
# 文件名中包含tmdbid
|
||
("/movies/The Vampire Diaries (2009) [tmdbid=18165]/The.Vampire.Diaries.S01E01.1080p.mkv", 18165),
|
||
# 目录名中包含tmdbid
|
||
("/movies/Inception (2010) [tmdbid-27205]/Inception.2010.1080p.mkv", 27205),
|
||
# 父目录名中包含tmdbid
|
||
("/movies/Breaking Bad (2008) [tmdb=1396]/Season 1/Breaking.Bad.S01E01.1080p.mkv", 1396),
|
||
# 祖父目录名中包含tmdbid
|
||
("/tv/Game of Thrones (2011) {tmdb=1399}/Season 1/Game.of.Thrones.S01E01.1080p.mkv", 1399),
|
||
# 测试{tmdb-xxx}格式
|
||
("/movies/Avatar (2009) {tmdb-19995}/Avatar.2009.1080p.mkv", 19995),
|
||
]
|
||
|
||
for path_str, expected_tmdbid in test_paths:
|
||
meta = MetaInfoPath(Path(path_str))
|
||
self.assertEqual(meta.tmdbid, expected_tmdbid,
|
||
f"路径 {path_str} 期望的tmdbid为 {expected_tmdbid}，实际识别为 {meta.tmdbid}")
|
||
|
||
def test_metainfopath_with_custom_words(self):
|
||
"""测试 MetaInfoPath 使用自定义识别词"""
|
||
# 测试替换词：将"测试替换"替换为空
|
||
custom_words = ["测试替换 => "]
|
||
path = Path("/movies/电影测试替换名称 (2024)/movie.mkv")
|
||
meta = MetaInfoPath(path, custom_words=custom_words)
|
||
# 验证替换生效：cn_name 不应包含"测试替换"
|
||
if meta.cn_name:
|
||
self.assertNotIn("测试替换", meta.cn_name)
|
||
|
||
def test_metainfopath_without_custom_words(self):
|
||
"""测试 MetaInfoPath 不传入自定义识别词"""
|
||
path = Path("/movies/Normal Movie (2024)/movie.mkv")
|
||
meta = MetaInfoPath(path)
|
||
# 验证正常识别，不报错
|
||
self.assertIsNotNone(meta)
|
||
|
||
def test_metainfopath_with_empty_custom_words(self):
|
||
"""测试 MetaInfoPath 传入空的自定义识别词"""
|
||
path = Path("/movies/Test Movie (2024)/movie.mkv")
|
||
meta = MetaInfoPath(path, custom_words=[])
|
||
# 验证不报错，正常识别
|
||
self.assertIsNotNone(meta)
|
||
|
||
def test_custom_words_apply_words_recording(self):
|
||
"""测试 apply_words 记录功能"""
|
||
custom_words = ["替换词 => 新词"]
|
||
title = "电影替换词.2024.mkv"
|
||
meta = MetaInfo(title=title, custom_words=custom_words)
|
||
# 验证 apply_words 属性存在
|
||
self.assertTrue(hasattr(meta, 'apply_words'))
|
||
# 如果替换词被应用，应该记录在 apply_words 中
|
||
if meta.apply_words:
|
||
self.assertIn("替换词 => 新词", meta.apply_words)
|