[npo] Improve formats extraction (Closes #3782)
[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 ..utils import (
7     unified_strdate,
8     parse_duration,
9     qualities,
10 )
11
12
13 class NPOIE(InfoExtractor):
14     IE_NAME = 'npo.nl'
15     _VALID_URL = r'https?://www\.npo\.nl/[^/]+/[^/]+/(?P<id>[^/?]+)'
16
17     _TESTS = [
18         {
19             'url': 'http://www.npo.nl/nieuwsuur/22-06-2014/VPWON_1220719',
20             'md5': '4b3f9c429157ec4775f2c9cb7b911016',
21             'info_dict': {
22                 'id': 'VPWON_1220719',
23                 'ext': 'm4v',
24                 'title': 'Nieuwsuur',
25                 'description': 'Dagelijks tussen tien en elf: nieuws, sport en achtergronden.',
26                 'upload_date': '20140622',
27             },
28         },
29         {
30             'url': 'http://www.npo.nl/de-mega-mike-mega-thomas-show/27-02-2009/VARA_101191800',
31             'md5': 'da50a5787dbfc1603c4ad80f31c5120b',
32             'info_dict': {
33                 'id': 'VARA_101191800',
34                 'ext': 'm4v',
35                 'title': 'De Mega Mike & Mega Thomas show',
36                 'description': 'md5:3b74c97fc9d6901d5a665aac0e5400f4',
37                 'upload_date': '20090227',
38                 'duration': 2400,
39             },
40         },
41         {
42             'url': 'http://www.npo.nl/tegenlicht/25-02-2013/VPWON_1169289',
43             'md5': 'f8065e4e5a7824068ed3c7e783178f2c',
44             'info_dict': {
45                 'id': 'VPWON_1169289',
46                 'ext': 'm4v',
47                 'title': 'Tegenlicht',
48                 'description': 'md5:d6476bceb17a8c103c76c3b708f05dd1',
49                 'upload_date': '20130225',
50                 'duration': 3000,
51             },
52         }
53     ]
54
55     def _real_extract(self, url):
56         mobj = re.match(self._VALID_URL, url)
57         video_id = mobj.group('id')
58
59         metadata = self._download_json(
60             'http://e.omroep.nl/metadata/aflevering/%s' % video_id,
61             video_id,
62             # We have to remove the javascript callback
63             transform_source=lambda j: re.sub(r'parseMetadata\((.*?)\);\n//.*$', r'\1', j)
64         )
65         token_page = self._download_webpage(
66             'http://ida.omroep.nl/npoplayer/i.js',
67             video_id,
68             note='Downloading token'
69         )
70         token = self._search_regex(r'npoplayer\.token = "(.+?)"', token_page, 'token')
71
72         formats = []
73         quality = qualities(['adaptive', 'wmv_sb', 'h264_sb', 'wmv_bb', 'h264_bb', 'wvc1_std', 'h264_std'])
74         for format_id in metadata['pubopties']:
75             format_info = self._download_json(
76                 'http://ida.omroep.nl/odi/?prid=%s&puboptions=%s&adaptive=yes&token=%s' % (video_id, format_id, token),
77                 video_id, 'Downloading %s JSON' % format_id)
78             if format_info.get('error_code', 0) or format_info.get('errorcode', 0):
79                 continue
80             streams = format_info.get('streams')
81             if streams:
82                 video_info = self._download_json(
83                     streams[0] + '&type=json',
84                     video_id, 'Downloading %s stream JSON' % format_id)
85             else:
86                 video_info = format_info
87             video_url = video_info.get('url')
88             if not video_url:
89                 continue
90             if format_id == 'adaptive':
91                 formats.extend(self._extract_m3u8_formats(video_url, video_id))
92             else:
93                 formats.append({
94                     'url': video_url,
95                     'format_id': format_id,
96                     'quality': quality(format_id),
97                 })
98         self._sort_formats(formats)
99
100         return {
101             'id': video_id,
102             'title': metadata['titel'],
103             'description': metadata['info'],
104             'thumbnail': metadata.get('images', [{'url': None}])[-1]['url'],
105             'upload_date': unified_strdate(metadata.get('gidsdatum')),
106             'duration': parse_duration(metadata.get('tijdsduur')),
107             'formats': formats,
108         }