[ooyala] extract more formats and metadata
[youtube-dl] / youtube_dl / extractor / ooyala.py
1 from __future__ import unicode_literals
2 import re
3 import base64
4
5 from .common import InfoExtractor
6 from ..utils import (
7     int_or_none,
8     float_or_none,
9 )
10
11
12 class OoyalaBaseIE(InfoExtractor):
13
14     def _extract(self, player_url, video_id):
15         print(player_url)
16         content_tree = self._download_json(player_url, video_id)['content_tree']
17         metadata = content_tree[list(content_tree)[0]]
18         embed_code = metadata['embed_code']
19         pcode = metadata.get('asset_pcode') or embed_code
20         video_info = {
21             'id': embed_code,
22             'title': metadata['title'],
23             'description': metadata.get('description'),
24             'thumbnail': metadata.get('thumbnail_image') or metadata.get('promo_image'),
25             'duration': int_or_none(metadata.get('duration')),
26         }
27
28         formats = []
29         for supported_format in ('mp4', 'm3u8', 'hds', 'rtmp'):
30             auth_data = self._download_json(
31                 'http://player.ooyala.com/sas/player_api/v1/authorization/embed_code/%s/%s?domain=www.example.org&supportedFormats=%s' % (pcode, embed_code, supported_format),
32                 video_id, 'Downloading %s JSON' % supported_format)
33
34             cur_auth_data = auth_data['authorization_data'][embed_code]
35
36             for stream in cur_auth_data['streams']:
37                 url = base64.b64decode(stream['url']['data'].encode('ascii')).decode('utf-8')
38                 delivery_type = stream['delivery_type']
39                 if delivery_type == 'remote_asset':
40                     video_info['url'] = url
41                     return video_info
42                 if delivery_type == 'hls':
43                     formats.extend(self._extract_m3u8_formats(url, embed_code, 'mp4', 'm3u8_native', 0, m3u8_id='hls', fatal=False))
44                 elif delivery_type == 'hds':
45                     formats.extend(self._extract_f4m_formats(url, embed_code, f4m_id='hds', fatal=False))
46                 else:
47                     formats.append({
48                         'url': url,
49                         'ext': stream.get('delivery_type'),
50                         'vcodec': stream.get('video_codec'),
51                         'format_id': stream.get('profile'),
52                         'width': int_or_none(stream.get('width')),
53                         'height': int_or_none(stream.get('height')),
54                         'abr': int_or_none(stream.get('audio_bitrate')),
55                         'vbr': int_or_none(stream.get('video_bitrate')),
56                         'fps': float_or_none(stream.get('framerate')),
57                     })
58         self._sort_formats(formats)
59
60         video_info['formats'] = formats
61         return video_info
62
63
64 class OoyalaIE(OoyalaBaseIE):
65     _VALID_URL = r'(?:ooyala:|https?://.+?\.ooyala\.com/.*?(?:embedCode|ec)=)(?P<id>.+?)(&|$)'
66
67     _TESTS = [
68         {
69             # From http://it.slashdot.org/story/13/04/25/178216/recovering-data-from-broken-hard-drives-and-ssds-video
70             'url': 'http://player.ooyala.com/player.js?embedCode=pxczE2YjpfHfn1f3M-ykG_AmJRRn0PD8',
71             'info_dict': {
72                 'id': 'pxczE2YjpfHfn1f3M-ykG_AmJRRn0PD8',
73                 'ext': 'mp4',
74                 'title': 'Explaining Data Recovery from Hard Drives and SSDs',
75                 'description': 'How badly damaged does a drive have to be to defeat Russell and his crew? Apparently, smashed to bits.',
76                 'duration': 853386,
77             },
78         }, {
79             # Only available for ipad
80             'url': 'http://player.ooyala.com/player.js?embedCode=x1b3lqZDq9y_7kMyC2Op5qo-p077tXD0',
81             'info_dict': {
82                 'id': 'x1b3lqZDq9y_7kMyC2Op5qo-p077tXD0',
83                 'ext': 'mp4',
84                 'title': 'Simulation Overview - Levels of Simulation',
85                 'duration': 194948,
86             },
87         },
88         {
89             # Information available only through SAS api
90             # From http://community.plm.automation.siemens.com/t5/News-NX-Manufacturing/Tool-Path-Divide/ba-p/4187
91             'url': 'http://player.ooyala.com/player.js?embedCode=FiOG81ZTrvckcchQxmalf4aQj590qTEx',
92             'md5': 'a84001441b35ea492bc03736e59e7935',
93             'info_dict': {
94                 'id': 'FiOG81ZTrvckcchQxmalf4aQj590qTEx',
95                 'ext': 'mp4',
96                 'title': 'Divide Tool Path.mp4',
97                 'duration': 204405,
98             }
99         }
100     ]
101
102     @staticmethod
103     def _url_for_embed_code(embed_code):
104         return 'http://player.ooyala.com/player.js?embedCode=%s' % embed_code
105
106     @classmethod
107     def _build_url_result(cls, embed_code):
108         return cls.url_result(cls._url_for_embed_code(embed_code),
109                               ie=cls.ie_key())
110
111     def _real_extract(self, url):
112         embed_code = self._match_id(url)
113         content_tree_url = 'http://player.ooyala.com/player_api/v1/content_tree/embed_code/%s/%s' % (embed_code, embed_code)
114         return self._extract(content_tree_url, embed_code)
115
116
117 class OoyalaExternalIE(OoyalaBaseIE):
118     _VALID_URL = r'''(?x)
119                     (?:
120                         ooyalaexternal:|
121                         https?://.+?\.ooyala\.com/.*?\bexternalId=
122                     )
123                     (?P<partner_id>[^:]+)
124                     :
125                     (?P<id>.+)
126                     (?:
127                         :|
128                         .*?&pcode=
129                     )
130                     (?P<pcode>.+?)
131                     (?:&|$)
132                     '''
133
134     _TEST = {
135         'url': 'https://player.ooyala.com/player.js?externalId=espn:10365079&pcode=1kNG061cgaoolOncv54OAO1ceO-I&adSetCode=91cDU6NuXTGKz3OdjOxFdAgJVtQcKJnI&callback=handleEvents&hasModuleParams=1&height=968&playerBrandingId=7af3bd04449c444c964f347f11873075&targetReplaceId=videoPlayer&width=1656&wmode=opaque&allowScriptAccess=always',
136         'info_dict': {
137             'id': 'FkYWtmazr6Ed8xmvILvKLWjd4QvYZpzG',
138             'ext': 'mp4',
139             'title': 'dm_140128_30for30Shorts___JudgingJewellv2',
140             'duration': 1302000,
141         },
142         'params': {
143             # m3u8 download
144             'skip_download': True,
145         },
146     }
147
148     def _real_extract(self, url):
149         partner_id, video_id, pcode = re.match(self._VALID_URL, url).groups()
150         content_tree_url = 'http://player.ooyala.com/player_api/v1/content_tree/external_id/%s/%s:%s' % (pcode, partner_id, video_id)
151         return self._extract(content_tree_url, video_id)