mirrors/yt-dlp-deezer
1
0
Fork 0
mirror of https://github.com/LucBerge/yt-dlp.git synced 2025-03-17 19:57:52 +03:00
Code Issues Projects Releases Packages Wiki Activity

[jsinterp] Handle default in switch better

Browse source
This commit is contained in:
pukkandan 2021-11-03 16:25:48 +05:30
parent c588b602d3
commit a1fc7ca074
No known key found for this signature in database
GPG key ID: 0F00D95A001F4698
3 changed files with 33 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

22
yt_dlp/jsinterp.py
View file

@ -228,21 +228,25 @@ class JSInterpreter(object):
switch_val, remaining = self._seperate_at_paren(expr[m.end() - 1:], ')')
switch_val = self.interpret_expression(switch_val, local_vars, allow_recursion)
body, expr = self._seperate_at_paren(remaining, '}')
body, default = body.split('default:') if 'default:' in body else (body, None)
items = body.split('case ')[1:]
if default:
items.append(f'default:{default}')
matched = False
for item in items:
case, stmt = [i.strip() for i in self._seperate(item, ':', 1)]
matched = matched or case == 'default' or switch_val == self.interpret_expression(case, local_vars, allow_recursion)
if matched:
items = body.replace('default:', 'case default:').split('case ')[1:]
for default in (False, True):
matched = False
for item in items:
case, stmt = [i.strip() for i in self._seperate(item, ':', 1)]
if default:
matched = matched or case == 'default'
elif not matched:
matched = case != 'default' and switch_val == self.interpret_expression(case, local_vars, allow_recursion)
if not matched:
continue
try:
ret, should_abort = self.interpret_statement(stmt, local_vars, allow_recursion - 1)
if should_abort:
return ret
except JS_Break:
break
if matched:
break
return self.interpret_statement(expr, local_vars, allow_recursion - 1)[0]
# Comma seperated statements