Better 3.2/33 store_locals handling

This commit is contained in:
rocky
2016-07-12 09:46:46 -04:00
parent 78f24f9c66
commit b99f196d18
3 changed files with 7 additions and 2 deletions

Binary file not shown.

View File

@@ -1962,11 +1962,16 @@ class SourceWalker(GenericASTTraversal, object):
code._tokens = None # save memory code._tokens = None # save memory
assert ast == 'stmts' assert ast == 'stmts'
if 3.0 <= self.version <= 3.3:
try:
if ast[0][0][0] == 'store_locals':
if self.hide_internal: del ast[0]
except:
pass
try: try:
if ast[0][0] == NAME_MODULE: if ast[0][0] == NAME_MODULE:
if self.hide_internal: del ast[0] if self.hide_internal: del ast[0]
elif ast[1][0] == NAME_MODULE:
if self.hide_internal: del ast[1]
pass pass
except: except:
pass pass