]> jfr.im git - yt-dlp.git/blame - yt_dlp/extractor/tv2.py
[misc] Add `hatch`, `ruff`, `pre-commit` and improve dev docs (#7409)
[yt-dlp.git] / yt_dlp / extractor / tv2.py
CommitLineData
588b82bb
S
1import re
2
bc0f937b 3from .common import InfoExtractor
3d2623a8 4from ..networking.exceptions import HTTPError
bc0f937b 5from ..utils import (
8d3a3a99 6 ExtractorError,
e897bd82 7 determine_ext,
bc0f937b 8 float_or_none,
e897bd82 9 int_or_none,
481c5c51 10 js_to_json,
bc0f937b 11 parse_iso8601,
588b82bb 12 remove_end,
8d3a3a99 13 strip_or_none,
9a621ddc 14 try_get,
bc0f937b
S
15)
16
17
18class TV2IE(InfoExtractor):
acf306d1 19 _VALID_URL = r'https?://(?:www\.)?tv2\.no/v(?:ideo)?\d*/(?:[^?#]+/)*(?P<id>\d+)'
2181983a 20 _TESTS = [{
acf306d1 21 'url': 'http://www.tv2.no/v/1791207/',
bc0f937b 22 'info_dict': {
acf306d1 23 'id': '1791207',
ea81966e 24 'ext': 'mp4',
acf306d1 25 'title': 'Her kolliderer romsonden med asteroiden ',
26 'description': 'En romsonde har krasjet inn i en asteroide i verdensrommet. Kollisjonen skjedde klokken 01:14 natt til tirsdag 27. september norsk tid. \n\nNasa kaller det sitt første forsøk på planetforsvar.',
27 'timestamp': 1664238190,
28 'upload_date': '20220927',
29 'duration': 146,
30 'thumbnail': r're:^https://.*$',
bc0f937b
S
31 'view_count': int,
32 'categories': list,
ed1a3905 33 },
e5d731f3 34 }, {
35 'url': 'http://www.tv2.no/v2/916509',
36 'only_matching': True,
acf306d1 37 }, {
38 'url': 'https://www.tv2.no/video/nyhetene/her-kolliderer-romsonden-med-asteroiden/1791207/',
39 'only_matching': True,
2181983a 40 }]
ea81966e 41 _PROTOCOLS = ('HLS', 'DASH')
8d3a3a99 42 _GEO_COUNTRIES = ['NO']
bc0f937b
S
43
44 def _real_extract(self, url):
45 video_id = self._match_id(url)
ea81966e
A
46 asset = self._download_json('https://sumo.tv2.no/rest/assets/' + video_id, video_id,
47 'Downloading metadata JSON')
48 title = asset['title']
2181983a 49 is_live = asset.get('live') is True
50
bc0f937b
S
51 formats = []
52 format_urls = []
8d3a3a99
RA
53 for protocol in self._PROTOCOLS:
54 try:
ea81966e
A
55 data = self._download_json('https://api.sumo.tv2.no/play/%s?stream=%s' % (video_id, protocol),
56 video_id, 'Downloading playabck JSON',
57 headers={'content-type': 'application/json'},
58 data='{"device":{"id":"1-1-1","name":"Nettleser (HTML)"}}'.encode())['playback']
8d3a3a99 59 except ExtractorError as e:
3d2623a8 60 if isinstance(e.cause, HTTPError) and e.cause.status == 401:
61 error = self._parse_json(e.cause.response.read().decode(), video_id)['error']
8d3a3a99
RA
62 error_code = error.get('code')
63 if error_code == 'ASSET_PLAYBACK_INVALID_GEO_LOCATION':
64 self.raise_geo_restricted(countries=self._GEO_COUNTRIES)
65 elif error_code == 'SESSION_NOT_AUTHENTICATED':
66 self.raise_login_required()
67 raise ExtractorError(error['description'])
68 raise
ea81966e 69 items = data.get('streams', [])
9a621ddc 70 for item in items:
bc0f937b
S
71 video_url = item.get('url')
72 if not video_url or video_url in format_urls:
73 continue
ea81966e 74 format_id = '%s-%s' % (protocol.lower(), item.get('type'))
bc0f937b
S
75 if not self._is_valid_url(video_url, video_id, format_id):
76 continue
77 format_urls.append(video_url)
78 ext = determine_ext(video_url)
79 if ext == 'f4m':
80 formats.extend(self._extract_f4m_formats(
dbc0b39b 81 video_url, video_id, f4m_id=format_id, fatal=False))
bc0f937b 82 elif ext == 'm3u8':
0b25af9b
RA
83 if not data.get('drmProtected'):
84 formats.extend(self._extract_m3u8_formats(
a5c0c202 85 video_url, video_id, 'mp4', live=is_live, m3u8_id=format_id, fatal=False))
8d3a3a99
RA
86 elif ext == 'mpd':
87 formats.extend(self._extract_mpd_formats(
88 video_url, video_id, format_id, fatal=False))
bc0f937b
S
89 elif ext == 'ism' or video_url.endswith('.ism/Manifest'):
90 pass
91 else:
92 formats.append({
93 'url': video_url,
94 'format_id': format_id,
bc0f937b 95 })
0b25af9b 96 if not formats and data.get('drmProtected'):
88acdbc2 97 self.report_drm(video_id)
bc0f937b 98
bc0f937b 99 thumbnails = [{
ea81966e
A
100 'id': type,
101 'url': thumb_url,
102 } for type, thumb_url in (asset.get('images') or {}).items()]
bc0f937b
S
103
104 return {
105 'id': video_id,
106 'url': video_url,
39ca3b5c 107 'title': title,
8d3a3a99 108 'description': strip_or_none(asset.get('description')),
bc0f937b 109 'thumbnails': thumbnails,
ea81966e 110 'timestamp': parse_iso8601(asset.get('live_broadcast_time') or asset.get('update_time')),
8d3a3a99
RA
111 'duration': float_or_none(asset.get('accurateDuration') or asset.get('duration')),
112 'view_count': int_or_none(asset.get('views')),
ea81966e 113 'categories': asset.get('tags', '').split(','),
bc0f937b 114 'formats': formats,
2181983a 115 'is_live': is_live,
bc0f937b 116 }
588b82bb
S
117
118
119class TV2ArticleIE(InfoExtractor):
acf306d1 120 _VALID_URL = r'https?://(?:www\.)?tv2\.no/(?!v(?:ideo)?\d*/)[^?#]+/(?P<id>\d+)'
588b82bb 121 _TESTS = [{
acf306d1 122 'url': 'https://www.tv2.no/underholdning/forraeder/katarina-flatland-angrer-etter-forraeder-exit/15095188/',
588b82bb 123 'info_dict': {
acf306d1 124 'id': '15095188',
125 'title': 'Katarina Flatland angrer etter Forræder-exit',
126 'description': 'SANDEFJORD (TV 2): Katarina Flatland (33) måtte følge i sine fars fotspor, da hun ble forvist fra Forræder.',
588b82bb
S
127 },
128 'playlist_count': 2,
129 }, {
130 'url': 'http://www.tv2.no/a/6930542',
131 'only_matching': True,
132 }]
133
134 def _real_extract(self, url):
135 playlist_id = self._match_id(url)
136
137 webpage = self._download_webpage(url, playlist_id)
138
481c5c51
S
139 # Old embed pattern (looks unused nowadays)
140 assets = re.findall(r'data-assetid=["\'](\d+)', webpage)
141
142 if not assets:
143 # New embed pattern
acf306d1 144 for v in re.findall(r'(?s)(?:TV2ContentboxVideo|TV2\.TV2Video)\(({.+?})\)', webpage):
481c5c51
S
145 video = self._parse_json(
146 v, playlist_id, transform_source=js_to_json, fatal=False)
147 if not video:
148 continue
149 asset = video.get('assetId')
150 if asset:
151 assets.append(asset)
152
588b82bb 153 entries = [
481c5c51
S
154 self.url_result('http://www.tv2.no/v/%s' % asset_id, 'TV2')
155 for asset_id in assets]
588b82bb
S
156
157 title = remove_end(self._og_search_title(webpage), ' - TV2.no')
158 description = remove_end(self._og_search_description(webpage), ' - TV2.no')
159
160 return self.playlist_result(entries, playlist_id, title, description)
8d3a3a99
RA
161
162
ea81966e 163class KatsomoIE(InfoExtractor):
df773c3d 164 _WORKING = False
2181983a 165 _VALID_URL = r'https?://(?:www\.)?(?:katsomo|mtv(uutiset)?)\.fi/(?:sarja/[0-9a-z-]+-\d+/[0-9a-z-]+-|(?:#!/)?jakso/(?:\d+/[^/]+/)?|video/prog)(?P<id>\d+)'
166 _TESTS = [{
8d3a3a99
RA
167 'url': 'https://www.mtv.fi/sarja/mtv-uutiset-live-33001002003/lahden-pelicans-teki-kovan-ratkaisun-ville-nieminen-pihalle-1181321',
168 'info_dict': {
169 'id': '1181321',
170 'ext': 'mp4',
2181983a 171 'title': 'Lahden Pelicans teki kovan ratkaisun – Ville Nieminen pihalle',
8d3a3a99
RA
172 'description': 'Päätöksen teki Pelicansin hallitus.',
173 'timestamp': 1575116484,
174 'upload_date': '20191130',
175 'duration': 37.12,
176 'view_count': int,
177 'categories': list,
178 },
179 'params': {
180 # m3u8 download
181 'skip_download': True,
182 },
2181983a 183 }, {
184 'url': 'http://www.katsomo.fi/#!/jakso/33001005/studio55-fi/658521/jukka-kuoppamaki-tekee-yha-lauluja-vaikka-lentokoneessa',
185 'only_matching': True,
186 }, {
187 'url': 'https://www.mtvuutiset.fi/video/prog1311159',
188 'only_matching': True,
189 }, {
190 'url': 'https://www.katsomo.fi/#!/jakso/1311159',
191 'only_matching': True,
192 }]
8d3a3a99
RA
193 _API_DOMAIN = 'api.katsomo.fi'
194 _PROTOCOLS = ('HLS', 'MPD')
195 _GEO_COUNTRIES = ['FI']
2181983a 196
ea81966e
A
197 def _real_extract(self, url):
198 video_id = self._match_id(url)
199 api_base = 'http://%s/api/web/asset/%s' % (self._API_DOMAIN, video_id)
200
201 asset = self._download_json(
202 api_base + '.json', video_id,
203 'Downloading metadata JSON')['asset']
204 title = asset.get('subtitle') or asset['title']
205 is_live = asset.get('live') is True
206
207 formats = []
208 format_urls = []
209 for protocol in self._PROTOCOLS:
210 try:
211 data = self._download_json(
212 api_base + '/play.json?protocol=%s&videoFormat=SMIL+ISMUSP' % protocol,
213 video_id, 'Downloading play JSON')['playback']
214 except ExtractorError as e:
3d2623a8 215 if isinstance(e.cause, HTTPError) and e.cause.status == 401:
216 error = self._parse_json(e.cause.response.read().decode(), video_id)['error']
ea81966e
A
217 error_code = error.get('code')
218 if error_code == 'ASSET_PLAYBACK_INVALID_GEO_LOCATION':
219 self.raise_geo_restricted(countries=self._GEO_COUNTRIES)
220 elif error_code == 'SESSION_NOT_AUTHENTICATED':
221 self.raise_login_required()
222 raise ExtractorError(error['description'])
223 raise
224 items = try_get(data, lambda x: x['items']['item'])
225 if not items:
226 continue
227 if not isinstance(items, list):
228 items = [items]
229 for item in items:
230 if not isinstance(item, dict):
231 continue
232 video_url = item.get('url')
233 if not video_url or video_url in format_urls:
234 continue
235 format_id = '%s-%s' % (protocol.lower(), item.get('mediaFormat'))
236 if not self._is_valid_url(video_url, video_id, format_id):
237 continue
238 format_urls.append(video_url)
239 ext = determine_ext(video_url)
240 if ext == 'f4m':
241 formats.extend(self._extract_f4m_formats(
242 video_url, video_id, f4m_id=format_id, fatal=False))
243 elif ext == 'm3u8':
244 if not data.get('drmProtected'):
245 formats.extend(self._extract_m3u8_formats(
a5c0c202 246 video_url, video_id, 'mp4', live=is_live, m3u8_id=format_id, fatal=False))
ea81966e
A
247 elif ext == 'mpd':
248 formats.extend(self._extract_mpd_formats(
249 video_url, video_id, format_id, fatal=False))
250 elif ext == 'ism' or video_url.endswith('.ism/Manifest'):
251 pass
252 else:
253 formats.append({
254 'url': video_url,
255 'format_id': format_id,
256 'tbr': int_or_none(item.get('bitrate')),
257 'filesize': int_or_none(item.get('fileSize')),
258 })
259 if not formats and data.get('drmProtected'):
260 self.report_drm(video_id)
ea81966e
A
261
262 thumbnails = [{
263 'id': thumbnail.get('@type'),
264 'url': thumbnail.get('url'),
265 } for _, thumbnail in (asset.get('imageVersions') or {}).items()]
266
267 return {
268 'id': video_id,
269 'url': video_url,
39ca3b5c 270 'title': title,
ea81966e
A
271 'description': strip_or_none(asset.get('description')),
272 'thumbnails': thumbnails,
273 'timestamp': parse_iso8601(asset.get('createTime')),
274 'duration': float_or_none(asset.get('accurateDuration') or asset.get('duration')),
275 'view_count': int_or_none(asset.get('views')),
276 'categories': asset.get('keywords', '').split(','),
277 'formats': formats,
278 'is_live': is_live,
279 }
280
2181983a 281
282class MTVUutisetArticleIE(InfoExtractor):
df773c3d 283 _WORKING = False
2181983a 284 _VALID_URL = r'https?://(?:www\.)mtvuutiset\.fi/artikkeli/[^/]+/(?P<id>\d+)'
285 _TESTS = [{
286 'url': 'https://www.mtvuutiset.fi/artikkeli/tallaisia-vaurioita-viking-amorellassa-on-useamman-osaston-alla-vetta/7931384',
287 'info_dict': {
288 'id': '1311159',
289 'ext': 'mp4',
290 'title': 'Viking Amorellan matkustajien evakuointi on alkanut – tältä operaatio näyttää laivalla',
291 'description': 'Viking Amorellan matkustajien evakuointi on alkanut – tältä operaatio näyttää laivalla',
292 'timestamp': 1600608966,
293 'upload_date': '20200920',
294 'duration': 153.7886666,
295 'view_count': int,
296 'categories': list,
297 },
298 'params': {
299 # m3u8 download
300 'skip_download': True,
301 },
302 }, {
303 # multiple Youtube embeds
304 'url': 'https://www.mtvuutiset.fi/artikkeli/50-vuotta-subarun-vastaiskua/6070962',
305 'only_matching': True,
306 }]
307
308 def _real_extract(self, url):
309 article_id = self._match_id(url)
310 article = self._download_json(
311 'http://api.mtvuutiset.fi/mtvuutiset/api/json/' + article_id,
312 article_id)
313
314 def entries():
315 for video in (article.get('videos') or []):
316 video_type = video.get('videotype')
317 video_url = video.get('url')
318 if not (video_url and video_type in ('katsomo', 'youtube')):
319 continue
320 yield self.url_result(
321 video_url, video_type.capitalize(), video.get('video_id'))
322
323 return self.playlist_result(
324 entries(), article_id, article.get('title'), article.get('description'))