Fix generators.

This commit is contained in:
Dave Halter
2016-10-16 04:17:11 +02:00
parent 4ca3556c3b
commit 0a0cb2a722
2 changed files with 9 additions and 5 deletions

View File

@@ -31,6 +31,7 @@ from jedi.evaluate.cache import CachedMetaClass, memoize_default
from jedi.evaluate import analysis
from jedi.evaluate import pep0484
from jedi import common
from jedi.evaluate.filters import DictFilter
class IterableWrapper(tree.Base):
@@ -56,10 +57,10 @@ class IterableWrapper(tree.Base):
try:
method = builtin_methods[name_str, self.type]
except KeyError:
dct[name_str] = [name]
dct[name_str] = name
else:
parent = BuiltinMethod(self, method, name.parent)
dct[name_str] = [helpers.FakeName(name_str, parent, is_definition=True)]
dct[name_str] = helpers.FakeName(name_str, parent, is_definition=True)
return dct
@@ -111,6 +112,10 @@ class GeneratorMixin(object):
gen_obj = compiled.get_special_object(self._evaluator, 'GENERATOR_OBJECT')
yield self._get_names_dict(gen_obj.names_dict)
def get_filters(self, search_global, until_position=None, origin_scope=None):
gen_obj = compiled.get_special_object(self._evaluator, 'GENERATOR_OBJECT')
yield DictFilter(self._get_names_dict(gen_obj.names_dict))
def py__bool__(self):
return True