From 9313fb9021979540e82e0314e1c77a4d2bc426f2 Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Fri, 5 Oct 2018 19:03:32 +0200 Subject: [PATCH] Avoid an issue with dict comprehensions --- jedi/evaluate/context/iterable.py | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/jedi/evaluate/context/iterable.py b/jedi/evaluate/context/iterable.py index 273d4cb6..aa176bad 100644 --- a/jedi/evaluate/context/iterable.py +++ b/jedi/evaluate/context/iterable.py @@ -266,6 +266,9 @@ class DictComprehension(_DictMixin, ComprehensionMixin, Sequence): return values raise SimpleGetItemNotFound() + def _dict_keys(self): + return ContextSet.from_sets(keys for keys, values in self._iterate()) + def _dict_values(self): return ContextSet.from_sets(values for keys, values in self._iterate()) @@ -290,6 +293,9 @@ class DictComprehension(_DictMixin, ComprehensionMixin, Sequence): return ContextSet([FakeSequence(self.evaluator, u'list', lazy_contexts)]) + def get_mapping_item_contexts(self): + return self._dict_keys(), self._dict_values() + class GeneratorComprehension(ComprehensionMixin, GeneratorBase): pass