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