]> jfr.im git - yt-dlp.git/blame - yt_dlp/extractor/vk.py
[ie/vk] Improve format extraction (#9885)
[yt-dlp.git] / yt_dlp / extractor / vk.py
CommitLineData
51815886 1import collections
59f63c8f 2import hashlib
60d142aa 3import re
60d142aa
JMF
4
5from .common import InfoExtractor
59f63c8f
M
6from .dailymotion import DailymotionIE
7from .odnoklassniki import OdnoklassnikiIE
8from .pladform import PladformIE
7991ae57 9from .sibnet import SibnetEmbedIE
59f63c8f
M
10from .vimeo import VimeoIE
11from .youtube import YoutubeIE
60d142aa 12from ..utils import (
9032dc28 13 ExtractorError,
8776349e 14 UserNotLive,
59f63c8f 15 clean_html,
2d19fb50 16 get_element_by_class,
a9c68545 17 get_element_html_by_id,
bf4b3b6b 18 int_or_none,
a9c68545 19 join_nonempty,
8776349e 20 parse_resolution,
ad1bc71a 21 str_or_none,
8117df4c 22 str_to_int,
a9c68545 23 try_call,
60d142aa 24 unescapeHTML,
a7ee8a00 25 unified_timestamp,
59f63c8f 26 update_url_query,
3052a30d 27 url_or_none,
6e6bc8da 28 urlencode_postdata,
a9c68545 29 urljoin,
8776349e 30 traverse_obj,
1cc79574 31)
60d142aa
JMF
32
33
2d19fb50
S
34class VKBaseIE(InfoExtractor):
35 _NETRC_MACHINE = 'vk'
36
59f63c8f
M
37 def _download_webpage_handle(self, url_or_request, video_id, *args, fatal=True, **kwargs):
38 response = super()._download_webpage_handle(url_or_request, video_id, *args, fatal=fatal, **kwargs)
3d2623a8 39 challenge_url, cookie = response[1].url if response else '', None
59f63c8f
M
40 if challenge_url.startswith('https://vk.com/429.html?'):
41 cookie = self._get_cookies(challenge_url).get('hash429')
42 if not cookie:
43 return response
44
45 hash429 = hashlib.md5(cookie.value.encode('ascii')).hexdigest()
46 self._request_webpage(
47 update_url_query(challenge_url, {'key': hash429}), video_id, fatal=fatal,
48 note='Resolving WAF challenge', errnote='Failed to bypass WAF challenge')
49 return super()._download_webpage_handle(url_or_request, video_id, *args, fatal=True, **kwargs)
50
52efa4b3 51 def _perform_login(self, username, password):
2d19fb50
S
52 login_page, url_handle = self._download_webpage_handle(
53 'https://vk.com', None, 'Downloading login page')
54
55 login_form = self._hidden_inputs(login_page)
56
57 login_form.update({
58 'email': username.encode('cp1251'),
59 'pass': password.encode('cp1251'),
60 })
61
e3c1266f
S
62 # vk serves two same remixlhk cookies in Set-Cookie header and expects
63 # first one to be actually set
64 self._apply_first_set_cookie_header(url_handle, 'remixlhk')
2d19fb50
S
65
66 login_page = self._download_webpage(
f0ffaa16 67 'https://vk.com/login', None,
e4d95865 68 note='Logging in',
2d19fb50
S
69 data=urlencode_postdata(login_form))
70
71 if re.search(r'onLoginFailed', login_page):
72 raise ExtractorError(
73 'Unable to login, incorrect username and/or password', expected=True)
74
3c989818 75 def _download_payload(self, path, video_id, data, fatal=True):
59f63c8f 76 endpoint = f'https://vk.com/{path}.php'
3c989818
RA
77 data['al'] = 1
78 code, payload = self._download_json(
59f63c8f
M
79 endpoint, video_id, data=urlencode_postdata(data), fatal=fatal,
80 headers={
81 'Referer': endpoint,
82 'X-Requested-With': 'XMLHttpRequest',
83 })['payload']
3c989818
RA
84 if code == '3':
85 self.raise_login_required()
86 elif code == '8':
87 raise ExtractorError(clean_html(payload[0][1:-1]), expected=True)
88 return payload
89
2d19fb50
S
90
91class VKIE(VKBaseIE):
1ecb5d1d
S
92 IE_NAME = 'vk'
93 IE_DESC = 'VK'
bfd973ec 94 _EMBED_REGEX = [r'<iframe[^>]+?src=(["\'])(?P<url>https?://vk\.com/video_ext\.php.+?)\1']
cf9cf7dd
S
95 _VALID_URL = r'''(?x)
96 https?://
97 (?:
04e88ca2 98 (?:
bdafd88d 99 (?:(?:m|new)\.)?vk\.com/video_|
b634ba74 100 (?:www\.)?daxab\.com/
04e88ca2 101 )
102 ext\.php\?(?P<embed_query>.*?\boid=(?P<oid>-?\d+).*?\bid=(?P<id>\d+).*)|
cf9cf7dd 103 (?:
21df2117 104 (?:(?:m|new)\.)?vk\.com/(?:.+?\?.*?z=)?(?:video|clip)|
b634ba74 105 (?:www\.)?daxab\.com/embed/
cf9cf7dd 106 )
af3cbd87 107 (?P<videoid>-?\d+_\d+)(?:.*\blist=(?P<list_id>([\da-f]+)|(ln-[\da-zA-Z]+)))?
cf9cf7dd
S
108 )
109 '''
7991ae57 110
9032dc28
S
111 _TESTS = [
112 {
113 'url': 'http://vk.com/videos-77521?z=video-77521_162222515%2Fclub77521',
9032dc28 114 'info_dict': {
220828f2 115 'id': '-77521_162222515',
09f934b0 116 'ext': 'mp4',
9032dc28 117 'title': 'ProtivoGunz - Хуёвая песня',
36300346 118 'uploader': 're:(?:Noize MC|Alexander Ilyashenko).*',
59f63c8f 119 'uploader_id': '39545378',
9032dc28 120 'duration': 195,
ad1bc71a 121 'timestamp': 1329049880,
42e1ff86 122 'upload_date': '20120212',
59f63c8f
M
123 'comment_count': int,
124 'like_count': int,
a9c68545 125 'thumbnail': r're:https?://.+(?:\.jpg|getVideoPreview.*)$',
9032dc28 126 },
59f63c8f 127 'params': {'skip_download': 'm3u8'},
60d142aa 128 },
9032dc28 129 {
c52331f3 130 'url': 'http://vk.com/video205387401_165548505',
9032dc28 131 'info_dict': {
220828f2 132 'id': '205387401_165548505',
9032dc28 133 'ext': 'mp4',
c52331f3 134 'title': 'No name',
ad1bc71a
RA
135 'uploader': 'Tom Cruise',
136 'uploader_id': '205387401',
c52331f3 137 'duration': 9,
ad1bc71a
RA
138 'timestamp': 1374364108,
139 'upload_date': '20130720',
59f63c8f
M
140 'comment_count': int,
141 'like_count': int,
a9c68545 142 'thumbnail': r're:https?://.+(?:\.jpg|getVideoPreview.*)$',
9032dc28
S
143 }
144 },
ca97a56e
S
145 {
146 'note': 'Embedded video',
3c989818 147 'url': 'https://vk.com/video_ext.php?oid=-77521&id=162222515&hash=87b046504ccd8bfa',
ca97a56e 148 'info_dict': {
3c989818 149 'id': '-77521_162222515',
ca97a56e 150 'ext': 'mp4',
3c989818
RA
151 'uploader': 're:(?:Noize MC|Alexander Ilyashenko).*',
152 'title': 'ProtivoGunz - Хуёвая песня',
153 'duration': 195,
154 'upload_date': '20120212',
155 'timestamp': 1329049880,
59f63c8f 156 'uploader_id': '39545378',
a9c68545 157 'thumbnail': r're:https?://.+(?:\.jpg|getVideoPreview.*)$',
04e88ca2 158 },
59f63c8f 159 'params': {'skip_download': 'm3u8'},
ca97a56e 160 },
af3cbd87 161 {
162 'url': 'https://vk.com/video-93049196_456239755?list=ln-cBjJ7S4jYYx3ADnmDT',
163 'info_dict': {
164 'id': '-93049196_456239755',
165 'ext': 'mp4',
166 'title': '8 серия (озвучка)',
167 'duration': 8383,
59f63c8f
M
168 'comment_count': int,
169 'uploader': 'Dizi2021',
170 'like_count': int,
171 'timestamp': 1640162189,
af3cbd87 172 'upload_date': '20211222',
59f63c8f 173 'uploader_id': '-93049196',
a9c68545 174 'thumbnail': r're:https?://.+(?:\.jpg|getVideoPreview.*)$',
af3cbd87 175 },
176 },
79913fde 177 {
a9c68545 178 'note': 'youtube embed',
9281f6d2
S
179 'url': 'https://vk.com/video276849682_170681728',
180 'info_dict': {
181 'id': 'V3K4mi0SYkc',
220828f2 182 'ext': 'mp4',
9281f6d2 183 'title': "DSWD Awards 'Children's Joy Foundation, Inc.' Certificate of Registration and License to Operate",
ad1bc71a 184 'description': 'md5:bf9c26cfa4acdfb146362682edd3827a',
220828f2 185 'duration': 178,
59f63c8f 186 'upload_date': '20130117',
ad1bc71a 187 'uploader': "Children's Joy Foundation Inc.",
9281f6d2
S
188 'uploader_id': 'thecjf',
189 'view_count': int,
59f63c8f
M
190 'channel_id': 'UCgzCNQ11TmR9V97ECnhi3gw',
191 'availability': 'public',
192 'like_count': int,
193 'live_status': 'not_live',
194 'playable_in_embed': True,
195 'channel': 'Children\'s Joy Foundation Inc.',
196 'uploader_url': 'http://www.youtube.com/user/thecjf',
197 'thumbnail': r're:https?://.+\.jpg$',
198 'tags': 'count:27',
199 'start_time': 0.0,
200 'categories': ['Nonprofits & Activism'],
201 'channel_url': 'https://www.youtube.com/channel/UCgzCNQ11TmR9V97ECnhi3gw',
a9c68545
M
202 'channel_follower_count': int,
203 'age_limit': 0,
204 },
205 },
206 {
207 'note': 'dailymotion embed',
208 'url': 'https://vk.com/video-95168827_456239103?list=cca524a0f0d5557e16',
209 'info_dict': {
210 'id': 'x8gfli0',
211 'ext': 'mp4',
212 'title': 'md5:45410f60ccd4b2760da98cb5fc777d70',
213 'description': 'md5:2e71c5c9413735cfa06cf1a166f16c84',
214 'uploader': 'Movies and cinema.',
215 'upload_date': '20221218',
216 'uploader_id': 'x1jdavv',
217 'timestamp': 1671387617,
59f63c8f 218 'age_limit': 0,
a9c68545
M
219 'duration': 2918,
220 'like_count': int,
221 'view_count': int,
222 'thumbnail': r're:https?://.+x1080$',
223 'tags': list
9281f6d2
S
224 },
225 },
e3845525 226 {
a9c68545 227 'url': 'https://vk.com/clips-74006511?z=clip-74006511_456247211',
e3845525 228 'info_dict': {
a9c68545 229 'id': '-74006511_456247211',
e3845525 230 'ext': 'mp4',
a9c68545
M
231 'comment_count': int,
232 'duration': 9,
233 'like_count': int,
234 'thumbnail': r're:https?://.+(?:\.jpg|getVideoPreview.*)$',
235 'timestamp': 1664995597,
236 'title': 'Clip by @madempress',
237 'upload_date': '20221005',
238 'uploader': 'Шальная императрица',
239 'uploader_id': '-74006511',
e3845525 240 },
e3845525 241 },
bf4b3b6b
S
242 {
243 # video key is extra_data not url\d+
244 'url': 'http://vk.com/video-110305615_171782105',
245 'md5': 'e13fcda136f99764872e739d13fac1d1',
246 'info_dict': {
220828f2 247 'id': '-110305615_171782105',
bf4b3b6b
S
248 'ext': 'mp4',
249 'title': 'S-Dance, репетиции к The way show',
250 'uploader': 'THE WAY SHOW | 17 апреля',
ad1bc71a
RA
251 'uploader_id': '-110305615',
252 'timestamp': 1454859345,
bf4b3b6b 253 'upload_date': '20160207',
ad1bc71a 254 },
59f63c8f 255 'skip': 'Removed',
bf4b3b6b 256 },
93aa0b63 257 {
a9c68545 258 'note': 'finished live stream, postlive_mp4',
93aa0b63 259 'url': 'https://vk.com/videos-387766?z=video-387766_456242764%2Fpl_-387766_-2',
93aa0b63 260 'info_dict': {
220828f2 261 'id': '-387766_456242764',
93aa0b63 262 'ext': 'mp4',
220828f2 263 'title': 'ИгроМир 2016 День 1 — Игромания Утром',
93aa0b63
S
264 'uploader': 'Игромания',
265 'duration': 5239,
220828f2
RA
266 'upload_date': '20160929',
267 'uploader_id': '-387766',
268 'timestamp': 1475137527,
59f63c8f
M
269 'thumbnail': r're:https?://.+\.jpg$',
270 'comment_count': int,
271 'like_count': int,
93aa0b63 272 },
3c989818
RA
273 'params': {
274 'skip_download': True,
275 },
93aa0b63 276 },
475f8a45 277 {
424ed37e 278 # live stream, hls and rtmp links, most likely already finished live
475f8a45
S
279 # stream by the time you are reading this comment
280 'url': 'https://vk.com/video-140332_456239111',
281 'only_matching': True,
282 },
a8363f3a
PH
283 {
284 # removed video, just testing that we match the pattern
285 'url': 'http://vk.com/feed?z=video-43215063_166094326%2Fbb50cacd3177146d7a',
286 'only_matching': True,
287 },
e58066e2
S
288 {
289 # age restricted video, requires vk account credentials
290 'url': 'https://vk.com/video205387401_164765225',
291 'only_matching': True,
292 },
a5e52a1f
S
293 {
294 # pladform embed
295 'url': 'https://vk.com/video-76116461_171554880',
296 'only_matching': True,
bdafd88d
S
297 },
298 {
299 'url': 'http://new.vk.com/video205387401_165548505',
300 'only_matching': True,
643dc0fc
CP
301 },
302 {
303 # This video is no longer available, because its author has been blocked.
304 'url': 'https://vk.com/video-10639516_456240611',
305 'only_matching': True,
a640c4d2 306 },
307 {
308 # The video is not available in your region.
309 'url': 'https://vk.com/video-51812607_171445436',
310 'only_matching': True,
21df2117 311 },
312 {
313 'url': 'https://vk.com/clip30014565_456240946',
314 'only_matching': True,
a640c4d2 315 }]
9032dc28 316
60d142aa 317 def _real_extract(self, url):
5ad28e7f 318 mobj = self._match_valid_url(url)
ca97a56e
S
319 video_id = mobj.group('videoid')
320
3c989818 321 mv_data = {}
04e88ca2 322 if video_id:
3c989818 323 data = {
59f63c8f 324 'act': 'show',
3c989818
RA
325 'video': video_id,
326 }
04e88ca2 327 # Some videos (removed?) can only be downloaded with list id specified
328 list_id = mobj.group('list_id')
329 if list_id:
3c989818
RA
330 data['list'] = list_id
331
332 payload = self._download_payload('al_video', video_id, data)
333 info_page = payload[1]
334 opts = payload[-1]
335 mv_data = opts.get('mvData') or {}
336 player = opts.get('player') or {}
04e88ca2 337 else:
ca97a56e 338 video_id = '%s_%s' % (mobj.group('oid'), mobj.group('id'))
9032dc28 339
3c989818
RA
340 info_page = self._download_webpage(
341 'http://vk.com/video_ext.php?' + mobj.group('embed_query'), video_id)
9032dc28 342
3c989818
RA
343 error_message = self._html_search_regex(
344 [r'(?s)<!><div[^>]+class="video_layer_message"[^>]*>(.+?)</div>',
345 r'(?s)<div[^>]+id="video_ext_msg"[^>]*>(.+?)</div>'],
346 info_page, 'error message', default=None)
347 if error_message:
348 raise ExtractorError(error_message, expected=True)
ee48b6a8 349
3c989818
RA
350 if re.search(r'<!>/login\.php\?.*\bact=security_check', info_page):
351 raise ExtractorError(
352 'You are trying to log in from an unusual location. You should confirm ownership at vk.com to log in with this IP.',
353 expected=True)
7f220b2f 354
3c989818 355 ERROR_COPYRIGHT = 'Video %s has been removed from public access due to rightholder complaint.'
1d1d60f6 356
3c989818
RA
357 ERRORS = {
358 r'>Видеозапись .*? была изъята из публичного доступа в связи с обращением правообладателя.<':
359 ERROR_COPYRIGHT,
1d1d60f6 360
3c989818
RA
361 r'>The video .*? was removed from public access by request of the copyright holder.<':
362 ERROR_COPYRIGHT,
3d36cea4 363
3c989818
RA
364 r'<!>Please log in or <':
365 'Video %s is only available for registered users, '
366 'use --username and --password options to provide account credentials.',
3d36cea4 367
3c989818
RA
368 r'<!>Unknown error':
369 'Video %s does not exist.',
1aa5172f 370
3c989818
RA
371 r'<!>Видео временно недоступно':
372 'Video %s is temporarily unavailable.',
d919fa33 373
3c989818
RA
374 r'<!>Access denied':
375 'Access denied to video %s.',
643dc0fc 376
3c989818
RA
377 r'<!>Видеозапись недоступна, так как её автор был заблокирован.':
378 'Video %s is no longer available, because its author has been blocked.',
643dc0fc 379
3c989818
RA
380 r'<!>This video is no longer available, because its author has been blocked.':
381 'Video %s is no longer available, because its author has been blocked.',
ad1bc71a 382
3c989818
RA
383 r'<!>This video is no longer available, because it has been deleted.':
384 'Video %s is no longer available, because it has been deleted.',
a640c4d2 385
3c989818
RA
386 r'<!>The video .+? is not available in your region.':
387 'Video %s is not available in your region.',
388 }
389
390 for error_re, error_msg in ERRORS.items():
391 if re.search(error_re, info_page):
392 raise ExtractorError(error_msg % video_id, expected=True)
9032dc28 393
3c989818
RA
394 player = self._parse_json(self._search_regex(
395 r'var\s+playerParams\s*=\s*({.+?})\s*;\s*\n',
396 info_page, 'player params'), video_id)
9334f8f1 397
5113b691 398 youtube_url = YoutubeIE._extract_url(info_page)
46478456 399 if youtube_url:
3c989818 400 return self.url_result(youtube_url, YoutubeIE.ie_key())
849086a1 401
09b9c45e 402 vimeo_url = VimeoIE._extract_url(url, info_page)
84663361 403 if vimeo_url is not None:
3c989818 404 return self.url_result(vimeo_url, VimeoIE.ie_key())
84663361 405
c4737bea
S
406 pladform_url = PladformIE._extract_url(info_page)
407 if pladform_url:
3c989818 408 return self.url_result(pladform_url, PladformIE.ie_key())
c4737bea 409
7a1818c9 410 m_rutube = re.search(
35972ba1 411 r'\ssrc="((?:https?:)?//rutube\.ru\\?/(?:video|play)\\?/embed(?:.*?))\\?"', info_page)
7a1818c9 412 if m_rutube is not None:
7a1818c9
PH
413 rutube_url = self._proto_relative_url(
414 m_rutube.group(1).replace('\\', ''))
415 return self.url_result(rutube_url)
416
43aebb7d 417 dailymotion_url = next(DailymotionIE._extract_embed_urls(url, info_page), None)
418 if dailymotion_url:
419 return self.url_result(dailymotion_url, DailymotionIE.ie_key())
e3845525 420
3c989818
RA
421 odnoklassniki_url = OdnoklassnikiIE._extract_url(info_page)
422 if odnoklassniki_url:
423 return self.url_result(odnoklassniki_url, OdnoklassnikiIE.ie_key())
424
7991ae57 425 sibnet_url = next(SibnetEmbedIE._extract_embed_urls(url, info_page), None)
43aebb7d 426 if sibnet_url:
427 return self.url_result(sibnet_url)
b73612a2 428
054932f4 429 m_opts = re.search(r'(?s)var\s+opts\s*=\s*({.+?});', info_page)
849086a1 430 if m_opts:
054932f4 431 m_opts_url = re.search(r"url\s*:\s*'((?!/\b)[^']+)", m_opts.group(1))
849086a1
S
432 if m_opts_url:
433 opts_url = m_opts_url.group(1)
434 if opts_url.startswith('//'):
435 opts_url = 'http:' + opts_url
436 return self.url_result(opts_url)
437
3c989818 438 data = player['params'][0]
475f8a45
S
439 title = unescapeHTML(data['md_title'])
440
424ed37e
S
441 # 2 = live
442 # 3 = post live (finished live)
9cdb0a33 443 is_live = data.get('live') == 2
475f8a45 444
a7ee8a00 445 timestamp = unified_timestamp(self._html_search_regex(
70d7b323 446 r'class=["\']mv_info_date[^>]+>([^<]+)(?:<|from)', info_page,
ad1bc71a 447 'upload date', default=None)) or int_or_none(data.get('date'))
3aa3953d 448
70d7b323
S
449 view_count = str_to_int(self._search_regex(
450 r'class=["\']mv_views_count[^>]+>\s*([\d,.]+)',
498a8a4c 451 info_page, 'view count', default=None))
8117df4c 452
bf4b3b6b 453 formats = []
df5c9e73 454 subtitles = {}
475f8a45 455 for format_id, format_url in data.items():
3052a30d
S
456 format_url = url_or_none(format_url)
457 if not format_url or not format_url.startswith(('http', '//', 'rtmp')):
bf4b3b6b 458 continue
3089bc74
S
459 if (format_id.startswith(('url', 'cache'))
460 or format_id in ('extra_data', 'live_mp4', 'postlive_mp4')):
475f8a45
S
461 height = int_or_none(self._search_regex(
462 r'^(?:url|cache)(\d+)', format_id, 'height', default=None))
463 formats.append({
464 'format_id': format_id,
465 'url': format_url,
df5c9e73 466 'ext': 'mp4',
467 'source_preference': 1,
475f8a45
S
468 'height': height,
469 })
470 elif format_id == 'hls':
df5c9e73 471 fmts, subs = self._extract_m3u8_formats_and_subtitles(
fb4fc449 472 format_url, video_id, 'mp4', 'm3u8_native',
df5c9e73 473 m3u8_id=format_id, fatal=False, live=is_live)
474 formats.extend(fmts)
475 self._merge_subtitles(subs, target=subtitles)
476 elif format_id.startswith('dash_'):
477 fmts, subs = self._extract_mpd_formats_and_subtitles(
478 format_url, video_id, mpd_id=format_id, fatal=False)
479 formats.extend(fmts)
480 self._merge_subtitles(subs, target=subtitles)
475f8a45
S
481 elif format_id == 'rtmp':
482 formats.append({
483 'format_id': format_id,
484 'url': format_url,
485 'ext': 'flv',
486 })
913f3292 487
5b6cb562 488 for sub in data.get('subs') or {}:
489 subtitles.setdefault(sub.get('lang', 'en'), []).append({
490 'ext': sub.get('title', '.srt').split('.')[-1],
491 'url': url_or_none(sub.get('url')),
492 })
493
60d142aa 494 return {
220828f2 495 'id': video_id,
913f3292 496 'formats': formats,
475f8a45 497 'title': title,
913f3292
PH
498 'thumbnail': data.get('jpg'),
499 'uploader': data.get('md_author'),
3c989818
RA
500 'uploader_id': str_or_none(data.get('author_id') or mv_data.get('authorId')),
501 'duration': int_or_none(data.get('duration') or mv_data.get('duration')),
a7ee8a00 502 'timestamp': timestamp,
8117df4c 503 'view_count': view_count,
3c989818
RA
504 'like_count': int_or_none(mv_data.get('likes')),
505 'comment_count': int_or_none(mv_data.get('commcount')),
9cdb0a33 506 'is_live': is_live,
5b6cb562 507 'subtitles': subtitles,
df5c9e73 508 '_format_sort_fields': ('res', 'source'),
60d142aa 509 }
469d4c89
WS
510
511
2d19fb50 512class VKUserVideosIE(VKBaseIE):
1ecb5d1d
S
513 IE_NAME = 'vk:uservideos'
514 IE_DESC = "VK - User's Videos"
5d14b734 515 _VALID_URL = r'https?://(?:(?:m|new)\.)?vk\.com/video/(?:playlist/)?(?P<id>[^?$#/&]+)(?!\?.*\bz=video)(?:[/?#&](?:.*?\bsection=(?P<section>\w+))?|$)'
469d4c89 516 _TEMPLATE_URL = 'https://vk.com/videos'
dc786d3d 517 _TESTS = [{
a70b71e8 518 'url': 'https://vk.com/video/@mobidevices',
0e6ec3ca 519 'info_dict': {
a70b71e8 520 'id': '-17892518_all',
0e6ec3ca 521 },
a70b71e8 522 'playlist_mincount': 1355,
0e6ec3ca 523 }, {
a70b71e8 524 'url': 'https://vk.com/video/@mobidevices?section=uploaded',
15ec6693 525 'info_dict': {
a70b71e8 526 'id': '-17892518_uploaded',
15ec6693 527 },
a70b71e8 528 'playlist_mincount': 182,
5d14b734
M
529 }, {
530 'url': 'https://vk.com/video/playlist/-174476437_2',
531 'info_dict': {
a9c68545 532 'id': '-174476437_playlist_2',
5d14b734
M
533 'title': 'Анонсы'
534 },
535 'playlist_mincount': 108,
dc786d3d 536 }]
0e6ec3ca 537 _VIDEO = collections.namedtuple('Video', ['owner_id', 'id'])
dc786d3d 538
a70b71e8
AG
539 def _entries(self, page_id, section):
540 video_list_json = self._download_payload('al_video', page_id, {
3c989818 541 'act': 'load_videos_silent',
a70b71e8 542 'offset': 0,
3c989818 543 'oid': page_id,
0e6ec3ca 544 'section': section,
a70b71e8
AG
545 })[0][section]
546 count = video_list_json['count']
547 total = video_list_json['total']
548 video_list = video_list_json['list']
549
550 while True:
551 for video in video_list:
552 v = self._VIDEO._make(video[:2])
553 video_id = '%d_%d' % (v.owner_id, v.id)
554 yield self.url_result(
555 'http://vk.com/video' + video_id, VKIE.ie_key(), video_id)
556 if count >= total:
557 break
558 video_list_json = self._download_payload('al_video', page_id, {
559 'act': 'load_videos_silent',
560 'offset': count,
561 'oid': page_id,
562 'section': section,
563 })[0][section]
564 count += video_list_json['count']
565 video_list = video_list_json['list']
0e6ec3ca
RA
566
567 def _real_extract(self, url):
a70b71e8
AG
568 u_id, section = self._match_valid_url(url).groups()
569 webpage = self._download_webpage(url, u_id)
5d14b734
M
570
571 if u_id.startswith('@'):
572 page_id = self._search_regex(r'data-owner-id\s?=\s?"([^"]+)"', webpage, 'page_id')
573 elif '_' in u_id:
574 page_id, section = u_id.split('_', 1)
a9c68545 575 section = f'playlist_{section}'
5d14b734
M
576 else:
577 raise ExtractorError('Invalid URL', expected=True)
578
0e6ec3ca
RA
579 if not section:
580 section = 'all'
581
5d14b734
M
582 playlist_title = clean_html(get_element_by_class('VideoInfoPanel__title', webpage))
583 return self.playlist_result(self._entries(page_id, section), '%s_%s' % (page_id, section), playlist_title)
2d19fb50
S
584
585
586class VKWallPostIE(VKBaseIE):
587 IE_NAME = 'vk:wallpost'
588 _VALID_URL = r'https?://(?:(?:(?:(?:m|new)\.)?vk\.com/(?:[^?]+\?.*\bw=)?wall(?P<id>-?\d+_\d+)))'
589 _TESTS = [{
590 # public page URL, audio playlist
591 'url': 'https://vk.com/bs.official?w=wall-23538238_35',
592 'info_dict': {
3c989818
RA
593 'id': '-23538238_35',
594 'title': 'Black Shadow - Wall post -23538238_35',
a9c68545 595 'description': 'md5:190c78f905a53e0de793d83933c6e67f',
2d19fb50
S
596 },
597 'playlist': [{
598 'md5': '5ba93864ec5b85f7ce19a9af4af080f6',
599 'info_dict': {
600 'id': '135220665_111806521',
a9c68545 601 'ext': 'm4a',
2d19fb50
S
602 'title': 'Black Shadow - Слепое Верование',
603 'duration': 370,
604 'uploader': 'Black Shadow',
605 'artist': 'Black Shadow',
606 'track': 'Слепое Верование',
607 },
608 }, {
609 'md5': '4cc7e804579122b17ea95af7834c9233',
610 'info_dict': {
611 'id': '135220665_111802303',
a9c68545 612 'ext': 'm4a',
2d19fb50
S
613 'title': 'Black Shadow - Война - Негасимое Бездны Пламя!',
614 'duration': 423,
615 'uploader': 'Black Shadow',
616 'artist': 'Black Shadow',
617 'track': 'Война - Негасимое Бездны Пламя!',
618 },
2d19fb50 619 }],
51815886 620 'params': {
3c989818 621 'skip_download': True,
51815886 622 },
2d19fb50 623 }, {
a9c68545
M
624 # single YouTube embed with irrelevant reaction videos
625 'url': 'https://vk.com/wall-32370614_7173954',
2d19fb50 626 'info_dict': {
a9c68545
M
627 'id': '-32370614_7173954',
628 'title': 'md5:9f93c405bbc00061d34007d78c75e3bc',
629 'description': 'md5:953b811f26fa9f21ee5856e2ea8e68fc',
2d19fb50
S
630 },
631 'playlist_count': 1,
2d19fb50
S
632 }, {
633 # wall page URL
634 'url': 'https://vk.com/wall-23538238_35',
635 'only_matching': True,
636 }, {
637 # mobile wall page URL
638 'url': 'https://m.vk.com/wall-23538238_35',
639 'only_matching': True,
640 }]
3c989818 641 _BASE64_CHARS = 'abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMN0PQRSTUVWXYZO123456789+/='
0e6ec3ca 642 _AUDIO = collections.namedtuple('Audio', ['id', 'owner_id', 'url', 'title', 'performer', 'duration', 'album_id', 'unk', 'author_link', 'lyrics', 'flags', 'context', 'extra', 'hashes', 'cover_url', 'ads'])
3c989818
RA
643
644 def _decode(self, enc):
645 dec = ''
646 e = n = 0
647 for c in enc:
648 r = self._BASE64_CHARS.index(c)
649 cond = n % 4
650 e = 64 * e + r if cond else r
651 n += 1
652 if cond:
653 dec += chr(255 & e >> (-2 * n & 6))
654 return dec
655
656 def _unmask_url(self, mask_url, vk_id):
657 if 'audio_api_unavailable' in mask_url:
658 extra = mask_url.split('?extra=')[1].split('#')
659 func, base = self._decode(extra[1]).split(chr(11))
3c989818
RA
660 mask_url = list(self._decode(extra[0]))
661 url_len = len(mask_url)
662 indexes = [None] * url_len
663 index = int(base) ^ vk_id
664 for n in range(url_len - 1, -1, -1):
665 index = (url_len * (n + 1) ^ index + n) % url_len
666 indexes[n] = index
667 for n in range(1, url_len):
668 c = mask_url[n]
669 index = indexes[url_len - 1 - n]
670 mask_url[n] = mask_url[index]
671 mask_url[index] = c
672 mask_url = ''.join(mask_url)
673 return mask_url
2d19fb50
S
674
675 def _real_extract(self, url):
676 post_id = self._match_id(url)
677
3c989818
RA
678 webpage = self._download_payload('wkview', post_id, {
679 'act': 'show',
680 'w': 'wall' + post_id,
681 })[1]
2d19fb50 682
a9c68545 683 uploader = clean_html(get_element_by_class('PostHeaderTitle__authorName', webpage))
2d19fb50
S
684
685 entries = []
686
3c989818
RA
687 for audio in re.findall(r'data-audio="([^"]+)', webpage):
688 audio = self._parse_json(unescapeHTML(audio), post_id)
a9c68545 689 if not audio['url']:
3c989818 690 continue
a9c68545
M
691 title = unescapeHTML(audio.get('title'))
692 artist = unescapeHTML(audio.get('artist'))
3c989818 693 entries.append({
a9c68545
M
694 'id': f'{audio["owner_id"]}_{audio["id"]}',
695 'title': join_nonempty(artist, title, delim=' - '),
696 'thumbnails': try_call(lambda: [{'url': u} for u in audio['coverUrl'].split(',')]),
697 'duration': int_or_none(audio.get('duration')),
3c989818 698 'uploader': uploader,
a9c68545 699 'artist': artist,
3c989818 700 'track': title,
a9c68545
M
701 'formats': [{
702 'url': audio['url'],
703 'ext': 'm4a',
704 'vcodec': 'none',
705 'acodec': 'mp3',
706 'container': 'm4a_dash',
707 }],
3c989818 708 })
2d19fb50 709
a9c68545
M
710 entries.extend(self.url_result(urljoin(url, entry), VKIE) for entry in set(re.findall(
711 r'<a[^>]+href=(?:["\'])(/video(?:-?[\d_]+)[^"\']*)',
712 get_element_html_by_id('wl_post_body', webpage))))
2d19fb50
S
713
714 return self.playlist_result(
a9c68545
M
715 entries, post_id, join_nonempty(uploader, f'Wall post {post_id}', delim=' - '),
716 clean_html(get_element_by_class('wall_post_text', webpage)))
8776349e 717
718
719class VKPlayBaseIE(InfoExtractor):
b15b0c1d 720 _BASE_URL_RE = r'https?://(?:vkplay\.live|live\.vkplay\.ru)/'
8776349e 721 _RESOLUTIONS = {
722 'tiny': '256x144',
723 'lowest': '426x240',
724 'low': '640x360',
725 'medium': '852x480',
726 'high': '1280x720',
727 'full_hd': '1920x1080',
728 'quad_hd': '2560x1440',
729 }
730
731 def _extract_from_initial_state(self, url, video_id, path):
732 webpage = self._download_webpage(url, video_id)
733 video_info = traverse_obj(self._search_json(
734 r'<script[^>]+\bid="initial-state"[^>]*>', webpage, 'initial state', video_id),
735 path, expected_type=dict)
736 if not video_info:
737 raise ExtractorError('Unable to extract video info from html inline initial state')
738 return video_info
739
740 def _extract_formats(self, stream_info, video_id):
741 formats = []
742 for stream in traverse_obj(stream_info, (
743 'data', 0, 'playerUrls', lambda _, v: url_or_none(v['url']) and v['type'])):
744 url = stream['url']
745 format_id = str_or_none(stream['type'])
746 if format_id in ('hls', 'live_hls', 'live_playback_hls') or '.m3u8' in url:
747 formats.extend(self._extract_m3u8_formats(url, video_id, m3u8_id=format_id, fatal=False))
748 elif format_id == 'dash':
749 formats.extend(self._extract_mpd_formats(url, video_id, mpd_id=format_id, fatal=False))
750 elif format_id in ('live_dash', 'live_playback_dash'):
751 self.write_debug(f'Not extracting unsupported format "{format_id}"')
752 else:
753 formats.append({
754 'url': url,
755 'ext': 'mp4',
756 'format_id': format_id,
757 **parse_resolution(self._RESOLUTIONS.get(format_id)),
758 })
759 return formats
760
761 def _extract_common_meta(self, stream_info):
762 return traverse_obj(stream_info, {
763 'id': ('id', {str_or_none}),
764 'title': ('title', {str}),
765 'release_timestamp': ('startTime', {int_or_none}),
766 'thumbnail': ('previewUrl', {url_or_none}),
767 'view_count': ('count', 'views', {int_or_none}),
768 'like_count': ('count', 'likes', {int_or_none}),
769 'categories': ('category', 'title', {str}, {lambda x: [x] if x else None}),
770 'uploader': (('user', ('blog', 'owner')), 'nick', {str}),
771 'uploader_id': (('user', ('blog', 'owner')), 'id', {str_or_none}),
772 'duration': ('duration', {int_or_none}),
773 'is_live': ('isOnline', {bool}),
774 'concurrent_view_count': ('count', 'viewers', {int_or_none}),
775 }, get_all=False)
776
777
778class VKPlayIE(VKPlayBaseIE):
b15b0c1d 779 _VALID_URL = rf'{VKPlayBaseIE._BASE_URL_RE}(?P<username>[^/#?]+)/record/(?P<id>[\da-f-]+)'
8776349e 780 _TESTS = [{
781 'url': 'https://vkplay.live/zitsmann/record/f5e6e3b5-dc52-4d14-965d-0680dd2882da',
782 'info_dict': {
783 'id': 'f5e6e3b5-dc52-4d14-965d-0680dd2882da',
784 'ext': 'mp4',
785 'title': 'Atomic Heart (пробуем!) спасибо подписчику EKZO!',
786 'uploader': 'ZitsmanN',
787 'uploader_id': '13159830',
788 'release_timestamp': 1683461378,
789 'release_date': '20230507',
b15b0c1d 790 'thumbnail': r're:https://[^/]+/public_video_stream/record/f5e6e3b5-dc52-4d14-965d-0680dd2882da/preview',
8776349e 791 'duration': 10608,
792 'view_count': int,
793 'like_count': int,
794 'categories': ['Atomic Heart'],
795 },
796 'params': {'skip_download': 'm3u8'},
b15b0c1d 797 }, {
798 'url': 'https://live.vkplay.ru/lebwa/record/33a4e4ce-e3ef-49db-bb14-f006cc6fabc9/records',
799 'only_matching': True,
8776349e 800 }]
801
802 def _real_extract(self, url):
803 username, video_id = self._match_valid_url(url).groups()
804
805 record_info = traverse_obj(self._download_json(
806 f'https://api.vkplay.live/v1/blog/{username}/public_video_stream/record/{video_id}', video_id, fatal=False),
807 ('data', 'record', {dict}))
808 if not record_info:
809 record_info = self._extract_from_initial_state(url, video_id, ('record', 'currentRecord', 'data'))
810
811 return {
812 **self._extract_common_meta(record_info),
813 'id': video_id,
814 'formats': self._extract_formats(record_info, video_id),
815 }
816
817
818class VKPlayLiveIE(VKPlayBaseIE):
b15b0c1d 819 _VALID_URL = rf'{VKPlayBaseIE._BASE_URL_RE}(?P<id>[^/#?]+)/?(?:[#?]|$)'
8776349e 820 _TESTS = [{
821 'url': 'https://vkplay.live/bayda',
822 'info_dict': {
823 'id': 'f02c321e-427b-408d-b12f-ae34e53e0ea2',
824 'ext': 'mp4',
825 'title': r're:эскапизм крута .*',
826 'uploader': 'Bayda',
f4f9f6d0 827 'uploader_id': '12279401',
8776349e 828 'release_timestamp': 1687209962,
829 'release_date': '20230619',
b15b0c1d 830 'thumbnail': r're:https://[^/]+/public_video_stream/12279401/preview',
8776349e 831 'view_count': int,
832 'concurrent_view_count': int,
833 'like_count': int,
834 'categories': ['EVE Online'],
835 'live_status': 'is_live',
836 },
837 'skip': 'livestream',
838 'params': {'skip_download': True},
b15b0c1d 839 }, {
840 'url': 'https://live.vkplay.ru/lebwa',
841 'only_matching': True,
8776349e 842 }]
843
844 def _real_extract(self, url):
845 username = self._match_id(url)
846
847 stream_info = self._download_json(
848 f'https://api.vkplay.live/v1/blog/{username}/public_video_stream', username, fatal=False)
849 if not stream_info:
850 stream_info = self._extract_from_initial_state(url, username, ('stream', 'stream', 'data', 'stream'))
851
852 formats = self._extract_formats(stream_info, username)
853 if not formats and not traverse_obj(stream_info, ('isOnline', {bool})):
854 raise UserNotLive(video_id=username)
855
856 return {
857 **self._extract_common_meta(stream_info),
858 'formats': formats,
859 }