c506bc5dd2402a95752bdf3223fe4a24cf9d06ae
[youtube-dl] / youtube_dl / extractor / canvas.py
1 from __future__ import unicode_literals
2
3 import re
4 import json
5
6 from .common import InfoExtractor
7 from .gigya import GigyaBaseIE
8 from ..compat import compat_HTTPError
9 from ..utils import (
10     ExtractorError,
11     strip_or_none,
12     float_or_none,
13     int_or_none,
14     merge_dicts,
15     parse_iso8601,
16 )
17
18
19 class CanvasIE(InfoExtractor):
20     _VALID_URL = r'https?://mediazone\.vrt\.be/api/v1/(?P<site_id>canvas|een|ketnet|vrt(?:video|nieuws)|sporza)/assets/(?P<id>[^/?#&]+)'
21     _TESTS = [{
22         'url': 'https://mediazone.vrt.be/api/v1/ketnet/assets/md-ast-4ac54990-ce66-4d00-a8ca-9eac86f4c475',
23         'md5': '90139b746a0a9bd7bb631283f6e2a64e',
24         'info_dict': {
25             'id': 'md-ast-4ac54990-ce66-4d00-a8ca-9eac86f4c475',
26             'display_id': 'md-ast-4ac54990-ce66-4d00-a8ca-9eac86f4c475',
27             'ext': 'flv',
28             'title': 'Nachtwacht: De Greystook',
29             'description': 'md5:1db3f5dc4c7109c821261e7512975be7',
30             'thumbnail': r're:^https?://.*\.jpg$',
31             'duration': 1468.03,
32         },
33         'expected_warnings': ['is not a supported codec', 'Unknown MIME type'],
34     }, {
35         'url': 'https://mediazone.vrt.be/api/v1/canvas/assets/mz-ast-5e5f90b6-2d72-4c40-82c2-e134f884e93e',
36         'only_matching': True,
37     }]
38     _HLS_ENTRY_PROTOCOLS_MAP = {
39         'HLS': 'm3u8_native',
40         'HLS_AES': 'm3u8',
41     }
42
43     def _real_extract(self, url):
44         mobj = re.match(self._VALID_URL, url)
45         site_id, video_id = mobj.group('site_id'), mobj.group('id')
46
47         data = self._download_json(
48             'https://mediazone.vrt.be/api/v1/%s/assets/%s'
49             % (site_id, video_id), video_id)
50
51         title = data['title']
52         description = data.get('description')
53
54         formats = []
55         for target in data['targetUrls']:
56             format_url, format_type = target.get('url'), target.get('type')
57             if not format_url or not format_type:
58                 continue
59             if format_type in self._HLS_ENTRY_PROTOCOLS_MAP:
60                 formats.extend(self._extract_m3u8_formats(
61                     format_url, video_id, 'mp4', self._HLS_ENTRY_PROTOCOLS_MAP[format_type],
62                     m3u8_id=format_type, fatal=False))
63             elif format_type == 'HDS':
64                 formats.extend(self._extract_f4m_formats(
65                     format_url, video_id, f4m_id=format_type, fatal=False))
66             elif format_type == 'MPEG_DASH':
67                 formats.extend(self._extract_mpd_formats(
68                     format_url, video_id, mpd_id=format_type, fatal=False))
69             elif format_type == 'HSS':
70                 formats.extend(self._extract_ism_formats(
71                     format_url, video_id, ism_id='mss', fatal=False))
72             else:
73                 formats.append({
74                     'format_id': format_type,
75                     'url': format_url,
76                 })
77         self._sort_formats(formats)
78
79         subtitles = {}
80         subtitle_urls = data.get('subtitleUrls')
81         if isinstance(subtitle_urls, list):
82             for subtitle in subtitle_urls:
83                 subtitle_url = subtitle.get('url')
84                 if subtitle_url and subtitle.get('type') == 'CLOSED':
85                     subtitles.setdefault('nl', []).append({'url': subtitle_url})
86
87         return {
88             'id': video_id,
89             'display_id': video_id,
90             'title': title,
91             'description': description,
92             'formats': formats,
93             'duration': float_or_none(data.get('duration'), 1000),
94             'thumbnail': data.get('posterImageUrl'),
95             'subtitles': subtitles,
96         }
97
98
99 class CanvasEenIE(InfoExtractor):
100     IE_DESC = 'canvas.be and een.be'
101     _VALID_URL = r'https?://(?:www\.)?(?P<site_id>canvas|een)\.be/(?:[^/]+/)*(?P<id>[^/?#&]+)'
102     _TESTS = [{
103         'url': 'http://www.canvas.be/video/de-afspraak/najaar-2015/de-afspraak-veilt-voor-de-warmste-week',
104         'md5': 'ed66976748d12350b118455979cca293',
105         'info_dict': {
106             'id': 'mz-ast-5e5f90b6-2d72-4c40-82c2-e134f884e93e',
107             'display_id': 'de-afspraak-veilt-voor-de-warmste-week',
108             'ext': 'flv',
109             'title': 'De afspraak veilt voor de Warmste Week',
110             'description': 'md5:24cb860c320dc2be7358e0e5aa317ba6',
111             'thumbnail': r're:^https?://.*\.jpg$',
112             'duration': 49.02,
113         },
114         'expected_warnings': ['is not a supported codec'],
115     }, {
116         # with subtitles
117         'url': 'http://www.canvas.be/video/panorama/2016/pieter-0167',
118         'info_dict': {
119             'id': 'mz-ast-5240ff21-2d30-4101-bba6-92b5ec67c625',
120             'display_id': 'pieter-0167',
121             'ext': 'mp4',
122             'title': 'Pieter 0167',
123             'description': 'md5:943cd30f48a5d29ba02c3a104dc4ec4e',
124             'thumbnail': r're:^https?://.*\.jpg$',
125             'duration': 2553.08,
126             'subtitles': {
127                 'nl': [{
128                     'ext': 'vtt',
129                 }],
130             },
131         },
132         'params': {
133             'skip_download': True,
134         },
135         'skip': 'Pagina niet gevonden',
136     }, {
137         'url': 'https://www.een.be/sorry-voor-alles/herbekijk-sorry-voor-alles',
138         'info_dict': {
139             'id': 'mz-ast-11a587f8-b921-4266-82e2-0bce3e80d07f',
140             'display_id': 'herbekijk-sorry-voor-alles',
141             'ext': 'mp4',
142             'title': 'Herbekijk Sorry voor alles',
143             'description': 'md5:8bb2805df8164e5eb95d6a7a29dc0dd3',
144             'thumbnail': r're:^https?://.*\.jpg$',
145             'duration': 3788.06,
146         },
147         'params': {
148             'skip_download': True,
149         },
150         'skip': 'Episode no longer available',
151     }, {
152         'url': 'https://www.canvas.be/check-point/najaar-2016/de-politie-uw-vriend',
153         'only_matching': True,
154     }]
155
156     def _real_extract(self, url):
157         mobj = re.match(self._VALID_URL, url)
158         site_id, display_id = mobj.group('site_id'), mobj.group('id')
159
160         webpage = self._download_webpage(url, display_id)
161
162         title = strip_or_none(self._search_regex(
163             r'<h1[^>]+class="video__body__header__title"[^>]*>(.+?)</h1>',
164             webpage, 'title', default=None) or self._og_search_title(
165             webpage, default=None))
166
167         video_id = self._html_search_regex(
168             r'data-video=(["\'])(?P<id>(?:(?!\1).)+)\1', webpage, 'video id',
169             group='id')
170
171         return {
172             '_type': 'url_transparent',
173             'url': 'https://mediazone.vrt.be/api/v1/%s/assets/%s' % (site_id, video_id),
174             'ie_key': CanvasIE.ie_key(),
175             'id': video_id,
176             'display_id': display_id,
177             'title': title,
178             'description': self._og_search_description(webpage),
179         }
180
181
182 class VrtNUIE(GigyaBaseIE):
183     IE_DESC = 'VrtNU.be'
184     _VALID_URL = r'https?://(?:www\.)?vrt\.be/(?P<site_id>vrtnu)/(?:[^/]+/)*(?P<id>[^/?#&]+)'
185     _TESTS = [{
186         'url': 'https://www.vrt.be/vrtnu/a-z/postbus-x/1/postbus-x-s1a1/',
187         'info_dict': {
188             'id': 'pbs-pub-2e2d8c27-df26-45c9-9dc6-90c78153044d$vid-90c932b1-e21d-4fb8-99b1-db7b49cf74de',
189             'ext': 'flv',
190             'title': 'De zwarte weduwe',
191             'description': 'md5:d90c21dced7db869a85db89a623998d4',
192             'duration': 1457.04,
193             'thumbnail': r're:^https?://.*\.jpg$',
194             'season': '1',
195             'season_number': 1,
196             'episode_number': 1,
197         },
198         'skip': 'This video is only available for registered users'
199     }]
200     _NETRC_MACHINE = 'vrtnu'
201     _APIKEY = '3_0Z2HujMtiWq_pkAjgnS2Md2E11a1AwZjYiBETtwNE-EoEHDINgtnvcAOpNgmrVGy'
202     _CONTEXT_ID = 'R3595707040'
203
204     def _real_initialize(self):
205         self._login()
206
207     def _login(self):
208         username, password = self._get_login_info()
209         if username is None:
210             return
211
212         auth_data = {
213             'APIKey': self._APIKEY,
214             'targetEnv': 'jssdk',
215             'loginID': username,
216             'password': password,
217             'authMode': 'cookie',
218         }
219
220         auth_info = self._gigya_login(auth_data)
221
222         # Sometimes authentication fails for no good reason, retry
223         login_attempt = 1
224         while login_attempt <= 3:
225             try:
226                 # When requesting a token, no actual token is returned, but the
227                 # necessary cookies are set.
228                 self._request_webpage(
229                     'https://token.vrt.be',
230                     None, note='Requesting a token', errnote='Could not get a token',
231                     headers={
232                         'Content-Type': 'application/json',
233                         'Referer': 'https://www.vrt.be/vrtnu/',
234                     },
235                     data=json.dumps({
236                         'uid': auth_info['UID'],
237                         'uidsig': auth_info['UIDSignature'],
238                         'ts': auth_info['signatureTimestamp'],
239                         'email': auth_info['profile']['email'],
240                     }).encode('utf-8'))
241             except ExtractorError as e:
242                 if isinstance(e.cause, compat_HTTPError) and e.cause.code == 401:
243                     login_attempt += 1
244                     self.report_warning('Authentication failed')
245                     self._sleep(1, None, msg_template='Waiting for %(timeout)s seconds before trying again')
246                 else:
247                     raise e
248             else:
249                 break
250
251     def _real_extract(self, url):
252         display_id = self._match_id(url)
253
254         webpage, urlh = self._download_webpage_handle(url, display_id)
255
256         info = self._search_json_ld(webpage, display_id, default={})
257
258         # title is optional here since it may be extracted by extractor
259         # that is delegated from here
260         title = strip_or_none(self._html_search_regex(
261             r'(?ms)<h1 class="content__heading">(.+?)</h1>',
262             webpage, 'title', default=None))
263
264         description = self._html_search_regex(
265             r'(?ms)<div class="content__description">(.+?)</div>',
266             webpage, 'description', default=None)
267
268         season = self._html_search_regex(
269             [r'''(?xms)<div\ class="tabs__tab\ tabs__tab--active">\s*
270                     <span>seizoen\ (.+?)</span>\s*
271                 </div>''',
272              r'<option value="seizoen (\d{1,3})" data-href="[^"]+?" selected>'],
273             webpage, 'season', default=None)
274
275         season_number = int_or_none(season)
276
277         episode_number = int_or_none(self._html_search_regex(
278             r'''(?xms)<div\ class="content__episode">\s*
279                     <abbr\ title="aflevering">afl</abbr>\s*<span>(\d+)</span>
280                 </div>''',
281             webpage, 'episode_number', default=None))
282
283         release_date = parse_iso8601(self._html_search_regex(
284             r'(?ms)<div class="content__broadcastdate">\s*<time\ datetime="(.+?)"',
285             webpage, 'release_date', default=None))
286
287         # If there's a ? or a # in the URL, remove them and everything after
288         clean_url = urlh.geturl().split('?')[0].split('#')[0].strip('/')
289         securevideo_url = clean_url + '.mssecurevideo.json'
290
291         try:
292             video = self._download_json(securevideo_url, display_id)
293         except ExtractorError as e:
294             if isinstance(e.cause, compat_HTTPError) and e.cause.code == 401:
295                 self.raise_login_required()
296             raise
297
298         # We are dealing with a '../<show>.relevant' URL
299         redirect_url = video.get('url')
300         if redirect_url:
301             return self.url_result(self._proto_relative_url(redirect_url, 'https:'))
302
303         # There is only one entry, but with an unknown key, so just get
304         # the first one
305         video_id = list(video.values())[0].get('videoid')
306
307         return merge_dicts(info, {
308             '_type': 'url_transparent',
309             'url': 'https://mediazone.vrt.be/api/v1/vrtvideo/assets/%s' % video_id,
310             'ie_key': CanvasIE.ie_key(),
311             'id': video_id,
312             'display_id': display_id,
313             'title': title,
314             'description': description,
315             'season': season,
316             'season_number': season_number,
317             'episode_number': episode_number,
318             'release_date': release_date,
319         })