Fix W504 and disable W503 (closes #20863)
[youtube-dl] / youtube_dl / extractor / kaltura.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import re
5 import base64
6
7 from .common import InfoExtractor
8 from ..compat import (
9     compat_urlparse,
10     compat_parse_qs,
11 )
12 from ..utils import (
13     clean_html,
14     ExtractorError,
15     int_or_none,
16     unsmuggle_url,
17     smuggle_url,
18 )
19
20
21 class KalturaIE(InfoExtractor):
22     _VALID_URL = r'''(?x)
23                 (?:
24                     kaltura:(?P<partner_id>\d+):(?P<id>[0-9a-z_]+)|
25                     https?://
26                         (:?(?:www|cdnapi(?:sec)?)\.)?kaltura\.com(?::\d+)?/
27                         (?:
28                             (?:
29                                 # flash player
30                                 index\.php/(?:kwidget|extwidget/preview)|
31                                 # html5 player
32                                 html5/html5lib/[^/]+/mwEmbedFrame\.php
33                             )
34                         )(?:/(?P<path>[^?]+))?(?:\?(?P<query>.*))?
35                 )
36                 '''
37     _SERVICE_URL = 'http://cdnapi.kaltura.com'
38     _SERVICE_BASE = '/api_v3/index.php'
39     # See https://github.com/kaltura/server/blob/master/plugins/content/caption/base/lib/model/enums/CaptionType.php
40     _CAPTION_TYPES = {
41         1: 'srt',
42         2: 'ttml',
43         3: 'vtt',
44     }
45     _TESTS = [
46         {
47             'url': 'kaltura:269692:1_1jc2y3e4',
48             'md5': '3adcbdb3dcc02d647539e53f284ba171',
49             'info_dict': {
50                 'id': '1_1jc2y3e4',
51                 'ext': 'mp4',
52                 'title': 'Straight from the Heart',
53                 'upload_date': '20131219',
54                 'uploader_id': 'mlundberg@wolfgangsvault.com',
55                 'description': 'The Allman Brothers Band, 12/16/1981',
56                 'thumbnail': 're:^https?://.*/thumbnail/.*',
57                 'timestamp': int,
58             },
59         },
60         {
61             'url': 'http://www.kaltura.com/index.php/kwidget/cache_st/1300318621/wid/_269692/uiconf_id/3873291/entry_id/1_1jc2y3e4',
62             'only_matching': True,
63         },
64         {
65             'url': 'https://cdnapisec.kaltura.com/index.php/kwidget/wid/_557781/uiconf_id/22845202/entry_id/1_plr1syf3',
66             'only_matching': True,
67         },
68         {
69             'url': 'https://cdnapisec.kaltura.com/html5/html5lib/v2.30.2/mwEmbedFrame.php/p/1337/uiconf_id/20540612/entry_id/1_sf5ovm7u?wid=_243342',
70             'only_matching': True,
71         },
72         {
73             # video with subtitles
74             'url': 'kaltura:111032:1_cw786r8q',
75             'only_matching': True,
76         },
77         {
78             # video with ttml subtitles (no fileExt)
79             'url': 'kaltura:1926081:0_l5ye1133',
80             'info_dict': {
81                 'id': '0_l5ye1133',
82                 'ext': 'mp4',
83                 'title': 'What Can You Do With Python?',
84                 'upload_date': '20160221',
85                 'uploader_id': 'stork',
86                 'thumbnail': 're:^https?://.*/thumbnail/.*',
87                 'timestamp': int,
88                 'subtitles': {
89                     'en': [{
90                         'ext': 'ttml',
91                     }],
92                 },
93             },
94             'skip': 'Gone. Maybe https://www.safaribooksonline.com/library/tutorials/introduction-to-python-anon/3469/',
95             'params': {
96                 'skip_download': True,
97             },
98         },
99         {
100             'url': 'https://www.kaltura.com/index.php/extwidget/preview/partner_id/1770401/uiconf_id/37307382/entry_id/0_58u8kme7/embed/iframe?&flashvars[streamerType]=auto',
101             'only_matching': True,
102         },
103         {
104             'url': 'https://www.kaltura.com:443/index.php/extwidget/preview/partner_id/1770401/uiconf_id/37307382/entry_id/0_58u8kme7/embed/iframe?&flashvars[streamerType]=auto',
105             'only_matching': True,
106         }
107     ]
108
109     @staticmethod
110     def _extract_url(webpage):
111         # Embed codes: https://knowledge.kaltura.com/embedding-kaltura-media-players-your-site
112         mobj = (
113             re.search(
114                 r"""(?xs)
115                     kWidget\.(?:thumb)?[Ee]mbed\(
116                     \{.*?
117                         (?P<q1>['"])wid(?P=q1)\s*:\s*
118                         (?P<q2>['"])_?(?P<partner_id>(?:(?!(?P=q2)).)+)(?P=q2),.*?
119                         (?P<q3>['"])entry_?[Ii]d(?P=q3)\s*:\s*
120                         (?P<q4>['"])(?P<id>(?:(?!(?P=q4)).)+)(?P=q4)(?:,|\s*\})
121                 """, webpage)
122             or re.search(
123                 r'''(?xs)
124                     (?P<q1>["'])
125                         (?:https?:)?//cdnapi(?:sec)?\.kaltura\.com(?::\d+)?/(?:(?!(?P=q1)).)*\b(?:p|partner_id)/(?P<partner_id>\d+)(?:(?!(?P=q1)).)*
126                     (?P=q1).*?
127                     (?:
128                         (?:
129                             entry_?[Ii]d|
130                             (?P<q2>["'])entry_?[Ii]d(?P=q2)
131                         )\s*:\s*|
132                         \[\s*(?P<q2_1>["'])entry_?[Ii]d(?P=q2_1)\s*\]\s*=\s*
133                     )
134                     (?P<q3>["'])(?P<id>(?:(?!(?P=q3)).)+)(?P=q3)
135                 ''', webpage)
136             or re.search(
137                 r'''(?xs)
138                     <(?:iframe[^>]+src|meta[^>]+\bcontent)=(?P<q1>["'])
139                       (?:https?:)?//(?:(?:www|cdnapi(?:sec)?)\.)?kaltura\.com/(?:(?!(?P=q1)).)*\b(?:p|partner_id)/(?P<partner_id>\d+)
140                       (?:(?!(?P=q1)).)*
141                       [?&;]entry_id=(?P<id>(?:(?!(?P=q1))[^&])+)
142                       (?:(?!(?P=q1)).)*
143                     (?P=q1)
144                 ''', webpage)
145         )
146         if mobj:
147             embed_info = mobj.groupdict()
148             for k, v in embed_info.items():
149                 embed_info[k] = v.strip()
150             url = 'kaltura:%(partner_id)s:%(id)s' % embed_info
151             escaped_pid = re.escape(embed_info['partner_id'])
152             service_url = re.search(
153                 r'<script[^>]+src=["\']((?:https?:)?//.+?)/p/%s/sp/%s00/embedIframeJs' % (escaped_pid, escaped_pid),
154                 webpage)
155             if service_url:
156                 url = smuggle_url(url, {'service_url': service_url.group(1)})
157             return url
158
159     def _kaltura_api_call(self, video_id, actions, service_url=None, *args, **kwargs):
160         params = actions[0]
161         if len(actions) > 1:
162             for i, a in enumerate(actions[1:], start=1):
163                 for k, v in a.items():
164                     params['%d:%s' % (i, k)] = v
165
166         data = self._download_json(
167             (service_url or self._SERVICE_URL) + self._SERVICE_BASE,
168             video_id, query=params, *args, **kwargs)
169
170         status = data if len(actions) == 1 else data[0]
171         if status.get('objectType') == 'KalturaAPIException':
172             raise ExtractorError(
173                 '%s said: %s' % (self.IE_NAME, status['message']))
174
175         return data
176
177     def _get_video_info(self, video_id, partner_id, service_url=None):
178         actions = [
179             {
180                 'action': 'null',
181                 'apiVersion': '3.1.5',
182                 'clientTag': 'kdp:v3.8.5',
183                 'format': 1,  # JSON, 2 = XML, 3 = PHP
184                 'service': 'multirequest',
185             },
186             {
187                 'expiry': 86400,
188                 'service': 'session',
189                 'action': 'startWidgetSession',
190                 'widgetId': '_%s' % partner_id,
191             },
192             {
193                 'action': 'get',
194                 'entryId': video_id,
195                 'service': 'baseentry',
196                 'ks': '{1:result:ks}',
197                 'responseProfile:fields': 'createdAt,dataUrl,duration,name,plays,thumbnailUrl,userId',
198                 'responseProfile:type': 1,
199             },
200             {
201                 'action': 'getbyentryid',
202                 'entryId': video_id,
203                 'service': 'flavorAsset',
204                 'ks': '{1:result:ks}',
205             },
206             {
207                 'action': 'list',
208                 'filter:entryIdEqual': video_id,
209                 'service': 'caption_captionasset',
210                 'ks': '{1:result:ks}',
211             },
212         ]
213         return self._kaltura_api_call(
214             video_id, actions, service_url, note='Downloading video info JSON')
215
216     def _real_extract(self, url):
217         url, smuggled_data = unsmuggle_url(url, {})
218
219         mobj = re.match(self._VALID_URL, url)
220         partner_id, entry_id = mobj.group('partner_id', 'id')
221         ks = None
222         captions = None
223         if partner_id and entry_id:
224             _, info, flavor_assets, captions = self._get_video_info(entry_id, partner_id, smuggled_data.get('service_url'))
225         else:
226             path, query = mobj.group('path', 'query')
227             if not path and not query:
228                 raise ExtractorError('Invalid URL', expected=True)
229             params = {}
230             if query:
231                 params = compat_parse_qs(query)
232             if path:
233                 splitted_path = path.split('/')
234                 params.update(dict((zip(splitted_path[::2], [[v] for v in splitted_path[1::2]]))))
235             if 'wid' in params:
236                 partner_id = params['wid'][0][1:]
237             elif 'p' in params:
238                 partner_id = params['p'][0]
239             elif 'partner_id' in params:
240                 partner_id = params['partner_id'][0]
241             else:
242                 raise ExtractorError('Invalid URL', expected=True)
243             if 'entry_id' in params:
244                 entry_id = params['entry_id'][0]
245                 _, info, flavor_assets, captions = self._get_video_info(entry_id, partner_id)
246             elif 'uiconf_id' in params and 'flashvars[referenceId]' in params:
247                 reference_id = params['flashvars[referenceId]'][0]
248                 webpage = self._download_webpage(url, reference_id)
249                 entry_data = self._parse_json(self._search_regex(
250                     r'window\.kalturaIframePackageData\s*=\s*({.*});',
251                     webpage, 'kalturaIframePackageData'),
252                     reference_id)['entryResult']
253                 info, flavor_assets = entry_data['meta'], entry_data['contextData']['flavorAssets']
254                 entry_id = info['id']
255                 # Unfortunately, data returned in kalturaIframePackageData lacks
256                 # captions so we will try requesting the complete data using
257                 # regular approach since we now know the entry_id
258                 try:
259                     _, info, flavor_assets, captions = self._get_video_info(
260                         entry_id, partner_id)
261                 except ExtractorError:
262                     # Regular scenario failed but we already have everything
263                     # extracted apart from captions and can process at least
264                     # with this
265                     pass
266             else:
267                 raise ExtractorError('Invalid URL', expected=True)
268             ks = params.get('flashvars[ks]', [None])[0]
269
270         source_url = smuggled_data.get('source_url')
271         if source_url:
272             referrer = base64.b64encode(
273                 '://'.join(compat_urlparse.urlparse(source_url)[:2])
274                 .encode('utf-8')).decode('utf-8')
275         else:
276             referrer = None
277
278         def sign_url(unsigned_url):
279             if ks:
280                 unsigned_url += '/ks/%s' % ks
281             if referrer:
282                 unsigned_url += '?referrer=%s' % referrer
283             return unsigned_url
284
285         data_url = info['dataUrl']
286         if '/flvclipper/' in data_url:
287             data_url = re.sub(r'/flvclipper/.*', '/serveFlavor', data_url)
288
289         formats = []
290         for f in flavor_assets:
291             # Continue if asset is not ready
292             if f.get('status') != 2:
293                 continue
294             # Original format that's not available (e.g. kaltura:1926081:0_c03e1b5g)
295             # skip for now.
296             if f.get('fileExt') == 'chun':
297                 continue
298             # DRM-protected video, cannot be decrypted
299             if f.get('fileExt') == 'wvm':
300                 continue
301             if not f.get('fileExt'):
302                 # QT indicates QuickTime; some videos have broken fileExt
303                 if f.get('containerFormat') == 'qt':
304                     f['fileExt'] = 'mov'
305                 else:
306                     f['fileExt'] = 'mp4'
307             video_url = sign_url(
308                 '%s/flavorId/%s' % (data_url, f['id']))
309             # audio-only has no videoCodecId (e.g. kaltura:1926081:0_c03e1b5g
310             # -f mp4-56)
311             vcodec = 'none' if 'videoCodecId' not in f and f.get(
312                 'frameRate') == 0 else f.get('videoCodecId')
313             formats.append({
314                 'format_id': '%(fileExt)s-%(bitrate)s' % f,
315                 'ext': f.get('fileExt'),
316                 'tbr': int_or_none(f['bitrate']),
317                 'fps': int_or_none(f.get('frameRate')),
318                 'filesize_approx': int_or_none(f.get('size'), invscale=1024),
319                 'container': f.get('containerFormat'),
320                 'vcodec': vcodec,
321                 'height': int_or_none(f.get('height')),
322                 'width': int_or_none(f.get('width')),
323                 'url': video_url,
324             })
325         if '/playManifest/' in data_url:
326             m3u8_url = sign_url(data_url.replace(
327                 'format/url', 'format/applehttp'))
328             formats.extend(self._extract_m3u8_formats(
329                 m3u8_url, entry_id, 'mp4', 'm3u8_native',
330                 m3u8_id='hls', fatal=False))
331
332         self._sort_formats(formats)
333
334         subtitles = {}
335         if captions:
336             for caption in captions.get('objects', []):
337                 # Continue if caption is not ready
338                 if caption.get('status') != 2:
339                     continue
340                 if not caption.get('id'):
341                     continue
342                 caption_format = int_or_none(caption.get('format'))
343                 subtitles.setdefault(caption.get('languageCode') or caption.get('language'), []).append({
344                     'url': '%s/api_v3/service/caption_captionasset/action/serve/captionAssetId/%s' % (self._SERVICE_URL, caption['id']),
345                     'ext': caption.get('fileExt') or self._CAPTION_TYPES.get(caption_format) or 'ttml',
346                 })
347
348         return {
349             'id': entry_id,
350             'title': info['name'],
351             'formats': formats,
352             'subtitles': subtitles,
353             'description': clean_html(info.get('description')),
354             'thumbnail': info.get('thumbnailUrl'),
355             'duration': info.get('duration'),
356             'timestamp': info.get('createdAt'),
357             'uploader_id': info.get('userId') if info.get('userId') != 'None' else None,
358             'view_count': info.get('plays'),
359         }