GeneratorExp also includes assignment context.

This commit is contained in:
whitequark 2015-06-13 12:58:45 +03:00
parent 4b4805265d
commit de6dff94cd
1 changed files with 7 additions and 6 deletions

View File

@ -58,12 +58,13 @@ class LocalExtractor(algorithm.Visitor):
self.in_root = True
self.generic_visit(node)
visit_ClassDef = visit_root # don't look at inner scopes
visit_FunctionDef = visit_root
visit_Lambda = visit_root
visit_DictComp = visit_root
visit_ListComp = visit_root
visit_SetComp = visit_root
visit_ClassDef = visit_root # don't look at inner scopes
visit_FunctionDef = visit_root
visit_Lambda = visit_root
visit_DictComp = visit_root
visit_ListComp = visit_root
visit_SetComp = visit_root
visit_GeneratorExp = visit_root
def _assignable(self, name):
if name not in self.typing_env and name not in self.nonlocal_: