]> git.bitcoin.ninja Git - youtube-dl/blob - youtube_dl/extractor/ceskatelevize.py
[brightcove:new] Add ability to smuggle geo_countries into URL
[youtube-dl] / youtube_dl / extractor / ceskatelevize.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5
6 from .common import InfoExtractor
7 from ..compat import (
8     compat_urllib_parse_unquote,
9     compat_urllib_parse_urlparse,
10 )
11 from ..utils import (
12     ExtractorError,
13     float_or_none,
14     sanitized_Request,
15     urlencode_postdata,
16     USER_AGENTS,
17 )
18
19
20 class CeskaTelevizeIE(InfoExtractor):
21     _VALID_URL = r'https?://(?:www\.)?ceskatelevize\.cz/(porady|ivysilani)/(?:[^/]+/)*(?P<id>[^/#?]+)/*(?:[#?].*)?$'
22     _TESTS = [{
23         'url': 'http://www.ceskatelevize.cz/ivysilani/ivysilani/10441294653-hyde-park-civilizace/214411058091220',
24         'info_dict': {
25             'id': '61924494877246241',
26             'ext': 'mp4',
27             'title': 'Hyde Park Civilizace: Život v Grónsku',
28             'description': 'md5:3fec8f6bb497be5cdb0c9e8781076626',
29             'thumbnail': r're:^https?://.*\.jpg',
30             'duration': 3350,
31         },
32         'params': {
33             # m3u8 download
34             'skip_download': True,
35         },
36     }, {
37         'url': 'http://www.ceskatelevize.cz/ivysilani/10441294653-hyde-park-civilizace/215411058090502/bonus/20641-bonus-01-en',
38         'info_dict': {
39             'id': '61924494877028507',
40             'ext': 'mp4',
41             'title': 'Hyde Park Civilizace: Bonus 01 - En',
42             'description': 'English Subtittles',
43             'thumbnail': r're:^https?://.*\.jpg',
44             'duration': 81.3,
45         },
46         'params': {
47             # m3u8 download
48             'skip_download': True,
49         },
50     }, {
51         # live stream
52         'url': 'http://www.ceskatelevize.cz/ivysilani/zive/ct4/',
53         'info_dict': {
54             'id': 402,
55             'ext': 'mp4',
56             'title': r're:^ČT Sport \d{4}-\d{2}-\d{2} \d{2}:\d{2}$',
57             'is_live': True,
58         },
59         'params': {
60             # m3u8 download
61             'skip_download': True,
62         },
63         'skip': 'Georestricted to Czech Republic',
64     }, {
65         # video with 18+ caution trailer
66         'url': 'http://www.ceskatelevize.cz/porady/10520528904-queer/215562210900007-bogotart/',
67         'info_dict': {
68             'id': '215562210900007-bogotart',
69             'title': 'Queer: Bogotart',
70             'description': 'Alternativní průvodce současným queer světem',
71         },
72         'playlist': [{
73             'info_dict': {
74                 'id': '61924494876844842',
75                 'ext': 'mp4',
76                 'title': 'Queer: Bogotart (Varování 18+)',
77                 'duration': 10.2,
78             },
79         }, {
80             'info_dict': {
81                 'id': '61924494877068022',
82                 'ext': 'mp4',
83                 'title': 'Queer: Bogotart (Queer)',
84                 'thumbnail': r're:^https?://.*\.jpg',
85                 'duration': 1558.3,
86             },
87         }],
88         'params': {
89             # m3u8 download
90             'skip_download': True,
91         },
92     }]
93
94     def _real_extract(self, url):
95         url = url.replace('/porady/', '/ivysilani/').replace('/video/', '')
96
97         mobj = re.match(self._VALID_URL, url)
98         playlist_id = mobj.group('id')
99
100         webpage = self._download_webpage(url, playlist_id)
101
102         NOT_AVAILABLE_STRING = 'This content is not available at your territory due to limited copyright.'
103         if '%s</p>' % NOT_AVAILABLE_STRING in webpage:
104             raise ExtractorError(NOT_AVAILABLE_STRING, expected=True)
105
106         typ = self._html_search_regex(
107             r'getPlaylistUrl\(\[\{"type":"(.+?)","id":".+?"\}\],', webpage, 'type')
108         episode_id = self._html_search_regex(
109             r'getPlaylistUrl\(\[\{"type":".+?","id":"(.+?)"\}\],', webpage, 'episode_id')
110
111         data = {
112             'playlist[0][type]': typ,
113             'playlist[0][id]': episode_id,
114             'requestUrl': compat_urllib_parse_urlparse(url).path,
115             'requestSource': 'iVysilani',
116         }
117
118         entries = []
119
120         for user_agent in (None, USER_AGENTS['Safari']):
121             req = sanitized_Request(
122                 'http://www.ceskatelevize.cz/ivysilani/ajax/get-client-playlist',
123                 data=urlencode_postdata(data))
124
125             req.add_header('Content-type', 'application/x-www-form-urlencoded')
126             req.add_header('x-addr', '127.0.0.1')
127             req.add_header('X-Requested-With', 'XMLHttpRequest')
128             if user_agent:
129                 req.add_header('User-Agent', user_agent)
130             req.add_header('Referer', url)
131
132             playlistpage = self._download_json(req, playlist_id, fatal=False)
133
134             if not playlistpage:
135                 continue
136
137             playlist_url = playlistpage['url']
138             if playlist_url == 'error_region':
139                 raise ExtractorError(NOT_AVAILABLE_STRING, expected=True)
140
141             req = sanitized_Request(compat_urllib_parse_unquote(playlist_url))
142             req.add_header('Referer', url)
143
144             playlist_title = self._og_search_title(webpage, default=None)
145             playlist_description = self._og_search_description(webpage, default=None)
146
147             playlist = self._download_json(req, playlist_id, fatal=False)
148             if not playlist:
149                 continue
150
151             playlist = playlist.get('playlist')
152             if not isinstance(playlist, list):
153                 continue
154
155             playlist_len = len(playlist)
156
157             for num, item in enumerate(playlist):
158                 is_live = item.get('type') == 'LIVE'
159                 formats = []
160                 for format_id, stream_url in item.get('streamUrls', {}).items():
161                     if 'playerType=flash' in stream_url:
162                         stream_formats = self._extract_m3u8_formats(
163                             stream_url, playlist_id, 'mp4',
164                             entry_protocol='m3u8' if is_live else 'm3u8_native',
165                             m3u8_id='hls-%s' % format_id, fatal=False)
166                     else:
167                         stream_formats = self._extract_mpd_formats(
168                             stream_url, playlist_id,
169                             mpd_id='dash-%s' % format_id, fatal=False)
170                     # See https://github.com/rg3/youtube-dl/issues/12119#issuecomment-280037031
171                     if format_id == 'audioDescription':
172                         for f in stream_formats:
173                             f['source_preference'] = -10
174                     formats.extend(stream_formats)
175
176                 if user_agent and len(entries) == playlist_len:
177                     entries[num]['formats'].extend(formats)
178                     continue
179
180                 item_id = item.get('id') or item['assetId']
181                 title = item['title']
182
183                 duration = float_or_none(item.get('duration'))
184                 thumbnail = item.get('previewImageUrl')
185
186                 subtitles = {}
187                 if item.get('type') == 'VOD':
188                     subs = item.get('subtitles')
189                     if subs:
190                         subtitles = self.extract_subtitles(episode_id, subs)
191
192                 if playlist_len == 1:
193                     final_title = playlist_title or title
194                     if is_live:
195                         final_title = self._live_title(final_title)
196                 else:
197                     final_title = '%s (%s)' % (playlist_title, title)
198
199                 entries.append({
200                     'id': item_id,
201                     'title': final_title,
202                     'description': playlist_description if playlist_len == 1 else None,
203                     'thumbnail': thumbnail,
204                     'duration': duration,
205                     'formats': formats,
206                     'subtitles': subtitles,
207                     'is_live': is_live,
208                 })
209
210         for e in entries:
211             self._sort_formats(e['formats'])
212
213         return self.playlist_result(entries, playlist_id, playlist_title, playlist_description)
214
215     def _get_subtitles(self, episode_id, subs):
216         original_subtitles = self._download_webpage(
217             subs[0]['url'], episode_id, 'Downloading subtitles')
218         srt_subs = self._fix_subtitles(original_subtitles)
219         return {
220             'cs': [{
221                 'ext': 'srt',
222                 'data': srt_subs,
223             }]
224         }
225
226     @staticmethod
227     def _fix_subtitles(subtitles):
228         """ Convert millisecond-based subtitles to SRT """
229
230         def _msectotimecode(msec):
231             """ Helper utility to convert milliseconds to timecode """
232             components = []
233             for divider in [1000, 60, 60, 100]:
234                 components.append(msec % divider)
235                 msec //= divider
236             return '{3:02}:{2:02}:{1:02},{0:03}'.format(*components)
237
238         def _fix_subtitle(subtitle):
239             for line in subtitle.splitlines():
240                 m = re.match(r'^\s*([0-9]+);\s*([0-9]+)\s+([0-9]+)\s*$', line)
241                 if m:
242                     yield m.group(1)
243                     start, stop = (_msectotimecode(int(t)) for t in m.groups()[1:])
244                     yield '{0} --> {1}'.format(start, stop)
245                 else:
246                     yield line
247
248         return '\r\n'.join(_fix_subtitle(subtitles))