Fix generator issues that were caused by the small refactoring

This commit is contained in:
Dave Halter
2019-05-28 23:27:25 +02:00
parent 3ef99863ee
commit 4b3262622b
3 changed files with 12 additions and 4 deletions
+4 -2
View File
@@ -146,8 +146,10 @@ class AbstractInstanceContext(Context):
def py__getitem__(self, index_context_set, contextualized_node):
names = self.get_function_slot_names(u'__getitem__')
if not names:
debug.warning('Found no __getitem__ on %s', self)
return NO_CONTEXTS
return super(AbstractInstanceContext, self).py__getitem__(
index_context_set,
contextualized_node,
)
args = ValuesArguments([index_context_set])
return ContextSet.from_sets(name.infer().execute(args) for name in names)
+6
View File
@@ -55,6 +55,12 @@ class GeneratorBase(LazyAttributeOverwrite, IterableMixin):
.execute_annotation()
return generator
def is_instance(self):
return False
def py__bool__(self):
return True
@publish_method('__iter__')
def py__iter__(self, contextualized_node=None):
return ContextSet([self])