[ok] Improve player regex
[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     unified_strdate,
8     int_or_none,
9     qualities,
10     unescapeHTML,
11 )
12
13
14 class OdnoklassnikiIE(InfoExtractor):
15     _VALID_URL = r'https?://(?:www\.)?(?:odnoklassniki|ok)\.ru/(?:video|web-api/video/moviePlayer)/(?P<id>[\d-]+)'
16     _TESTS = [{
17         # metadata in JSON
18         'url': 'http://ok.ru/video/20079905452',
19         'md5': '6ba728d85d60aa2e6dd37c9e70fdc6bc',
20         'info_dict': {
21             'id': '20079905452',
22             'ext': 'mp4',
23             'title': 'Культура меняет нас (прекрасный ролик!))',
24             'duration': 100,
25             'upload_date': '20141207',
26             'uploader_id': '330537914540',
27             'uploader': 'Виталий Добровольский',
28             'like_count': int,
29             'age_limit': 0,
30         },
31     }, {
32         # metadataUrl
33         'url': 'http://ok.ru/video/63567059965189-0',
34         'md5': '9676cf86eff5391d35dea675d224e131',
35         'info_dict': {
36             'id': '63567059965189-0',
37             'ext': 'mp4',
38             'title': 'Девушка без комплексов ...',
39             'duration': 191,
40             'upload_date': '20150518',
41             'uploader_id': '534380003155',
42             'uploader': '☭ Андрей Мещанинов ☭',
43             'like_count': int,
44             'age_limit': 0,
45         },
46     }, {
47         'url': 'http://ok.ru/web-api/video/moviePlayer/20079905452',
48         'only_matching': True,
49     }, {
50         'url': 'http://www.ok.ru/video/20648036891',
51         'only_matching': True,
52     }]
53
54     def _real_extract(self, url):
55         video_id = self._match_id(url)
56
57         webpage = self._download_webpage(
58             'http://ok.ru/video/%s' % video_id, video_id)
59
60         player = self._parse_json(
61             unescapeHTML(self._search_regex(
62                 r'data-options=(?P<quote>["\'])(?P<player>{.+?%s.+?})(?P=quote)' % video_id,
63                 webpage, 'player', group='player')),
64             video_id)
65
66         flashvars = player['flashvars']
67
68         metadata = flashvars.get('metadata')
69         if metadata:
70             metadata = self._parse_json(metadata, video_id)
71         else:
72             metadata = self._download_json(
73                 compat_urllib_parse_unquote(flashvars['metadataUrl']),
74                 video_id, 'Downloading metadata JSON')
75
76         movie = metadata['movie']
77         title = movie['title']
78         thumbnail = movie.get('poster')
79         duration = int_or_none(movie.get('duration'))
80
81         author = metadata.get('author', {})
82         uploader_id = author.get('id')
83         uploader = author.get('name')
84
85         upload_date = unified_strdate(self._html_search_meta(
86             'ya:ovs:upload_date', webpage, 'upload date', default=None))
87
88         age_limit = None
89         adult = self._html_search_meta(
90             'ya:ovs:adult', webpage, 'age limit', default=None)
91         if adult:
92             age_limit = 18 if adult == 'true' else 0
93
94         like_count = int_or_none(metadata.get('likeCount'))
95
96         quality = qualities(('mobile', 'lowest', 'low', 'sd', 'hd'))
97
98         formats = [{
99             'url': f['url'],
100             'ext': 'mp4',
101             'format_id': f['name'],
102             'quality': quality(f['name']),
103         } for f in metadata['videos']]
104
105         return {
106             'id': video_id,
107             'title': title,
108             'thumbnail': thumbnail,
109             'duration': duration,
110             'upload_date': upload_date,
111             'uploader': uploader,
112             'uploader_id': uploader_id,
113             'like_count': like_count,
114             'age_limit': age_limit,
115             'formats': formats,
116         }