mirrors/yt-dlp
1
0
Fork 0
mirror of https://github.com/yt-dlp/yt-dlp.git synced 2024-11-08 01:13:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply suggestions from code review

Browse source 
Co-authored-by: Simon Sawicki <accounts@grub4k.xyz>
This commit is contained in:
GD-Slime 2023-11-09 18:52:12 +08:00 committed by GitHub
parent 73324baa2d
commit 2c0052097f
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 24 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

20
yt_dlp/extractor/bilibili.py
View file

@ -51,7 +51,7 @@ def extract_formats(self, play_info, is_dash=True):
if not is_dash: if not is_dash:
formats = [] formats = []
for qn in traverse_obj(play_info, 'videos'): for qn in traverse_obj(play_info, 'videos'):
video = traverse_obj(play_info, 'videos').get(qn) video = traverse_obj(play_info, ('videos', qn, ...))
segments = [] segments = []
file_total_size = 0 file_total_size = 0
@ -72,7 +72,7 @@ def extract_formats(self, play_info, is_dash=True):
'filesize': int_or_none(file_total_size) 'filesize': int_or_none(file_total_size)
}) })
missing_formats = format_names.keys() - set(traverse_obj(play_info, 'accept_quality')) missing_formats = format_names.keys() - set(traverse_obj(play_info, ('accept_quality', ...)))
else: else:
audios = traverse_obj(play_info, ('dash', (None, 'dolby'), 'audio', ..., {dict})) audios = traverse_obj(play_info, ('dash', (None, 'dolby'), 'audio', ..., {dict}))
@ -188,11 +188,9 @@ def _enc_wbi(self, params: dict, video_id=None):
session_data = self._download_json('https://api.bilibili.com/x/web-interface/nav', session_data = self._download_json('https://api.bilibili.com/x/web-interface/nav',
video_id, note='wbi signature...', fatal=False) video_id, note='wbi signature...', fatal=False)
key_from_url = lambda x: x[x.rfind('/') + 1:].split('.')[0] key_from_url = lambda x: x.rpartition('/')[2].partition('.')[0]
img_key = traverse_obj( lookup = traverse_obj(
session_data, ('data', 'wbi_img', 'img_url', {key_from_url})) session_data, ('data', 'wbi_img', ('img_url', 'sub_url'), {key_from_url}, ...))
sub_key = traverse_obj(
session_data, ('data', 'wbi_img', 'sub_url', {key_from_url}))
mixin_key_enc_tab = [ mixin_key_enc_tab = [
46, 47, 18, 2, 53, 8, 23, 32, 15, 50, 10, 31, 58, 3, 45, 35, 27, 43, 5, 49, 46, 47, 18, 2, 53, 8, 23, 32, 15, 50, 10, 31, 58, 3, 45, 35, 27, 43, 5, 49,
@ -200,13 +198,11 @@ def _enc_wbi(self, params: dict, video_id=None):
61, 26, 17, 0, 1, 60, 51, 30, 4, 22, 25, 54, 21, 56, 59, 6, 63, 57, 62, 11, 61, 26, 17, 0, 1, 60, 51, 30, 4, 22, 25, 54, 21, 56, 59, 6, 63, 57, 62, 11,
36, 20, 34, 44, 52 36, 20, 34, 44, 52
] ]
mixin_key = functools.reduce(lambda s, i: s + (img_key + sub_key)[i], mixin_key_enc_tab, '')[:32] mixin_key = ''.join(lookup[i] for i in mixin_key_enc_tab)[:32]
params['wts'] = round(time.time()) params['wts'] = round(time.time())
params = dict(sorted(params.items()))
params = { params = {
k: ''.join(filter(lambda char: char not in "!'()*", str(v))) k: ''.join(filter('!\'()*'.__contains__, str(v)))
for k, v for k, v in sorted(params.items())
in params.items()
} }
query = urllib.parse.urlencode(params) query = urllib.parse.urlencode(params)
params['w_rid'] = hashlib.md5((query + mixin_key).encode()).hexdigest() params['w_rid'] = hashlib.md5((query + mixin_key).encode()).hexdigest()