github/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-30 11:51:30 -05:00
Code Issues Projects Releases Packages Wiki Activity

[viki] Fix extraction

Browse source 
Closes #381
Code from: 59e583f7e8
This commit is contained in:
pukkandan 2021-06-07 12:19:30 +05:30
parent 87ea7dfc04
commit 89ee4cf8ae
No known key found for this signature in database
GPG key ID: 0F00D95A001F4698
1 changed files with 38 additions and 20 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

58
yt_dlp/extractor/viki.py
View file

@ -142,6 +142,21 @@ class VikiIE(VikiBaseIE):
IE_NAME = 'viki' IE_NAME = 'viki'
_VALID_URL = r'%s(?:videos|player)/(?P<id>[0-9]+v)' % VikiBaseIE._VALID_URL_BASE _VALID_URL = r'%s(?:videos|player)/(?P<id>[0-9]+v)' % VikiBaseIE._VALID_URL_BASE
_TESTS = [{ _TESTS = [{
'url': 'https://www.viki.com/videos/1175236v-choosing-spouse-by-lottery-episode-1',
'info_dict': {
'id': '1175236v',
'ext': 'mp4',
'title': 'Choosing Spouse by Lottery - Episode 1',
'timestamp': 1606463239,
'age_limit': 13,
'uploader': 'FCC',
'upload_date': '20201127',
},
'params': {
'format': 'bestvideo',
},
'expected_warnings': ['Unknown MIME type image/jpeg in DASH manifest'],
}, {
'url': 'http://www.viki.com/videos/1023585v-heirs-episode-14', 'url': 'http://www.viki.com/videos/1023585v-heirs-episode-14',
'info_dict': { 'info_dict': {
'id': '1023585v', 'id': '1023585v',
@ -255,8 +270,14 @@ class VikiIE(VikiBaseIE):
def _real_extract(self, url): def _real_extract(self, url):
video_id = self._match_id(url) video_id = self._match_id(url)
video = self._call_api( resp = self._download_json(
'videos/%s.json' % video_id, video_id, 'Downloading video JSON') 'https://www.viki.com/api/videos/' + video_id,
video_id, 'Downloading video JSON', headers={
'x-client-user-agent': std_headers['User-Agent'],
'x-viki-app-ver': '3.0.0',
})
video = resp['video']
self._check_errors(video) self._check_errors(video)
title = self.dict_selection(video.get('titles', {}), 'en', allow_fallback=False) title = self.dict_selection(video.get('titles', {}), 'en', allow_fallback=False)
@ -310,8 +331,6 @@ def add_format(format_id, format_dict, protocol='http'):
format_url = format_dict.get('url') format_url = format_dict.get('url')
if not format_url: if not format_url:
return return
format_drms = format_dict.get('drms')
format_stream_id = format_dict.get('id')
qs = compat_parse_qs(compat_urllib_parse_urlparse(format_url).query) qs = compat_parse_qs(compat_urllib_parse_urlparse(format_url).query)
stream = qs.get('stream', [None])[0] stream = qs.get('stream', [None])[0]
if stream: if stream:
@ -345,8 +364,6 @@ def add_format(format_id, format_dict, protocol='http'):
'play_path': mobj.group('playpath'), 'play_path': mobj.group('playpath'),
'app': mobj.group('app'), 'app': mobj.group('app'),
'page_url': url, 'page_url': url,
'drms': format_drms,
'stream_id': format_stream_id,
}) })
else: else:
urlh = self._request_webpage( urlh = self._request_webpage(
@ -356,25 +373,26 @@ def add_format(format_id, format_dict, protocol='http'):
'format_id': '%s-%s' % (format_id, protocol), 'format_id': '%s-%s' % (format_id, protocol),
'height': int_or_none(self._search_regex( 'height': int_or_none(self._search_regex(
r'^(\d+)[pP]$', format_id, 'height', default=None)), r'^(\d+)[pP]$', format_id, 'height', default=None)),
'drms': format_drms,
'stream_id': format_stream_id,
'filesize': int_or_none(urlh.headers.get('Content-Length')), 'filesize': int_or_none(urlh.headers.get('Content-Length')),
}) })
streams = self._call_api( for format_id, format_dict in (resp.get('streams') or {}).items():
'videos/%s/streams.json' % video_id, video_id, add_format(format_id, format_dict)
'Downloading video streams JSON') if not formats:
streams = self._call_api(
'videos/%s/streams.json' % video_id, video_id,
'Downloading video streams JSON')
if 'external' in streams: if 'external' in streams:
result.update({ result.update({
'_type': 'url_transparent', '_type': 'url_transparent',
'url': streams['external']['url'], 'url': streams['external']['url'],
}) })
return result return result
for format_id, stream_dict in streams.items(): for format_id, stream_dict in streams.items():
for protocol, format_dict in stream_dict.items(): for protocol, format_dict in stream_dict.items():
add_format(format_id, format_dict, protocol) add_format(format_id, format_dict, protocol)
self._sort_formats(formats) self._sort_formats(formats)
result['formats'] = formats result['formats'] = formats