]> git.bitcoin.ninja Git - youtube-dl/commitdiff
[jsinterp] Avoid double key lookup for setting new key
authorLucas Moura <lucas.moura128@gmail.com>
Sat, 18 Jun 2016 20:01:47 +0000 (17:01 -0300)
committerSergey M․ <dstftw@gmail.com>
Sat, 18 Jun 2016 20:29:45 +0000 (03:29 +0700)
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.

youtube_dl/jsinterp.py

index a7440c58242079ea1c6874e1bed0abe756fdc814..4a5a0dbc3fca59e06a268e51c3a0161ede74db70 100644 (file)
@@ -131,9 +131,8 @@ class JSInterpreter(object):
             if variable in local_vars:
                 obj = local_vars[variable]
             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:
                 # Member access
@@ -204,8 +203,7 @@ class JSInterpreter(object):
             argvals = tuple([
                 int(v) if v.isdigit() else local_vars[v]
                 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)
 
         raise ExtractorError('Unsupported JS expression %r' % expr)