50234798bbe5176d881b5a42357b92ef36c16fe3
[youtube-dl] / youtube_dl / extractor / mailru.py
1 # coding: utf-8
2 from __future__ import unicode_literals
3
4 import itertools
5 import json
6 import re
7
8 from .common import InfoExtractor
9 from ..compat import compat_urllib_parse_unquote
10 from ..utils import (
11     int_or_none,
12     parse_duration,
13     remove_end,
14     try_get,
15 )
16
17
18 class MailRuIE(InfoExtractor):
19     IE_NAME = 'mailru'
20     IE_DESC = 'Видео@Mail.Ru'
21     _VALID_URL = r'''(?x)
22                     https?://
23                         (?:(?:www|m)\.)?my\.mail\.ru/+
24                         (?:
25                             video/.*\#video=/?(?P<idv1>(?:[^/]+/){3}\d+)|
26                             (?:(?P<idv2prefix>(?:[^/]+/+){2})video/(?P<idv2suffix>[^/]+/\d+))\.html|
27                             (?:video/embed|\+/video/meta)/(?P<metaid>\d+)
28                         )
29                     '''
30     _TESTS = [
31         {
32             'url': 'http://my.mail.ru/video/top#video=/mail/sonypicturesrus/75/76',
33             'md5': 'dea205f03120046894db4ebb6159879a',
34             'info_dict': {
35                 'id': '46301138_76',
36                 'ext': 'mp4',
37                 'title': 'Новый Человек-Паук. Высокое напряжение. Восстание Электро',
38                 'timestamp': 1393235077,
39                 'upload_date': '20140224',
40                 'uploader': 'sonypicturesrus',
41                 'uploader_id': 'sonypicturesrus@mail.ru',
42                 'duration': 184,
43             },
44             'skip': 'Not accessible from Travis CI server',
45         },
46         {
47             'url': 'http://my.mail.ru/corp/hitech/video/news_hi-tech_mail_ru/1263.html',
48             'md5': '00a91a58c3402204dcced523777b475f',
49             'info_dict': {
50                 'id': '46843144_1263',
51                 'ext': 'mp4',
52                 'title': 'Samsung Galaxy S5 Hammer Smash Fail Battery Explosion',
53                 'timestamp': 1397039888,
54                 'upload_date': '20140409',
55                 'uploader': 'hitech',
56                 'uploader_id': 'hitech@corp.mail.ru',
57                 'duration': 245,
58             },
59             'skip': 'Not accessible from Travis CI server',
60         },
61         {
62             # only available via metaUrl API
63             'url': 'http://my.mail.ru/mail/720pizle/video/_myvideo/502.html',
64             'md5': '3b26d2491c6949d031a32b96bd97c096',
65             'info_dict': {
66                 'id': '56664382_502',
67                 'ext': 'mp4',
68                 'title': ':8336',
69                 'timestamp': 1449094163,
70                 'upload_date': '20151202',
71                 'uploader': '720pizle@mail.ru',
72                 'uploader_id': '720pizle@mail.ru',
73                 'duration': 6001,
74             },
75             'skip': 'Not accessible from Travis CI server',
76         },
77         {
78             'url': 'http://m.my.mail.ru/mail/3sktvtr/video/_myvideo/138.html',
79             'only_matching': True,
80         },
81         {
82             'url': 'https://my.mail.ru/video/embed/7949340477499637815',
83             'only_matching': True,
84         },
85         {
86             'url': 'http://my.mail.ru/+/video/meta/7949340477499637815',
87             'only_matching': True,
88         },
89         {
90             'url': 'https://my.mail.ru//list/sinyutin10/video/_myvideo/4.html',
91             'only_matching': True,
92         },
93         {
94             'url': 'https://my.mail.ru//list//sinyutin10/video/_myvideo/4.html',
95             'only_matching': True,
96         }
97     ]
98
99     def _real_extract(self, url):
100         mobj = re.match(self._VALID_URL, url)
101         meta_id = mobj.group('metaid')
102
103         video_id = None
104         if meta_id:
105             meta_url = 'https://my.mail.ru/+/video/meta/%s' % meta_id
106         else:
107             video_id = mobj.group('idv1')
108             if not video_id:
109                 video_id = mobj.group('idv2prefix') + mobj.group('idv2suffix')
110             webpage = self._download_webpage(url, video_id)
111             page_config = self._parse_json(self._search_regex(
112                 r'(?s)<script[^>]+class="sp-video__page-config"[^>]*>(.+?)</script>',
113                 webpage, 'page config', default='{}'), video_id, fatal=False)
114             if page_config:
115                 meta_url = page_config.get('metaUrl') or page_config.get('video', {}).get('metaUrl')
116             else:
117                 meta_url = None
118
119         video_data = None
120         if meta_url:
121             video_data = self._download_json(
122                 meta_url, video_id or meta_id, 'Downloading video meta JSON',
123                 fatal=not video_id)
124
125         # Fallback old approach
126         if not video_data:
127             video_data = self._download_json(
128                 'http://api.video.mail.ru/videos/%s.json?new=1' % video_id,
129                 video_id, 'Downloading video JSON')
130
131         formats = []
132         for f in video_data['videos']:
133             video_url = f.get('url')
134             if not video_url:
135                 continue
136             format_id = f.get('key')
137             height = int_or_none(self._search_regex(
138                 r'^(\d+)[pP]$', format_id, 'height', default=None)) if format_id else None
139             formats.append({
140                 'url': video_url,
141                 'format_id': format_id,
142                 'height': height,
143             })
144         self._sort_formats(formats)
145
146         meta_data = video_data['meta']
147         title = remove_end(meta_data['title'], '.mp4')
148
149         author = video_data.get('author')
150         uploader = author.get('name')
151         uploader_id = author.get('id') or author.get('email')
152         view_count = int_or_none(video_data.get('viewsCount') or video_data.get('views_count'))
153
154         acc_id = meta_data.get('accId')
155         item_id = meta_data.get('itemId')
156         content_id = '%s_%s' % (acc_id, item_id) if acc_id and item_id else video_id
157
158         thumbnail = meta_data.get('poster')
159         duration = int_or_none(meta_data.get('duration'))
160         timestamp = int_or_none(meta_data.get('timestamp'))
161
162         return {
163             'id': content_id,
164             'title': title,
165             'thumbnail': thumbnail,
166             'timestamp': timestamp,
167             'uploader': uploader,
168             'uploader_id': uploader_id,
169             'duration': duration,
170             'view_count': view_count,
171             'formats': formats,
172         }
173
174
175 class MailRuMusicSearchBaseIE(InfoExtractor):
176     def _search(self, query, url, audio_id, limit=100, offset=0):
177         search = self._download_json(
178             'https://my.mail.ru/cgi-bin/my/ajax', audio_id,
179             'Downloading songs JSON page %d' % (offset // limit + 1),
180             headers={
181                 'Referer': url,
182                 'X-Requested-With': 'XMLHttpRequest',
183             }, query={
184                 'xemail': '',
185                 'ajax_call': '1',
186                 'func_name': 'music.search',
187                 'mna': '',
188                 'mnb': '',
189                 'arg_query': query,
190                 'arg_extended': '1',
191                 'arg_search_params': json.dumps({
192                     'music': {
193                         'limit': limit,
194                         'offset': offset,
195                     },
196                 }),
197                 'arg_limit': limit,
198                 'arg_offset': offset,
199             })
200         return next(e for e in search if isinstance(e, dict))
201
202     @staticmethod
203     def _extract_track(t, fatal=True):
204         audio_url = t['URL'] if fatal else t.get('URL')
205         if not audio_url:
206             return
207
208         audio_id = t['File'] if fatal else t.get('File')
209         if not audio_id:
210             return
211
212         thumbnail = t.get('AlbumCoverURL') or t.get('FiledAlbumCover')
213         uploader = t.get('OwnerName') or t.get('OwnerName_Text_HTML')
214         uploader_id = t.get('UploaderID')
215         duration = int_or_none(t.get('DurationInSeconds')) or parse_duration(
216             t.get('Duration') or t.get('DurationStr'))
217         view_count = int_or_none(t.get('PlayCount') or t.get('PlayCount_hr'))
218
219         track = t.get('Name') or t.get('Name_Text_HTML')
220         artist = t.get('Author') or t.get('Author_Text_HTML')
221
222         if track:
223             title = '%s - %s' % (artist, track) if artist else track
224         else:
225             title = audio_id
226
227         return {
228             'extractor_key': MailRuMusicIE.ie_key(),
229             'id': audio_id,
230             'title': title,
231             'thumbnail': thumbnail,
232             'uploader': uploader,
233             'uploader_id': uploader_id,
234             'duration': duration,
235             'view_count': view_count,
236             'vcodec': 'none',
237             'abr': int_or_none(t.get('BitRate')),
238             'track': track,
239             'artist': artist,
240             'album': t.get('Album'),
241             'url': audio_url,
242         }
243
244
245 class MailRuMusicIE(MailRuMusicSearchBaseIE):
246     IE_NAME = 'mailru:music'
247     IE_DESC = 'Музыка@Mail.Ru'
248     _VALID_URL = r'https?://my\.mail\.ru/+music/+songs/+[^/?#&]+-(?P<id>[\da-f]+)'
249     _TESTS = [{
250         'url': 'https://my.mail.ru/music/songs/%D0%BC8%D0%BB8%D1%82%D1%85-l-a-h-luciferian-aesthetics-of-herrschaft-single-2017-4e31f7125d0dfaef505d947642366893',
251         'md5': '0f8c22ef8c5d665b13ac709e63025610',
252         'info_dict': {
253             'id': '4e31f7125d0dfaef505d947642366893',
254             'ext': 'mp3',
255             'title': 'L.A.H. (Luciferian Aesthetics of Herrschaft) single, 2017 - М8Л8ТХ',
256             'uploader': 'Игорь Мудрый',
257             'uploader_id': '1459196328',
258             'duration': 280,
259             'view_count': int,
260             'vcodec': 'none',
261             'abr': 320,
262             'track': 'L.A.H. (Luciferian Aesthetics of Herrschaft) single, 2017',
263             'artist': 'М8Л8ТХ',
264         },
265     }]
266
267     def _real_extract(self, url):
268         audio_id = self._match_id(url)
269
270         webpage = self._download_webpage(url, audio_id)
271
272         title = self._og_search_title(webpage)
273         music_data = self._search(title, url, audio_id)['MusicData']
274         t = next(t for t in music_data if t.get('File') == audio_id)
275
276         info = self._extract_track(t)
277         info['title'] = title
278         return info
279
280
281 class MailRuMusicSearchIE(MailRuMusicSearchBaseIE):
282     IE_NAME = 'mailru:music:search'
283     IE_DESC = 'Музыка@Mail.Ru'
284     _VALID_URL = r'https?://my\.mail\.ru/+music/+search/+(?P<id>[^/?#&]+)'
285     _TESTS = [{
286         'url': 'https://my.mail.ru/music/search/black%20shadow',
287         'info_dict': {
288             'id': 'black shadow',
289         },
290         'playlist_mincount': 532,
291     }]
292
293     def _real_extract(self, url):
294         query = compat_urllib_parse_unquote(self._match_id(url))
295
296         entries = []
297
298         LIMIT = 100
299         offset = 0
300
301         for _ in itertools.count(1):
302             search = self._search(query, url, query, LIMIT, offset)
303
304             music_data = search.get('MusicData')
305             if not music_data or not isinstance(music_data, list):
306                 break
307
308             for t in music_data:
309                 track = self._extract_track(t, fatal=False)
310                 if track:
311                     entries.append(track)
312
313             total = try_get(
314                 search, lambda x: x['Results']['music']['Total'], int)
315
316             if total is not None:
317                 if offset > total:
318                     break
319
320             offset += LIMIT
321
322         return self.playlist_result(entries, query)