[spiegel] Modernize
[youtube-dl] / youtube_dl / extractor / spiegel.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 compat_urlparse
8
9
10 class SpiegelIE(InfoExtractor):
11     _VALID_URL = r'https?://(?:www\.)?spiegel\.de/video/[^/]*-(?P<id>[0-9]+)(?:\.html)?(?:#.*)?$'
12     _TESTS = [{
13         'url': 'http://www.spiegel.de/video/vulkan-tungurahua-in-ecuador-ist-wieder-aktiv-video-1259285.html',
14         'md5': '2c2754212136f35fb4b19767d242f66e',
15         'info_dict': {
16             'id': '1259285',
17             'ext': 'mp4',
18             'title': 'Vulkanausbruch in Ecuador: Der "Feuerschlund" ist wieder aktiv',
19             'description': 'md5:8029d8310232196eb235d27575a8b9f4',
20             'duration': 49,
21         },
22     }, {
23         'url': 'http://www.spiegel.de/video/schach-wm-videoanalyse-des-fuenften-spiels-video-1309159.html',
24         'md5': 'f2cdf638d7aa47654e251e1aee360af1',
25         'info_dict': {
26             'id': '1309159',
27             'ext': 'mp4',
28             'title': 'Schach-WM in der Videoanalyse: Carlsen nutzt die Fehlgriffe des Titelverteidigers',
29             'description': 'md5:c2322b65e58f385a820c10fa03b2d088',
30             'duration': 983,
31         },
32     }]
33
34     def _real_extract(self, url):
35         video_id = self._match_id(url)
36         webpage = self._download_webpage(url, video_id)
37
38         title = self._html_search_regex(
39             r'<div class="module-title">(.*?)</div>', webpage, 'title')
40         description = self._html_search_meta('description', webpage, 'description')
41
42         base_url = self._search_regex(
43             r'var\s+server\s*=\s*"([^"]+)\"', webpage, 'server URL')
44
45         xml_url = base_url + video_id + '.xml'
46         idoc = self._download_xml(xml_url, video_id)
47
48         formats = [
49             {
50                 'format_id': n.tag.rpartition('type')[2],
51                 'url': base_url + n.find('./filename').text,
52                 'width': int(n.find('./width').text),
53                 'height': int(n.find('./height').text),
54                 'abr': int(n.find('./audiobitrate').text),
55                 'vbr': int(n.find('./videobitrate').text),
56                 'vcodec': n.find('./codec').text,
57                 'acodec': 'MP4A',
58             }
59             for n in list(idoc)
60             # Blacklist type 6, it's extremely LQ and not available on the same server
61             if n.tag.startswith('type') and n.tag != 'type6'
62         ]
63         duration = float(idoc[0].findall('./duration')[0].text)
64
65         self._sort_formats(formats)
66
67         return {
68             'id': video_id,
69             'title': title,
70             'description': description,
71             'duration': duration,
72             'formats': formats,
73         }
74
75
76 class SpiegelArticleIE(InfoExtractor):
77     _VALID_URL = 'https?://www\.spiegel\.de/(?!video/)[^?#]*?-(?P<id>[0-9]+)\.html'
78     IE_NAME = 'Spiegel:Article'
79     IE_DESC = 'Articles on spiegel.de'
80     _TEST = {
81         'url': 'http://www.spiegel.de/sport/sonst/badminton-wm-die-randsportart-soll-populaerer-werden-a-987092.html',
82         'info_dict': {
83             'id': '1516455',
84             'ext': 'mp4',
85             'title': 'Faszination Badminton: Nennt es bloß nicht Federball',
86             'description': 're:^Patrick Kämnitz gehört.{100,}',
87         },
88     }
89
90     def _real_extract(self, url):
91         video_id = self._match_id(url)
92
93         webpage = self._download_webpage(url, video_id)
94         video_link = self._search_regex(
95             r'<a href="([^"]+)" onclick="return spOpenVideo\(this,', webpage,
96             'video page URL')
97         video_url = compat_urlparse.urljoin(
98             self.http_scheme() + '//spiegel.de/', video_link)
99
100         return {
101             '_type': 'url',
102             'url': video_url,
103         }