[nrktv] Add support for playlists (Closes #4656)
[youtube-dl] / youtube_dl / extractor / nrk.py
1 # encoding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7 from ..utils import (
8     ExtractorError,
9     float_or_none,
10     unified_strdate,
11 )
12
13
14 class NRKIE(InfoExtractor):
15     _VALID_URL = r'http://(?:www\.)?nrk\.no/(?:video|lyd)/[^/]+/(?P<id>[\dA-F]{16})'
16
17     _TESTS = [
18         {
19             'url': 'http://www.nrk.no/video/dompap_og_andre_fugler_i_piip_show/D0FA54B5C8B6CE59/emne/piipshow/',
20             'md5': 'a6eac35052f3b242bb6bb7f43aed5886',
21             'info_dict': {
22                 'id': '150533',
23                 'ext': 'flv',
24                 'title': 'Dompap og andre fugler i Piip-Show',
25                 'description': 'md5:d9261ba34c43b61c812cb6b0269a5c8f'
26             }
27         },
28         {
29             'url': 'http://www.nrk.no/lyd/lyd_av_oppleser_for_blinde/AEFDDD5473BA0198/',
30             'md5': '3471f2a51718195164e88f46bf427668',
31             'info_dict': {
32                 'id': '154915',
33                 'ext': 'flv',
34                 'title': 'Slik høres internett ut når du er blind',
35                 'description': 'md5:a621f5cc1bd75c8d5104cb048c6b8568',
36             }
37         },
38     ]
39
40     def _real_extract(self, url):
41         mobj = re.match(self._VALID_URL, url)
42         video_id = mobj.group('id')
43
44         page = self._download_webpage(url, video_id)
45
46         video_id = self._html_search_regex(r'<div class="nrk-video" data-nrk-id="(\d+)">', page, 'video id')
47
48         data = self._download_json(
49             'http://v7.psapi.nrk.no/mediaelement/%s' % video_id, video_id, 'Downloading media JSON')
50
51         if data['usageRights']['isGeoBlocked']:
52             raise ExtractorError('NRK har ikke rettig-heter til å vise dette programmet utenfor Norge', expected=True)
53
54         video_url = data['mediaUrl'] + '?hdcore=3.1.1&plugin=aasp-3.1.1.69.124'
55
56         images = data.get('images')
57         if images:
58             thumbnails = images['webImages']
59             thumbnails.sort(key=lambda image: image['pixelWidth'])
60             thumbnail = thumbnails[-1]['imageUrl']
61         else:
62             thumbnail = None
63
64         return {
65             'id': video_id,
66             'url': video_url,
67             'ext': 'flv',
68             'title': data['title'],
69             'description': data['description'],
70             'thumbnail': thumbnail,
71         }
72
73
74 class NRKTVIE(InfoExtractor):
75     _VALID_URL = r'http://tv\.nrk(?:super)?\.no/(?:serie/[^/]+|program)/(?P<id>[a-zA-Z]{4}\d{8})(?:/\d{2}-\d{2}-\d{4})?(?:#del=(?P<part_id>\d+))?'
76
77     _TESTS = [
78         {
79             'url': 'http://tv.nrk.no/serie/20-spoersmaal-tv/MUHH48000314/23-05-2014',
80             'md5': 'adf2c5454fa2bf032f47a9f8fb351342',
81             'info_dict': {
82                 'id': 'MUHH48000314',
83                 'ext': 'flv',
84                 'title': '20 spørsmål',
85                 'description': 'md5:bdea103bc35494c143c6a9acdd84887a',
86                 'upload_date': '20140523',
87                 'duration': 1741.52,
88             },
89         },
90         {
91             'url': 'http://tv.nrk.no/program/mdfp15000514',
92             'md5': '383650ece2b25ecec996ad7b5bb2a384',
93             'info_dict': {
94                 'id': 'mdfp15000514',
95                 'ext': 'flv',
96                 'title': 'Kunnskapskanalen: Grunnlovsjubiléet - Stor ståhei for ingenting',
97                 'description': 'md5:654c12511f035aed1e42bdf5db3b206a',
98                 'upload_date': '20140524',
99                 'duration': 4605.0,
100             },
101         },
102         {
103             # single playlist video
104             'url': 'http://tv.nrk.no/serie/tour-de-ski/MSPO40010515/06-01-2015#del=2',
105             'md5': 'adbd1dbd813edaf532b0a253780719c2',
106             'info_dict': {
107                 'id': 'MSPO40010515-part2',
108                 'ext': 'flv',
109                 'title': 'Tour de Ski: Sprint fri teknikk, kvinner og menn 06.01.2015 (del 2:2)',
110                 'description': 'md5:238b67b97a4ac7d7b4bf0edf8cc57d26',
111                 'upload_date': '20150106',
112             },
113             'skip': 'Only works from Norway',
114             'params': {
115                 'proxy': '127.0.0.1:8118',
116             },
117         },
118         {
119             'url': 'http://tv.nrk.no/serie/tour-de-ski/MSPO40010515/06-01-2015',
120             'playlist': [
121                 {
122                     'md5': '9480285eff92d64f06e02a5367970a7a',
123                     'info_dict': {
124                         'id': 'MSPO40010515-part1',
125                         'ext': 'flv',
126                         'title': 'Tour de Ski: Sprint fri teknikk, kvinner og menn 06.01.2015 (del 1:2)',
127                         'description': 'md5:238b67b97a4ac7d7b4bf0edf8cc57d26',
128                         'upload_date': '20150106',
129                     },
130                 },
131                 {
132                     'md5': 'adbd1dbd813edaf532b0a253780719c2',
133                     'info_dict': {
134                         'id': 'MSPO40010515-part2',
135                         'ext': 'flv',
136                         'title': 'Tour de Ski: Sprint fri teknikk, kvinner og menn 06.01.2015 (del 2:2)',
137                         'description': 'md5:238b67b97a4ac7d7b4bf0edf8cc57d26',
138                         'upload_date': '20150106',
139                     },
140                 },
141             ],
142             'info_dict': {
143                 'id': 'MSPO40010515',
144                 'title': 'Tour de Ski: Sprint fri teknikk, kvinner og menn',
145                 'description': 'md5:238b67b97a4ac7d7b4bf0edf8cc57d26',
146                 'upload_date': '20150106',
147                 'duration': 6947.5199999999995,
148             },
149             'skip': 'Only works from Norway',
150             'params': {
151                 'proxy': '127.0.0.1:8118',
152             },
153         }
154     ]
155
156     def _extract_f4m(self, manifest_url, video_id):
157         return self._extract_f4m_formats(manifest_url + '?hdcore=3.1.1&plugin=aasp-3.1.1.69.124', video_id)
158
159     def _real_extract(self, url):
160         mobj = re.match(self._VALID_URL, url)
161         video_id = mobj.group('id')
162         part_id = mobj.group('part_id')
163
164         webpage = self._download_webpage(url, video_id)
165
166         title = self._html_search_meta(
167             'title', webpage, 'title')
168         description = self._html_search_meta(
169             'description', webpage, 'description')
170
171         thumbnail = self._html_search_regex(
172             r'data-posterimage="([^"]+)"',
173             webpage, 'thumbnail', fatal=False)
174         upload_date = unified_strdate(self._html_search_meta(
175             'rightsfrom', webpage, 'upload date', fatal=False))
176         duration = float_or_none(self._html_search_regex(
177             r'data-duration="([^"]+)"',
178             webpage, 'duration', fatal=False))
179
180         # playlist
181         parts = re.findall(
182             r'<a href="#del=(\d+)"[^>]+data-argument="([^"]+)">([^<]+)</a>', webpage)
183         if parts:
184             entries = []
185             for current_part_id, stream_url, part_title in parts:
186                 if part_id and current_part_id != part_id:
187                     continue
188                 video_part_id = '%s-part%s' % (video_id, current_part_id)
189                 formats = self._extract_f4m(stream_url, video_part_id)
190                 entries.append({
191                     'id': video_part_id,
192                     'title': part_title,
193                     'description': description,
194                     'thumbnail': thumbnail,
195                     'upload_date': upload_date,
196                     'formats': formats,
197                 })
198             if part_id:
199                 if entries:
200                     return entries[0]
201             else:
202                 playlist = self.playlist_result(entries, video_id, title, description)
203                 playlist.update({
204                     'thumbnail': thumbnail,
205                     'upload_date': upload_date,
206                     'duration': duration,
207                 })
208                 return playlist
209
210         formats = []
211
212         f4m_url = re.search(r'data-media="([^"]+)"', webpage)
213         if f4m_url:
214             formats.extend(self._extract_f4m(f4m_url.group(1), video_id))
215
216         m3u8_url = re.search(r'data-hls-media="([^"]+)"', webpage)
217         if m3u8_url:
218             formats.extend(self._extract_m3u8_formats(m3u8_url.group(1), video_id, 'mp4'))
219
220         self._sort_formats(formats)
221
222         return {
223             'id': video_id,
224             'title': title,
225             'description': description,
226             'thumbnail': thumbnail,
227             'upload_date': upload_date,
228             'duration': duration,
229             'formats': formats,
230         }