[ok] Capture error message
[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 compat_urllib_parse_unquote
6 from ..utils import (
7     ExtractorError,
8     unified_strdate,
9     int_or_none,
10     qualities,
11     unescapeHTML,
12 )
13
14
15 class OdnoklassnikiIE(InfoExtractor):
16     _VALID_URL = r'https?://(?:www\.)?(?:odnoklassniki|ok)\.ru/(?:video|web-api/video/moviePlayer)/(?P<id>[\d-]+)'
17     _TESTS = [{
18         # metadata in JSON
19         'url': 'http://ok.ru/video/20079905452',
20         'md5': '6ba728d85d60aa2e6dd37c9e70fdc6bc',
21         'info_dict': {
22             'id': '20079905452',
23             'ext': 'mp4',
24             'title': 'Культура меняет нас (прекрасный ролик!))',
25             'duration': 100,
26             'upload_date': '20141207',
27             'uploader_id': '330537914540',
28             'uploader': 'Виталий Добровольский',
29             'like_count': int,
30             'age_limit': 0,
31         },
32     }, {
33         # metadataUrl
34         'url': 'http://ok.ru/video/63567059965189-0',
35         'md5': '9676cf86eff5391d35dea675d224e131',
36         'info_dict': {
37             'id': '63567059965189-0',
38             'ext': 'mp4',
39             'title': 'Девушка без комплексов ...',
40             'duration': 191,
41             'upload_date': '20150518',
42             'uploader_id': '534380003155',
43             'uploader': '☭ Андрей Мещанинов ☭',
44             'like_count': int,
45             'age_limit': 0,
46         },
47     }, {
48         # YouTube embed (metadataUrl, provider == USER_YOUTUBE)
49         'url': 'http://ok.ru/video/64211978996595-1',
50         'md5': '5d7475d428845cd2e13bae6f1a992278',
51         'info_dict': {
52             'id': '64211978996595-1',
53             'ext': 'mp4',
54             'title': 'Космическая среда от 26 августа 2015',
55             'description': 'md5:848eb8b85e5e3471a3a803dae1343ed0',
56             'duration': 440,
57             'upload_date': '20150826',
58             'uploader_id': '750099571',
59             'uploader': 'Алина П',
60             'age_limit': 0,
61         },
62     }, {
63         'url': 'http://ok.ru/web-api/video/moviePlayer/20079905452',
64         'only_matching': True,
65     }, {
66         'url': 'http://www.ok.ru/video/20648036891',
67         'only_matching': True,
68     }]
69
70     def _real_extract(self, url):
71         video_id = self._match_id(url)
72
73         webpage = self._download_webpage(
74             'http://ok.ru/video/%s' % video_id, video_id)
75
76         error = self._search_regex(
77             r'[^>]+class="vp_video_stub_txt"[^>]*>([^<]+)<',
78             webpage, 'error', default=None)
79         if error:
80             raise ExtractorError(error, expected=True)
81
82         player = self._parse_json(
83             unescapeHTML(self._search_regex(
84                 r'data-options=(?P<quote>["\'])(?P<player>{.+?%s.+?})(?P=quote)' % video_id,
85                 webpage, 'player', group='player')),
86             video_id)
87
88         flashvars = player['flashvars']
89
90         metadata = flashvars.get('metadata')
91         if metadata:
92             metadata = self._parse_json(metadata, video_id)
93         else:
94             metadata = self._download_json(
95                 compat_urllib_parse_unquote(flashvars['metadataUrl']),
96                 video_id, 'Downloading metadata JSON')
97
98         movie = metadata['movie']
99         title = movie['title']
100         thumbnail = movie.get('poster')
101         duration = int_or_none(movie.get('duration'))
102
103         author = metadata.get('author', {})
104         uploader_id = author.get('id')
105         uploader = author.get('name')
106
107         upload_date = unified_strdate(self._html_search_meta(
108             'ya:ovs:upload_date', webpage, 'upload date', default=None))
109
110         age_limit = None
111         adult = self._html_search_meta(
112             'ya:ovs:adult', webpage, 'age limit', default=None)
113         if adult:
114             age_limit = 18 if adult == 'true' else 0
115
116         like_count = int_or_none(metadata.get('likeCount'))
117
118         info = {
119             'id': video_id,
120             'title': title,
121             'thumbnail': thumbnail,
122             'duration': duration,
123             'upload_date': upload_date,
124             'uploader': uploader,
125             'uploader_id': uploader_id,
126             'like_count': like_count,
127             'age_limit': age_limit,
128         }
129
130         if metadata.get('provider') == 'USER_YOUTUBE':
131             info.update({
132                 '_type': 'url_transparent',
133                 'url': movie['contentId'],
134             })
135             return info
136
137         quality = qualities(('mobile', 'lowest', 'low', 'sd', 'hd'))
138
139         formats = [{
140             'url': f['url'],
141             'ext': 'mp4',
142             'format_id': f['name'],
143             'quality': quality(f['name']),
144         } for f in metadata['videos']]
145         self._sort_formats(formats)
146
147         info['formats'] = formats
148         return info