1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
|
# -*- coding: utf-8 -*-
import pytest
from subliminal_patch.providers.supersubtitles import SuperSubtitlesProvider
from subliminal_patch.providers.supersubtitles import SuperSubtitlesSubtitle
from subliminal_patch.core import Episode
from subzero.language import Language
@pytest.fixture
def episode():
episode = {
"name": "/tv/All of Us Are Dead/Season 1/All of Us Are Dead - S01E11 - Episode 11 WEBDL-1080p.mp4",
"source": "Web",
"release_group": None,
"resolution": "1080p",
"video_codec": None,
"audio_codec": None,
"imdb_id": None,
"subtitle_languages": set(),
"streaming_service": None,
"edition": None,
"series": "All of Us Are Dead",
"season": 1,
"episode": 11,
"title": "Episode 11",
"year": None,
"original_series": True,
"tvdb_id": None,
"series_tvdb_id": None,
"series_imdb_id": None,
"alternative_series": [],
}
return Episode(**episode)
def test_list_episode_subtitles(episode):
language = Language.fromalpha2("en")
with SuperSubtitlesProvider() as provider:
assert provider.list_subtitles(episode, {language})
def test_download_episode_subtitle(episode):
subtitle = SuperSubtitlesSubtitle(
Language.fromalpha2("en"),
"https://www.feliratok.eu/index.php?action=letolt&felirat=1643361676",
1643361676,
"All of us are dead",
1,
11,
"",
[
"NF.WEB-DL.1080p-TEPES",
"NF.WEBRip.1080p-TEPES",
"WEBRip-ION10",
"WEBRip-ION265",
"WEBRip.1080p-RARBG",
],
"",
"",
"",
asked_for_episode=True,
)
assert subtitle.get_matches(episode)
with SuperSubtitlesProvider() as provider:
provider.download_subtitle(subtitle)
assert subtitle.is_valid()
def test_list_and_download_movie_subtitles(movies):
movie = movies["dune"]
language = Language.fromalpha2("en")
with SuperSubtitlesProvider() as provider:
assert provider.list_subtitles(movie, {language})
def test_download_movie_subtitle(movies):
movie = movies["dune"]
subtitle = SuperSubtitlesSubtitle(
Language.fromalpha2("en"),
"https://www.feliratok.eu/index.php?action=letolt&felirat=1634579718",
1634579718,
"Dune",
0,
0,
"",
[
"NF.WEB-DL.1080p-TEPES",
"NF.WEBRip.1080p-TEPES",
"WEBRip-ION10",
"WEBRip-ION265",
"WEBRip.1080p-RARBG",
],
"",
"",
"",
asked_for_episode=None,
)
assert subtitle.get_matches(movie)
with SuperSubtitlesProvider() as provider:
provider.download_subtitle(subtitle)
assert subtitle.is_valid()
def test_subtitle_reprs(movies):
subtitle = SuperSubtitlesSubtitle(
Language.fromalpha2("en"),
"https://www.feliratok.eu/index.php?action=letolt&felirat=1634579718",
1634579718,
"Dune",
0,
0,
"",
[
"NF.WEB-DL.1080p-TEPES",
"NF.WEBRip.1080p-TEPES",
"WEBRip-ION10",
"WEBRip-ION265",
"WEBRip.1080p-RARBG",
],
"",
"",
"",
asked_for_episode=None,
)
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)
|
