[vpro] Improve playlist extraction
[youtube-dl] / youtube_dl / extractor / npo.py
1 from __future__ import unicode_literals
2
3 import re
4
5 from .common import InfoExtractor
6 from ..compat import (
7     compat_urllib_request,
8     compat_urllib_parse,
9 )
10 from ..utils import (
11     fix_xml_ampersands,
12     parse_duration,
13     qualities,
14     strip_jsonp,
15     unified_strdate,
16     url_basename,
17 )
18
19
20 class NPOBaseIE(InfoExtractor):
21     def _get_token(self, video_id):
22         token_page = self._download_webpage(
23             'http://ida.omroep.nl/npoplayer/i.js',
24             video_id, note='Downloading token')
25         token = self._search_regex(
26             r'npoplayer\.token = "(.+?)"', token_page, 'token')
27         # Decryption algorithm extracted from http://npoplayer.omroep.nl/csjs/npoplayer-min.js
28         token_l = list(token)
29         first = second = None
30         for i in range(5, len(token_l) - 4):
31             if token_l[i].isdigit():
32                 if first is None:
33                     first = i
34                 elif second is None:
35                     second = i
36         if first is None or second is None:
37             first = 12
38             second = 13
39
40         token_l[first], token_l[second] = token_l[second], token_l[first]
41
42         return ''.join(token_l)
43
44
45 class NPOIE(NPOBaseIE):
46     IE_NAME = 'npo'
47     IE_DESC = 'npo.nl and ntr.nl'
48     _VALID_URL = r'''(?x)
49                     (?:
50                         npo:|
51                         https?://
52                             (?:www\.)?
53                             (?:
54                                 npo\.nl/(?!live|radio)(?:[^/]+/){2}|
55                                 ntr\.nl/(?:[^/]+/){2,}|
56                                 omroepwnl\.nl/video/fragment/[^/]+__
57                             )
58                         )
59                         (?P<id>[^/?#]+)
60                 '''
61
62     _TESTS = [
63         {
64             'url': 'http://www.npo.nl/nieuwsuur/22-06-2014/VPWON_1220719',
65             'md5': '4b3f9c429157ec4775f2c9cb7b911016',
66             'info_dict': {
67                 'id': 'VPWON_1220719',
68                 'ext': 'm4v',
69                 'title': 'Nieuwsuur',
70                 'description': 'Dagelijks tussen tien en elf: nieuws, sport en achtergronden.',
71                 'upload_date': '20140622',
72             },
73         },
74         {
75             'url': 'http://www.npo.nl/de-mega-mike-mega-thomas-show/27-02-2009/VARA_101191800',
76             'md5': 'da50a5787dbfc1603c4ad80f31c5120b',
77             'info_dict': {
78                 'id': 'VARA_101191800',
79                 'ext': 'm4v',
80                 'title': 'De Mega Mike & Mega Thomas show',
81                 'description': 'md5:3b74c97fc9d6901d5a665aac0e5400f4',
82                 'upload_date': '20090227',
83                 'duration': 2400,
84             },
85         },
86         {
87             'url': 'http://www.npo.nl/tegenlicht/25-02-2013/VPWON_1169289',
88             'md5': 'f8065e4e5a7824068ed3c7e783178f2c',
89             'info_dict': {
90                 'id': 'VPWON_1169289',
91                 'ext': 'm4v',
92                 'title': 'Tegenlicht',
93                 'description': 'md5:52cf4eefbc96fffcbdc06d024147abea',
94                 'upload_date': '20130225',
95                 'duration': 3000,
96             },
97         },
98         {
99             'url': 'http://www.npo.nl/de-nieuwe-mens-deel-1/21-07-2010/WO_VPRO_043706',
100             'info_dict': {
101                 'id': 'WO_VPRO_043706',
102                 'ext': 'wmv',
103                 'title': 'De nieuwe mens - Deel 1',
104                 'description': 'md5:518ae51ba1293ffb80d8d8ce90b74e4b',
105                 'duration': 4680,
106             },
107             'params': {
108                 # mplayer mms download
109                 'skip_download': True,
110             }
111         },
112         # non asf in streams
113         {
114             'url': 'http://www.npo.nl/hoe-gaat-europa-verder-na-parijs/10-01-2015/WO_NOS_762771',
115             'md5': 'b3da13de374cbe2d5332a7e910bef97f',
116             'info_dict': {
117                 'id': 'WO_NOS_762771',
118                 'ext': 'mp4',
119                 'title': 'Hoe gaat Europa verder na Parijs?',
120             },
121         },
122         {
123             'url': 'http://www.ntr.nl/Aap-Poot-Pies/27/detail/Aap-poot-pies/VPWON_1233944#content',
124             'md5': '01c6a2841675995da1f0cf776f03a9c3',
125             'info_dict': {
126                 'id': 'VPWON_1233944',
127                 'ext': 'm4v',
128                 'title': 'Aap, poot, pies',
129                 'description': 'md5:c9c8005d1869ae65b858e82c01a91fde',
130                 'upload_date': '20150508',
131                 'duration': 599,
132             },
133         },
134         {
135             'url': 'http://www.omroepwnl.nl/video/fragment/vandaag-de-dag-verkiezingen__POMS_WNL_853698',
136             'md5': 'd30cd8417b8b9bca1fdff27428860d08',
137             'info_dict': {
138                 'id': 'POW_00996502',
139                 'ext': 'm4v',
140                 'title': '''"Dit is wel een 'landslide'..."''',
141                 'description': 'md5:f8d66d537dfb641380226e31ca57b8e8',
142                 'upload_date': '20150508',
143                 'duration': 462,
144             },
145         }
146     ]
147
148     def _real_extract(self, url):
149         video_id = self._match_id(url)
150         return self._get_info(video_id)
151
152     def _get_info(self, video_id):
153         metadata = self._download_json(
154             'http://e.omroep.nl/metadata/%s' % video_id,
155             video_id,
156             # We have to remove the javascript callback
157             transform_source=strip_jsonp,
158         )
159
160         # For some videos actual video id (prid) is different (e.g. for
161         # http://www.omroepwnl.nl/video/fragment/vandaag-de-dag-verkiezingen__POMS_WNL_853698
162         # video id is POMS_WNL_853698 but prid is POW_00996502)
163         video_id = metadata.get('prid') or video_id
164
165         token = self._get_token(video_id)
166
167         formats = []
168
169         pubopties = metadata.get('pubopties')
170         if pubopties:
171             quality = qualities(['adaptive', 'wmv_sb', 'h264_sb', 'wmv_bb', 'h264_bb', 'wvc1_std', 'h264_std'])
172             for format_id in pubopties:
173                 format_info = self._download_json(
174                     'http://ida.omroep.nl/odi/?prid=%s&puboptions=%s&adaptive=yes&token=%s'
175                     % (video_id, format_id, token),
176                     video_id, 'Downloading %s JSON' % format_id)
177                 if format_info.get('error_code', 0) or format_info.get('errorcode', 0):
178                     continue
179                 streams = format_info.get('streams')
180                 if streams:
181                     video_info = self._download_json(
182                         streams[0] + '&type=json',
183                         video_id, 'Downloading %s stream JSON' % format_id)
184                 else:
185                     video_info = format_info
186                 video_url = video_info.get('url')
187                 if not video_url:
188                     continue
189                 if format_id == 'adaptive':
190                     formats.extend(self._extract_m3u8_formats(video_url, video_id))
191                 else:
192                     formats.append({
193                         'url': video_url,
194                         'format_id': format_id,
195                         'quality': quality(format_id),
196                     })
197
198         streams = metadata.get('streams')
199         if streams:
200             for i, stream in enumerate(streams):
201                 stream_url = stream.get('url')
202                 if not stream_url:
203                     continue
204                 if '.asf' not in stream_url:
205                     formats.append({
206                         'url': stream_url,
207                         'quality': stream.get('kwaliteit'),
208                     })
209                     continue
210                 asx = self._download_xml(
211                     stream_url, video_id,
212                     'Downloading stream %d ASX playlist' % i,
213                     transform_source=fix_xml_ampersands)
214                 ref = asx.find('./ENTRY/Ref')
215                 if ref is None:
216                     continue
217                 video_url = ref.get('href')
218                 if not video_url:
219                     continue
220                 formats.append({
221                     'url': video_url,
222                     'ext': stream.get('formaat', 'asf'),
223                     'quality': stream.get('kwaliteit'),
224                 })
225
226         self._sort_formats(formats)
227
228         subtitles = {}
229         if metadata.get('tt888') == 'ja':
230             subtitles['nl'] = [{
231                 'ext': 'vtt',
232                 'url': 'http://e.omroep.nl/tt888/%s' % video_id,
233             }]
234
235         return {
236             'id': video_id,
237             # prefer aflevering_titel if any since titel may be too generic, e.g.
238             # http://tegenlicht.vpro.nl/afleveringen/2014-2015/access-to-africa.html
239             'title': metadata.get('aflevering_titel') or metadata['titel'],
240             'description': metadata.get('info'),
241             'thumbnail': metadata.get('images', [{'url': None}])[-1]['url'],
242             'upload_date': unified_strdate(metadata.get('gidsdatum')),
243             'duration': parse_duration(metadata.get('tijdsduur')),
244             'formats': formats,
245             'subtitles': subtitles,
246         }
247
248
249 class NPOLiveIE(NPOBaseIE):
250     IE_NAME = 'npo.nl:live'
251     _VALID_URL = r'https?://(?:www\.)?npo\.nl/live/(?P<id>.+)'
252
253     _TEST = {
254         'url': 'http://www.npo.nl/live/npo-1',
255         'info_dict': {
256             'id': 'LI_NEDERLAND1_136692',
257             'display_id': 'npo-1',
258             'ext': 'mp4',
259             'title': 're:^Nederland 1 [0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}$',
260             'description': 'Livestream',
261             'is_live': True,
262         },
263         'params': {
264             'skip_download': True,
265         }
266     }
267
268     def _real_extract(self, url):
269         display_id = self._match_id(url)
270
271         webpage = self._download_webpage(url, display_id)
272
273         live_id = self._search_regex(
274             r'data-prid="([^"]+)"', webpage, 'live id')
275
276         metadata = self._download_json(
277             'http://e.omroep.nl/metadata/%s' % live_id,
278             display_id, transform_source=strip_jsonp)
279
280         token = self._get_token(display_id)
281
282         formats = []
283
284         streams = metadata.get('streams')
285         if streams:
286             for stream in streams:
287                 stream_type = stream.get('type').lower()
288                 # smooth streaming is not supported
289                 if stream_type in ['ss', 'ms']:
290                     continue
291                 stream_info = self._download_json(
292                     'http://ida.omroep.nl/aapi/?stream=%s&token=%s&type=jsonp'
293                     % (stream.get('url'), token),
294                     display_id, 'Downloading %s JSON' % stream_type)
295                 if stream_info.get('error_code', 0) or stream_info.get('errorcode', 0):
296                     continue
297                 stream_url = self._download_json(
298                     stream_info['stream'], display_id,
299                     'Downloading %s URL' % stream_type,
300                     'Unable to download %s URL' % stream_type,
301                     transform_source=strip_jsonp, fatal=False)
302                 if not stream_url:
303                     continue
304                 if stream_type == 'hds':
305                     f4m_formats = self._extract_f4m_formats(stream_url, display_id)
306                     # f4m downloader downloads only piece of live stream
307                     for f4m_format in f4m_formats:
308                         f4m_format['preference'] = -1
309                     formats.extend(f4m_formats)
310                 elif stream_type == 'hls':
311                     formats.extend(self._extract_m3u8_formats(stream_url, display_id, 'mp4'))
312                 else:
313                     formats.append({
314                         'url': stream_url,
315                         'preference': -10,
316                     })
317
318         self._sort_formats(formats)
319
320         return {
321             'id': live_id,
322             'display_id': display_id,
323             'title': self._live_title(metadata['titel']),
324             'description': metadata['info'],
325             'thumbnail': metadata.get('images', [{'url': None}])[-1]['url'],
326             'formats': formats,
327             'is_live': True,
328         }
329
330
331 class NPORadioIE(InfoExtractor):
332     IE_NAME = 'npo.nl:radio'
333     _VALID_URL = r'https?://(?:www\.)?npo\.nl/radio/(?P<id>[^/]+)/?$'
334
335     _TEST = {
336         'url': 'http://www.npo.nl/radio/radio-1',
337         'info_dict': {
338             'id': 'radio-1',
339             'ext': 'mp3',
340             'title': 're:^NPO Radio 1 [0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}$',
341             'is_live': True,
342         },
343         'params': {
344             'skip_download': True,
345         }
346     }
347
348     @staticmethod
349     def _html_get_attribute_regex(attribute):
350         return r'{0}\s*=\s*\'([^\']+)\''.format(attribute)
351
352     def _real_extract(self, url):
353         video_id = self._match_id(url)
354
355         webpage = self._download_webpage(url, video_id)
356
357         title = self._html_search_regex(
358             self._html_get_attribute_regex('data-channel'), webpage, 'title')
359
360         stream = self._parse_json(
361             self._html_search_regex(self._html_get_attribute_regex('data-streams'), webpage, 'data-streams'),
362             video_id)
363
364         codec = stream.get('codec')
365
366         return {
367             'id': video_id,
368             'url': stream['url'],
369             'title': self._live_title(title),
370             'acodec': codec,
371             'ext': codec,
372             'is_live': True,
373         }
374
375
376 class NPORadioFragmentIE(InfoExtractor):
377     IE_NAME = 'npo.nl:radio:fragment'
378     _VALID_URL = r'https?://(?:www\.)?npo\.nl/radio/[^/]+/fragment/(?P<id>\d+)'
379
380     _TEST = {
381         'url': 'http://www.npo.nl/radio/radio-5/fragment/174356',
382         'md5': 'dd8cc470dad764d0fdc70a9a1e2d18c2',
383         'info_dict': {
384             'id': '174356',
385             'ext': 'mp3',
386             'title': 'Jubileumconcert Willeke Alberti',
387         },
388     }
389
390     def _real_extract(self, url):
391         audio_id = self._match_id(url)
392
393         webpage = self._download_webpage(url, audio_id)
394
395         title = self._html_search_regex(
396             r'href="/radio/[^/]+/fragment/%s" title="([^"]+)"' % audio_id,
397             webpage, 'title')
398
399         audio_url = self._search_regex(
400             r"data-streams='([^']+)'", webpage, 'audio url')
401
402         return {
403             'id': audio_id,
404             'url': audio_url,
405             'title': title,
406         }
407
408
409 class VPROIE(NPOIE):
410     _VALID_URL = r'https?://(?:www\.)?(?:tegenlicht\.)?vpro\.nl/(?:[^/]+/){2,}(?P<id>[^/]+)\.html'
411
412     _TESTS = [
413         {
414             'url': 'http://tegenlicht.vpro.nl/afleveringen/2012-2013/de-toekomst-komt-uit-afrika.html',
415             'md5': 'f8065e4e5a7824068ed3c7e783178f2c',
416             'info_dict': {
417                 'id': 'VPWON_1169289',
418                 'ext': 'm4v',
419                 'title': 'De toekomst komt uit Afrika',
420                 'description': 'md5:52cf4eefbc96fffcbdc06d024147abea',
421                 'upload_date': '20130225',
422             },
423         },
424         {
425             'url': 'http://www.vpro.nl/programmas/2doc/2015/sergio-herman.html',
426             'info_dict': {
427                 'id': 'sergio-herman',
428                 'title': 'Sergio Herman: Fucking perfect',
429             },
430             'playlist_count': 2,
431         },
432         {
433             # playlist with youtube embed
434             'url': 'http://www.vpro.nl/programmas/2doc/2015/education-education.html',
435             'info_dict': {
436                 'id': 'education-education',
437                 'title': '2Doc',
438             },
439             'playlist_count': 2,
440         }
441     ]
442
443     def _real_extract(self, url):
444         playlist_id = self._match_id(url)
445
446         webpage = self._download_webpage(url, playlist_id)
447
448         entries = [
449             self.url_result('npo:%s' % video_id if not video_id.startswith('http') else video_id)
450             for video_id in re.findall(r'data-media-id="([^"]+)"', webpage)
451         ]
452
453         playlist_title = self._search_regex(
454             r'<title>\s*([^>]+?)\s*-\s*Teledoc\s*-\s*VPRO\s*</title>',
455             webpage, 'playlist title', default=None) or self._og_search_title(webpage)
456
457         return self.playlist_result(entries, playlist_id, playlist_title)
458
459
460 class WNLIE(InfoExtractor):
461     _VALID_URL = r'https?://(?:www\.)?omroepwnl\.nl/video/detail/(?P<id>[^/]+)__\d+'
462
463     _TEST = {
464         'url': 'http://www.omroepwnl.nl/video/detail/vandaag-de-dag-6-mei__060515',
465         'info_dict': {
466             'id': 'vandaag-de-dag-6-mei',
467             'title': 'Vandaag de Dag 6 mei',
468         },
469         'playlist_count': 4,
470     }
471
472     def _real_extract(self, url):
473         playlist_id = self._match_id(url)
474
475         webpage = self._download_webpage(url, playlist_id)
476
477         entries = [
478             self.url_result('npo:%s' % video_id, 'NPO')
479             for video_id, part in re.findall(
480                 r'<a[^>]+href="([^"]+)"[^>]+class="js-mid"[^>]*>(Deel \d+)', webpage)
481         ]
482
483         playlist_title = self._html_search_regex(
484             r'(?s)<h1[^>]+class="subject"[^>]*>(.+?)</h1>',
485             webpage, 'playlist title')
486
487         return self.playlist_result(entries, playlist_id, playlist_title)