Merge branch 'shahid' of https://github.com/remitamine/youtube-dl into remitamine...
[youtube-dl] / youtube_dl / extractor / mdr.py
1 from __future__ import unicode_literals
2
3 import re
4
5 from .common import InfoExtractor
6
7
8 class MDRIE(InfoExtractor):
9     _VALID_URL = r'^(?P<domain>https?://(?:www\.)?mdr\.de)/(?:.*)/(?P<type>video|audio)(?P<video_id>[^/_]+)(?:_|\.html)'
10
11     # No tests, MDR regularily deletes its videos
12     _TEST = {
13         'url': 'http://www.mdr.de/fakt/video189002.html',
14         'only_matching': True,
15     }
16
17     def _real_extract(self, url):
18         m = re.match(self._VALID_URL, url)
19         video_id = m.group('video_id')
20         domain = m.group('domain')
21
22         # determine title and media streams from webpage
23         html = self._download_webpage(url, video_id)
24
25         title = self._html_search_regex(r'<h[12]>(.*?)</h[12]>', html, 'title')
26         xmlurl = self._search_regex(
27             r'dataURL:\'(/(?:.+)/(?:video|audio)[0-9]+-avCustom.xml)', html, 'XML URL')
28
29         doc = self._download_xml(domain + xmlurl, video_id)
30         formats = []
31         for a in doc.findall('./assets/asset'):
32             url_el = a.find('./progressiveDownloadUrl')
33             if url_el is None:
34                 continue
35             abr = int(a.find('bitrateAudio').text) // 1000
36             media_type = a.find('mediaType').text
37             format = {
38                 'abr': abr,
39                 'filesize': int(a.find('fileSize').text),
40                 'url': url_el.text,
41             }
42
43             vbr_el = a.find('bitrateVideo')
44             if vbr_el is None:
45                 format.update({
46                     'vcodec': 'none',
47                     'format_id': '%s-%d' % (media_type, abr),
48                 })
49             else:
50                 vbr = int(vbr_el.text) // 1000
51                 format.update({
52                     'vbr': vbr,
53                     'width': int(a.find('frameWidth').text),
54                     'height': int(a.find('frameHeight').text),
55                     'format_id': '%s-%d' % (media_type, vbr),
56                 })
57             formats.append(format)
58         self._sort_formats(formats)
59
60         return {
61             'id': video_id,
62             'title': title,
63             'formats': formats,
64         }