Fix comprehension parent issues, fixes #1215

2019-08-23 14:34:16 +02:00
parent ecc574025c
commit 7573e2033a
2 changed files with 11 additions and 3 deletions
--- a/jedi/api/classes.py
+++ b/jedi/api/classes.py
@@ -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>" % (