Fix comprehension parent issues, fixes #1215

This commit is contained in:
Dave Halter
2019-08-23 14:34:16 +02:00
parent ecc574025c
commit 7573e2033a
2 changed files with 11 additions and 3 deletions

View File

@@ -361,9 +361,10 @@ class BaseDefinition(object):
context = self._name.parent_context
if context is None:
return None
# TODO private access!
return Definition(self._inference_state, context._value.name)
while context.name is None:
# Happens for comprehension contexts
context = context.parent_context
return Definition(self._inference_state, context.name)
def __repr__(self):
return "<%s %sname=%r, description=%r>" % (