[wdr] Simplify extraction
[youtube-dl] / youtube_dl / extractor / wdr.py
1 # -*- coding: utf-8 -*-
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7 from ..utils import (
8     determine_ext,
9     js_to_json,
10     strip_jsonp,
11     unified_strdate,
12     ExtractorError,
13     update_url_query,
14     urlhandle_detect_ext,
15 )
16
17
18 class WDRIE(InfoExtractor):
19     _CURRENT_MAUS_URL = r'https?://(?:www\.)wdrmaus.de/(?:[^/]+/){1,2}[^/?#]+\.php5'
20     _PAGE_REGEX = r'/(?:mediathek/)?(?P<media_type>[^/]+)/(?P<type>[^/]+)/(?P<display_id>.+)\.html'
21     _VALID_URL = r'(?P<page_url>https?://(?:www\d\.)?wdr\d?\.de)' + _PAGE_REGEX + '|' + _CURRENT_MAUS_URL
22
23     _TESTS = [
24         {
25             'url': 'http://www1.wdr.de/mediathek/video/sendungen/doku-am-freitag/video-geheimnis-aachener-dom-100.html',
26             # HDS download, MD5 is unstable
27             'info_dict': {
28                 'id': 'mdb-1058683',
29                 'ext': 'flv',
30                 'display_id': 'doku-am-freitag/video-geheimnis-aachener-dom-100',
31                 'title': 'Geheimnis Aachener Dom',
32                 'alt_title': 'Doku am Freitag',
33                 'upload_date': '20160304',
34                 'description': 'md5:87be8ff14d8dfd7a7ee46f0299b52318',
35                 'is_live': False,
36                 'subtitles': {'de': [{
37                     'url': 'http://ondemand-ww.wdr.de/medp/fsk0/105/1058683/1058683_12220974.xml'
38                 }]},
39             },
40         },
41         {
42             'url': 'http://www1.wdr.de/mediathek/audio/wdr3/wdr3-gespraech-am-samstag/audio-schriftstellerin-juli-zeh-100.html',
43             'md5': 'f4c1f96d01cf285240f53ea4309663d8',
44             'info_dict': {
45                 'id': 'mdb-1072000',
46                 'ext': 'mp3',
47                 'display_id': 'wdr3-gespraech-am-samstag/audio-schriftstellerin-juli-zeh-100',
48                 'title': 'Schriftstellerin Juli Zeh',
49                 'alt_title': 'WDR 3 Gespräch am Samstag',
50                 'upload_date': '20160312',
51                 'description': 'md5:e127d320bc2b1f149be697ce044a3dd7',
52                 'is_live': False,
53                 'subtitles': {}
54             },
55         },
56         {
57             'url': 'http://www1.wdr.de/mediathek/video/live/index.html',
58             'info_dict': {
59                 'id': 'mdb-103364',
60                 'ext': 'mp4',
61                 'display_id': 'index',
62                 'title': r're:^WDR Fernsehen im Livestream [0-9]{4}-[0-9]{2}-[0-9]{2} [0-9]{2}:[0-9]{2}$',
63                 'alt_title': 'WDR Fernsehen Live',
64                 'upload_date': None,
65                 'description': 'md5:ae2ff888510623bf8d4b115f95a9b7c9',
66                 'is_live': True,
67                 'subtitles': {}
68             },
69             'params': {
70                 'skip_download': True,  # m3u8 download
71             },
72         },
73         {
74             'url': 'http://www1.wdr.de/mediathek/video/sendungen/aktuelle-stunde/aktuelle-stunde-120.html',
75             'playlist_mincount': 8,
76             'info_dict': {
77                 'id': 'aktuelle-stunde/aktuelle-stunde-120',
78             },
79         },
80         {
81             'url': 'http://www.wdrmaus.de/aktuelle-sendung/index.php5',
82             'info_dict': {
83                 'id': 'mdb-1096487',
84                 'ext': 'flv',
85                 'upload_date': 're:^[0-9]{8}$',
86                 'title': 're:^Die Sendung mit der Maus vom [0-9.]{10}$',
87                 'description': '- Die Sendung mit der Maus -',
88             },
89             'skip': 'The id changes from week to week because of the new episode'
90         },
91         {
92             'url': 'http://www.wdrmaus.de/sachgeschichten/sachgeschichten/achterbahn.php5',
93             # HDS download, MD5 is unstable
94             'info_dict': {
95                 'id': 'mdb-186083',
96                 'ext': 'flv',
97                 'upload_date': '20130919',
98                 'title': 'Sachgeschichte - Achterbahn ',
99                 'description': '- Die Sendung mit der Maus -',
100             },
101         },
102         {
103             'url': 'http://www1.wdr.de/radio/player/radioplayer116~_layout-popupVersion.html',
104             # Live stream, MD5 unstable
105             'info_dict': {
106                 'id': 'mdb-869971',
107                 'ext': 'flv',
108                 'title': 'Funkhaus Europa Livestream',
109                 'description': 'md5:2309992a6716c347891c045be50992e4',
110                 'upload_date': '20160101',
111             },
112         }
113     ]
114
115     def _real_extract(self, url):
116         mobj = re.match(self._VALID_URL, url)
117         url_type = mobj.group('type')
118         page_url = mobj.group('page_url')
119         display_id = mobj.group('display_id')
120         webpage = self._download_webpage(url, display_id)
121
122         # for wdr.de the data-extension is in a tag with the class "mediaLink"
123         # for wdr.de radio players, in a tag with the class "wdrrPlayerPlayBtn"
124         # for wdrmaus its in a link to the page in a multiline "videoLink"-tag
125         json_metadata = self._html_search_regex(
126             r'class=(?:"(?:mediaLink|wdrrPlayerPlayBtn)\b[^"]*"[^>]+|"videoLink\b[^"]*"[\s]*>\n[^\n]*)data-extension="([^"]+)"',
127             webpage, 'media link', default=None, flags=re.MULTILINE)
128
129         if not json_metadata:
130             entries = [
131                 self.url_result(page_url + href[0], 'WDR')
132                 for href in re.findall(
133                     r'<a href="(%s)"[^>]+data-extension=' % self._PAGE_REGEX,
134                     webpage)
135             ]
136
137             if entries:  # Playlist page
138                 return self.playlist_result(entries, playlist_id=display_id)
139
140             raise ExtractorError('No downloadable streams found', expected=True)
141
142         media_link_obj = self._parse_json(json_metadata, display_id,
143                                           transform_source=js_to_json)
144         jsonp_url = media_link_obj['mediaObj']['url']
145
146         metadata = self._download_json(
147             jsonp_url, 'metadata', transform_source=strip_jsonp)
148
149         metadata_tracker_data = metadata['trackerData']
150         metadata_media_resource = metadata['mediaResource']
151
152         formats = []
153
154         # check if the metadata contains a direct URL to a file
155         for kind, media_resource in metadata_media_resource.items():
156             if kind not in ('dflt', 'alt'):
157                 continue
158
159             for tag_name, medium_url in media_resource.items():
160                 if tag_name not in ('videoURL', 'audioURL'):
161                     continue
162
163                 ext = determine_ext(medium_url)
164                 if ext == 'm3u8':
165                     m3u_fmt = self._extract_m3u8_formats(
166                         medium_url, display_id, 'mp4', 'm3u8_native',
167                         m3u8_id='hls')
168                     formats.extend(m3u_fmt)
169                 elif ext == 'f4m':
170                     manifest_url = update_url_query(
171                         medium_url, {'hdcore': '3.2.0', 'plugin': 'aasp-3.2.0.77.18'})
172                     formats.extend(self._extract_f4m_formats(
173                         manifest_url, display_id, f4m_id='hds', fatal=False))
174                 elif ext == 'smil':
175                     formats.extend(self._extract_smil_formats(
176                         medium_url, 'stream', fatal=False))
177                 else:
178                     a_format = {
179                         'url': medium_url
180                     }
181                     if ext == 'unknown_video':
182                         urlh = self._request_webpage(
183                             medium_url, display_id, note='Determining extension')
184                         ext = urlhandle_detect_ext(urlh)
185                         a_format['ext'] = ext
186                     formats.append(a_format)
187
188         subtitles = {}
189         caption_url = metadata_media_resource.get('captionURL')
190         if caption_url:
191             subtitles['de'] = [{
192                 'url': caption_url
193             }]
194
195         title = metadata_tracker_data.get('trackerClipTitle')
196         is_live = url_type == 'live'
197
198         if is_live:
199             title = self._live_title(title)
200             upload_date = None
201         elif 'trackerClipAirTime' in metadata_tracker_data:
202             upload_date = metadata_tracker_data['trackerClipAirTime']
203         else:
204             upload_date = self._html_search_meta('DC.Date', webpage, 'upload date')
205
206         if upload_date:
207             upload_date = unified_strdate(upload_date)
208
209         self._sort_formats(formats)
210
211         return {
212             'id': metadata_tracker_data.get('trackerClipId', display_id),
213             'display_id': display_id,
214             'title': title,
215             'alt_title': metadata_tracker_data.get('trackerClipSubcategory'),
216             'formats': formats,
217             'upload_date': upload_date,
218             'description': self._html_search_meta('Description', webpage),
219             'is_live': is_live,
220             'subtitles': subtitles,
221         }
222
223
224 class WDRMobileIE(InfoExtractor):
225     _VALID_URL = r'''(?x)
226         https?://mobile-ondemand\.wdr\.de/
227         .*?/fsk(?P<age_limit>[0-9]+)
228         /[0-9]+/[0-9]+/
229         (?P<id>[0-9]+)_(?P<title>[0-9]+)'''
230     IE_NAME = 'wdr:mobile'
231     _TEST = {
232         'url': 'http://mobile-ondemand.wdr.de/CMS2010/mdb/ondemand/weltweit/fsk0/42/421735/421735_4283021.mp4',
233         'info_dict': {
234             'title': '4283021',
235             'id': '421735',
236             'ext': 'mp4',
237             'age_limit': 0,
238         },
239         'skip': 'Problems with loading data.'
240     }
241
242     def _real_extract(self, url):
243         mobj = re.match(self._VALID_URL, url)
244         return {
245             'id': mobj.group('id'),
246             'title': mobj.group('title'),
247             'age_limit': int(mobj.group('age_limit')),
248             'url': url,
249             'http_headers': {
250                 'User-Agent': 'mobile',
251             },
252         }