[odnoklassniki] extract m3u8 formats
[youtube-dl] / youtube_dl / extractor / odnoklassniki.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 from .common import InfoExtractor
5 from ..compat import (
6     compat_etree_fromstring,
7     compat_parse_qs,
8     compat_urllib_parse_unquote,
9     compat_urllib_parse_urlparse,
10 )
11 from ..utils import (
12     ExtractorError,
13     unified_strdate,
14     int_or_none,
15     qualities,
16     unescapeHTML,
17 )
18
19
20 class OdnoklassnikiIE(InfoExtractor):
21     _VALID_URL = r'https?://(?:(?:www|m|mobile)\.)?(?:odnoklassniki|ok)\.ru/(?:video(?:embed)?|web-api/video/moviePlayer)/(?P<id>[\d-]+)'
22     _TESTS = [{
23         # metadata in JSON
24         'url': 'http://ok.ru/video/20079905452',
25         'md5': '6ba728d85d60aa2e6dd37c9e70fdc6bc',
26         'info_dict': {
27             'id': '20079905452',
28             'ext': 'mp4',
29             'title': 'Культура меняет нас (прекрасный ролик!))',
30             'duration': 100,
31             'upload_date': '20141207',
32             'uploader_id': '330537914540',
33             'uploader': 'Виталий Добровольский',
34             'like_count': int,
35             'age_limit': 0,
36         },
37         'skip': 'Video has been blocked',
38     }, {
39         # metadataUrl
40         'url': 'http://ok.ru/video/63567059965189-0?fromTime=5',
41         'md5': '9676cf86eff5391d35dea675d224e131',
42         'info_dict': {
43             'id': '63567059965189-0',
44             'ext': 'mp4',
45             'title': 'Девушка без комплексов ...',
46             'duration': 191,
47             'upload_date': '20150518',
48             'uploader_id': '534380003155',
49             'uploader': '☭ Андрей Мещанинов ☭',
50             'like_count': int,
51             'age_limit': 0,
52             'start_time': 5,
53         },
54     }, {
55         # YouTube embed (metadataUrl, provider == USER_YOUTUBE)
56         'url': 'http://ok.ru/video/64211978996595-1',
57         'md5': '5d7475d428845cd2e13bae6f1a992278',
58         'info_dict': {
59             'id': '64211978996595-1',
60             'ext': 'mp4',
61             'title': 'Космическая среда от 26 августа 2015',
62             'description': 'md5:848eb8b85e5e3471a3a803dae1343ed0',
63             'duration': 440,
64             'upload_date': '20150826',
65             'uploader_id': '750099571',
66             'uploader': 'Алина П',
67             'age_limit': 0,
68         },
69     }, {
70         # YouTube embed (metadata, provider == USER_YOUTUBE, no metadata.movie.title field)
71         'url': 'http://ok.ru/video/62036049272859-0',
72         'info_dict': {
73             'id': '62036049272859-0',
74             'ext': 'mp4',
75             'title': 'МУЗЫКА     ДОЖДЯ .',
76             'description': 'md5:6f1867132bd96e33bf53eda1091e8ed0',
77             'upload_date': '20120106',
78             'uploader_id': '473534735899',
79             'uploader': 'МARINA D',
80             'age_limit': 0,
81         },
82         'params': {
83             'skip_download': True,
84         },
85     }, {
86         'url': 'http://ok.ru/web-api/video/moviePlayer/20079905452',
87         'only_matching': True,
88     }, {
89         'url': 'http://www.ok.ru/video/20648036891',
90         'only_matching': True,
91     }, {
92         'url': 'http://www.ok.ru/videoembed/20648036891',
93         'only_matching': True,
94     }, {
95         'url': 'http://m.ok.ru/video/20079905452',
96         'only_matching': True,
97     }, {
98         'url': 'http://mobile.ok.ru/video/20079905452',
99         'only_matching': True,
100     }]
101
102     def _real_extract(self, url):
103         start_time = int_or_none(compat_parse_qs(
104             compat_urllib_parse_urlparse(url).query).get('fromTime', [None])[0])
105
106         video_id = self._match_id(url)
107
108         webpage = self._download_webpage(
109             'http://ok.ru/video/%s' % video_id, video_id)
110
111         error = self._search_regex(
112             r'[^>]+class="vp_video_stub_txt"[^>]*>([^<]+)<',
113             webpage, 'error', default=None)
114         if error:
115             raise ExtractorError(error, expected=True)
116
117         player = self._parse_json(
118             unescapeHTML(self._search_regex(
119                 r'data-options=(?P<quote>["\'])(?P<player>{.+?%s.+?})(?P=quote)' % video_id,
120                 webpage, 'player', group='player')),
121             video_id)
122
123         flashvars = player['flashvars']
124
125         metadata = flashvars.get('metadata')
126         if metadata:
127             metadata = self._parse_json(metadata, video_id)
128         else:
129             metadata = self._download_json(
130                 compat_urllib_parse_unquote(flashvars['metadataUrl']),
131                 video_id, 'Downloading metadata JSON')
132
133         movie = metadata['movie']
134
135         # Some embedded videos may not contain title in movie dict (e.g.
136         # http://ok.ru/video/62036049272859-0) thus we allow missing title
137         # here and it's going to be extracted later by an extractor that
138         # will process the actual embed.
139         provider = metadata.get('provider')
140         title = movie['title'] if provider == 'UPLOADED_ODKL' else movie.get('title')
141
142         thumbnail = movie.get('poster')
143         duration = int_or_none(movie.get('duration'))
144
145         author = metadata.get('author', {})
146         uploader_id = author.get('id')
147         uploader = author.get('name')
148
149         upload_date = unified_strdate(self._html_search_meta(
150             'ya:ovs:upload_date', webpage, 'upload date', default=None))
151
152         age_limit = None
153         adult = self._html_search_meta(
154             'ya:ovs:adult', webpage, 'age limit', default=None)
155         if adult:
156             age_limit = 18 if adult == 'true' else 0
157
158         like_count = int_or_none(metadata.get('likeCount'))
159
160         info = {
161             'id': video_id,
162             'title': title,
163             'thumbnail': thumbnail,
164             'duration': duration,
165             'upload_date': upload_date,
166             'uploader': uploader,
167             'uploader_id': uploader_id,
168             'like_count': like_count,
169             'age_limit': age_limit,
170             'start_time': start_time,
171         }
172
173         if provider == 'USER_YOUTUBE':
174             info.update({
175                 '_type': 'url_transparent',
176                 'url': movie['contentId'],
177             })
178             return info
179
180         quality = qualities(('4', '0', '1', '2', '3', '5'))
181
182         formats = [{
183             'url': f['url'],
184             'ext': 'mp4',
185             'format_id': f['name'],
186         } for f in metadata['videos']]
187
188         m3u8_url = metadata.get('hlsManifestUrl')
189         if m3u8_url:
190             formats.extend(self._extract_m3u8_formats(
191                 m3u8_url, video_id, 'mp4', 'm3u8_native',
192                 m3u8_id='hls', fatal=False))
193
194         dash_manifest = metadata.get('metadataEmbedded')
195         if dash_manifest:
196             formats.extend(self._parse_mpd_formats(
197                 compat_etree_fromstring(dash_manifest), 'mpd'))
198
199         for fmt in formats:
200             fmt_type = self._search_regex(
201                 r'\btype[/=](\d)', fmt['url'],
202                 'format type', default=None)
203             if fmt_type:
204                 fmt['quality'] = quality(fmt_type)
205
206         self._sort_formats(formats)
207
208         info['formats'] = formats
209         return info