[nova] Add extractor
[youtube-dl] / youtube_dl / extractor / nova.py
1 # encoding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7
8
9 class NovaIE(InfoExtractor):
10     IE_DESC = 'TN.cz, Prásk.tv, Nova.cz, Novaplus.cz, FANDA.tv, Krásná.cz and Doma.cz'
11     _VALID_URL = 'http://(?:[^.]+\.)?(?P<site>tv(?:noviny)?|tn|novaplus|vymena|fanda|krasna|doma|prask)\.nova\.cz/(?:[^/]+/)+(?P<id>[^/]+)(?:\.html|/?)'
12     _TESTS = [{
13         'url': 'http://tvnoviny.nova.cz/clanek/novinky/co-na-sebe-sportaci-praskli-vime-jestli-pujde-hrdlicka-na-materskou.html',
14         'info_dict': {
15             'id': '1608920',
16             'display_id': 'co-na-sebe-sportaci-praskli-vime-jestli-pujde-hrdlicka-na-materskou',
17             'ext': 'flv',
18             'title': 'Duel: Michal Hrdlička a Petr Suchoň',
19             'description': 'md5:d0cc509858eee1b1374111c588c6f5d5',
20             'thumbnail': 're:^https?://.*\.(?:jpg)',
21         },
22         'params': {
23             # rtmp download
24             'skip_download': True,
25         }
26     }, {
27         'url': 'http://tn.nova.cz/clanek/tajemstvi-ukryte-v-podzemi-specialni-nemocnice-v-prazske-krci.html',
28         'md5': '1dd7b9d5ea27bc361f110cd855a19bd3',
29         'info_dict': {
30             'id': '1757139',
31             'display_id': 'tajemstvi-ukryte-v-podzemi-specialni-nemocnice-v-prazske-krci',
32             'ext': 'mp4',
33             'title': 'Podzemní nemocnice v pražské Krči',
34             'description': 'md5:f0a42dd239c26f61c28f19e62d20ef53',
35             'thumbnail': 're:^https?://.*\.(?:jpg)',
36         }
37     }, {
38         'url': 'http://novaplus.nova.cz/porad/policie-modrava/video/5591-policie-modrava-15-dil-blondynka-na-hrbitove/',
39         'info_dict': {
40             'id': '1756825',
41             'display_id': '5591-policie-modrava-15-dil-blondynka-na-hrbitove',
42             'ext': 'flv',
43             'title': 'Policie Modrava - 15. díl - Blondýnka na hřbitově',
44             'description': 'md5:d804ba6b30bc7da2705b1fea961bddfe',
45             'thumbnail': 're:^https?://.*\.(?:jpg)',
46         },
47         'params': {
48             # rtmp download
49             'skip_download': True,
50         }
51     }, {
52         'url': 'http://sport.tn.nova.cz/clanek/sport/hokej/nhl/zivot-jde-dal-hodnotil-po-vyrazeni-z-playoff-jiri-sekac.html',
53         'only_matching': True,
54     }, {
55         'url': 'http://fanda.nova.cz/clanek/fun-and-games/krvavy-epos-zaklinac-3-divoky-hon-vychazi-vyhrajte-ho-pro-sebe.html',
56         'only_matching': True,
57     }, {
58         'url': 'http://doma.nova.cz/clanek/zdravi/prijdte-se-zapsat-do-registru-kostni-drene-jiz-ve-stredu-3-cervna.html',
59         'only_matching': True,
60     }, {
61         'url': 'http://prask.nova.cz/clanek/novinky/co-si-na-sobe-nase-hvezdy-nechaly-pojistit.html',
62         'only_matching': True,
63     }]
64
65     def _real_extract(self, url):
66         mobj = re.match(self._VALID_URL, url)
67         display_id = mobj.group('id')
68         site = mobj.group('site')
69
70         webpage = self._download_webpage(url, display_id)
71
72         video_id = self._search_regex(
73             [r"(?:media|video_id)\s*:\s*'(\d+)'",
74              r'media=(\d+)',
75              r'id="article_video_(\d+)"',
76              r'id="player_(\d+)"'],
77             webpage, 'video id')
78
79         config_url = self._search_regex(
80             r'src="(http://tn\.nova\.cz/bin/player/videojs/config\.php\?[^"]+)"',
81             webpage, 'config url', default=None)
82
83         if not config_url:
84             DEFAULT_SITE_ID = '23000'
85             SITES = {
86                 'tvnoviny': DEFAULT_SITE_ID,
87                 'novaplus': DEFAULT_SITE_ID,
88                 'vymena': DEFAULT_SITE_ID,
89                 'krasna': DEFAULT_SITE_ID,
90                 'fanda': '30',
91                 'tn': '30',
92                 'doma': '30',
93             }
94
95             site_id = self._search_regex(
96                 r'site=(\d+)', webpage, 'site id', default=None) or SITES.get(site, DEFAULT_SITE_ID)
97
98             config_url = ('http://tn.nova.cz/bin/player/videojs/config.php?site=%s&media=%s&jsVar=vjsconfig'
99                           % (site_id, video_id))
100
101         config = self._download_json(
102             config_url, display_id,
103             'Downloading config JSON',
104             transform_source=lambda s: re.sub(r'var\s+[\da-zA-Z_]+\s*=\s*({.+?});', r'\1', s))
105
106         mediafile = config['mediafile']
107         video_url = mediafile['src']
108
109         m = re.search(r'^(?P<url>rtmpe?://[^/]+/(?P<app>[^/]+?))/&*(?P<playpath>.+)$', video_url)
110         if m:
111             formats = [{
112                 'url': m.group('url'),
113                 'app': m.group('app'),
114                 'play_path': m.group('playpath'),
115                 'player_path': 'http://tvnoviny.nova.cz/static/shared/app/videojs/video-js.swf',
116                 'ext': 'flv',
117             }]
118         else:
119             formats = [{
120                 'url': video_url,
121             }]
122         self._sort_formats(formats)
123
124         title = mediafile.get('meta', {}).get('title') or self._og_search_title(webpage)
125         description = self._og_search_description(webpage)
126         thumbnail = config.get('poster')
127
128         return {
129             'id': video_id,
130             'display_id': display_id,
131             'title': title,
132             'description': description,
133             'thumbnail': thumbnail,
134             'formats': formats,
135         }