[ndr] add support for embeds
[youtube-dl] / youtube_dl / extractor / ndr.py
1 # encoding: utf-8
2 from __future__ import unicode_literals
3
4 from .common import InfoExtractor
5 from ..utils import (
6     ExtractorError,
7     int_or_none,
8 )
9
10
11 preferences = {'xl': 4, 'l': 3, 'm': 2, 's': 1, 'xs': 0,}
12
13
14 class NDRBaseIE(InfoExtractor):
15
16     def extract_video_info(self, playlist, video_id):
17         formats = []
18         streamType = playlist.get('config').get('streamType')
19         if streamType == 'httpVideo':
20             for key, f in playlist.items():
21                 if key != 'config':
22                     src = f['src']
23                     if '.f4m' in src:
24                         formats.extend(self._extract_f4m_formats(src, video_id))
25                     elif '.m3u8' in src:
26                         formats.extend(self._extract_m3u8_formats(src, video_id, fatal=False))
27                     else:
28                         quality = f.get('quality')
29                         formats.append({
30                             'url': src,
31                             'format_id': quality,
32                             'preference': preferences.get(quality),
33                         })
34         elif streamType == 'httpAudio':
35             for key, f in playlist.items():
36                 if key != 'config':
37                     formats.append({
38                         'url': f['src'],
39                         'format_id': 'mp3',
40                         'vcodec': 'none',
41                     })
42         else:
43             raise ExtractorError('No media links available for %s' % video_id)
44
45         self._sort_formats(formats)
46
47         config = playlist.get('config')
48
49         title = config['title']
50         duration = int_or_none(config.get('duration'))
51         thumbnails = [{
52             'id': thumbnail.get('quality'),
53             'url': thumbnail.get('src'),
54             'preference': preferences.get(thumbnail.get('quality'))
55         } for thumbnail in config.get('poster').values()]
56
57         return {
58             'id': video_id,
59             'title': title,
60             'thumbnails': thumbnails,
61             'duration': duration,
62             'formats': formats,
63         }
64
65     def _real_extract(self, url):
66         video_id = self._match_id(url)
67
68         json_data = self._download_json('http://www.ndr.de/%s-ppjson.json' % video_id, video_id, fatal=False)
69
70         if not json_data:
71             webpage = self._download_webpage(url, video_id)
72             embed_url = self._html_search_regex(r'<iframe[^>]+id="pp_\w+"[^>]+src="(/.*)"', webpage, 'embed url', None, False)
73             if not embed_url:
74                 embed_url = self._html_search_meta('embedURL', webpage, fatal=False)
75             if embed_url:
76                 if embed_url.startswith('/'):
77                     return self.url_result('http://www.ndr.de%s' % embed_url, 'NDREmbed')
78                 else:
79                     return self.url_result(embed_url, 'NDREmbed')
80             raise ExtractorError('No media links available for %s' % video_id)
81
82         return self.extract_video_info(json_data['playlist'], video_id)
83
84
85 class NDRIE(NDRBaseIE):
86     IE_NAME = 'ndr'
87     IE_DESC = 'NDR.de - Mediathek'
88     _VALID_URL = r'https?://www\.ndr\.de/.+?,(?P<id>\w+)\.html'
89
90     _TESTS = [
91         {
92             'url': 'http://www.ndr.de/fernsehen/sendungen/nordmagazin/Kartoffeltage-in-der-Lewitz,nordmagazin25866.html',
93             'md5': '5bc5f5b92c82c0f8b26cddca34f8bb2c',
94             'note': 'Video file',
95             'info_dict': {
96                 'id': 'nordmagazin25866',
97                 'ext': 'mp4',
98                 'title': 'Kartoffeltage in der Lewitz',
99                 'duration': 166,
100             },
101             'skip': '404 Not found',
102         },
103         {
104             'url': 'http://www.ndr.de/fernsehen/Party-Poette-und-Parade,hafengeburtstag988.html',
105             'md5': 'dadc003c55ae12a5d2f6bd436cd73f59',
106             'info_dict': {
107                 'id': 'hafengeburtstag988',
108                 'ext': 'mp4',
109                 'title': 'Party, Pötte und Parade',
110                 'duration': 3498,
111             },
112         },
113         {
114             'url': 'http://www.ndr.de/info/La-Valette-entgeht-der-Hinrichtung,audio51535.html',
115             'md5': 'bb3cd38e24fbcc866d13b50ca59307b8',
116             'note': 'Audio file',
117             'info_dict': {
118                 'id': 'audio51535',
119                 'ext': 'mp3',
120                 'title': 'La Valette entgeht der Hinrichtung',
121                 'duration': 884,
122             }
123         }
124     ]
125
126
127 class NJoyIE(NDRBaseIE):
128     IE_NAME = 'N-JOY'
129     _VALID_URL = r'https?://www\.n-joy\.de/.+?,(?P<id>\w+)\.html'
130
131     _TEST = {
132         'url': 'http://www.n-joy.de/entertainment/comedy/comedy_contest/Benaissa-beim-NDR-Comedy-Contest,comedycontest2480.html',
133         'md5': 'cb63be60cd6f9dd75218803146d8dc67',
134         'info_dict': {
135             'id': 'comedycontest2480',
136             'ext': 'mp4',
137             'title': 'Benaissa beim NDR Comedy Contest',
138             'duration': 654,
139         }
140     }
141
142
143 class NDREmbedBaseIE(NDRBaseIE):
144
145     def _real_extract(self, url):
146         video_id = self._match_id(url)
147         json_data = self._download_json('http://www.ndr.de/%s-ppjson.json' % video_id, video_id, fatal=False)
148         if not json_data:
149             raise ExtractorError('No media links available for %s' % video_id)
150         return self.extract_video_info(json_data['playlist'], video_id)
151
152
153 class NDREmbedIE(NDREmbedBaseIE):
154     IE_NAME = 'ndr:embed'
155     _VALID_URL = r'https?://www\.ndr\.de/(?:[^/]+/)+(?P<id>\w+)'
156
157     _TEST = {
158         'url': 'http://www.ndr.de/fernsehen/sendungen/ndr_aktuell/ndraktuell28488-player.html',
159         'md5': 'cb63be60cd6f9dd75218803146d8dc67',
160         'info_dict': {
161             'id': 'ndraktuell28488',
162             'ext': 'mp4',
163             'title': 'Norddeutschland begrüßt Flüchtlinge',
164             'duration': 132,
165         }
166     }
167
168
169 class NJoyEmbedIE(NDREmbedBaseIE):
170     IE_NAME = 'N-JOY:embed'
171     _VALID_URL = r'https?://www\.n-joy\.de/(?:[^/]+/)(?P<id>\w+)'
172
173     _TEST = {
174         'url': 'http://www.n-joy.de/entertainment/film/portraet374-player_image-832d9b79-fa8a-4026-92e2-e0fd99deb2f9_theme-n-joy.html',
175         'md5': 'cb63be60cd6f9dd75218803146d8dc67',
176         'info_dict': {
177             'id': 'portraet374',
178             'ext': 'mp4',
179             'title': 'Viviane Andereggen - "Schuld um Schuld"',
180             'duration': 129,
181         }
182     }