Browse Source

[jsinterp] Avoid double key lookup for setting new key

In order to add a new key to both __objects and __functions dicts on jsinterp.py, it is
necessary to first verify if a key was present and if not, create the key and
assign it to a value.

However, this can be done with a single step using dict setdefault method.
master
Lucas Moura 8 years ago
committed by Sergey M․
parent
commit
7c05097633
No known key found for this signature in database GPG Key ID: 2C393E0F18A9236D
1 changed files with 3 additions and 5 deletions
  1. 8
      youtube_dl/jsinterp.py

8
youtube_dl/jsinterp.py

@ -131,9 +131,8 @@ class JSInterpreter(object):
if variable in local_vars: if variable in local_vars:
obj = local_vars[variable] obj = local_vars[variable]
else: else:
if variable not in self._objects:
self._objects[variable] = self.extract_object(variable)
obj = self._objects[variable]
obj = self._objects.setdefault(
variable, self.extract_object(variable))
if arg_str is None: if arg_str is None:
# Member access # Member access
@ -204,8 +203,7 @@ class JSInterpreter(object):
argvals = tuple([ argvals = tuple([
int(v) if v.isdigit() else local_vars[v] int(v) if v.isdigit() else local_vars[v]
for v in m.group('args').split(',')]) for v in m.group('args').split(',')])
if fname not in self._functions:
self._functions[fname] = self.extract_function(fname)
self._functions.setdefault(fname, self.extract_function(fname))
return self._functions[fname](argvals) return self._functions[fname](argvals)
raise ExtractorError('Unsupported JS expression %r' % expr) raise ExtractorError('Unsupported JS expression %r' % expr)

Loading…
Cancel
Save