SuperSubtitles provider: fix episode matching

This commit is contained in:
Vitiko 2023-03-30 17:26:21 -04:00
parent ef67cd4792
commit 1427a8ab73
2 changed files with 43 additions and 1 deletions

View File

@ -95,7 +95,7 @@ class SuperSubtitlesSubtitle(Subtitle):
def get_matches(self, video):
matches = set()
update_matches(matches, video, self.releases)
update_matches(matches, video, self.release_info)
# episode
if isinstance(video, Episode):

View File

@ -130,3 +130,45 @@ def test_subtitle_reprs(movies):
)
assert isinstance(subtitle.__repr__(), str)
assert isinstance(subtitle.__str__(), str)
@pytest.fixture
def video_2():
return Episode(
**{
"name": "/tv/La.Brea/Season.02/La.Brea.S02E13.720p.WEB.H264-CAKES.mkv",
"source": "Web",
"release_group": "CAKES",
"resolution": "720p",
"video_codec": "H.264",
"audio_codec": "Dolby Digital Plus",
"imdb_id": None,
"size": 1598748631,
"original_name": "la.brea.s02e13.720p.web.h264-cakes.mkv",
"hints": {"title": "La Brea", "type": "episode", "single_value": True},
"streaming_service": None,
"edition": None,
"original_path": "/tv/La.Brea/Season.02/la.brea.s02e13.720p.web.h264-cakes.mkv",
"other": None,
"series": "La Brea",
"season": 2,
"episode": 13,
"title": "The Journey (1)",
"year": 2021,
"tvdb_id": None,
"series_tvdb_id": 395029,
"series_imdb_id": "tt11640018",
"alternative_series": [],
"used_scene_name": True,
}
)
def test_list_video_2(video_2):
lang = Language.fromalpha2("hu")
with SuperSubtitlesProvider() as provider:
subs = provider.list_subtitles(video_2, {lang})
for sub in subs:
matches = sub.get_matches(video_2)
assert {"season", "episode", "series"}.issubset(matches)