From 11fa71bac89041332110e89340fa41657d377a1f Mon Sep 17 00:00:00 2001 From: Dave Halter Date: Thu, 20 Nov 2014 14:56:05 +0100 Subject: [PATCH] Fixes for nested star imports. --- jedi/evaluate/representation.py | 8 ++++++-- 1 file changed, 6 insertions(+), 2 deletions(-) diff --git a/jedi/evaluate/representation.py b/jedi/evaluate/representation.py index f1d03521..e985a954 100644 --- a/jedi/evaluate/representation.py +++ b/jedi/evaluate/representation.py @@ -722,13 +722,17 @@ class ModuleWrapper(use_metaclass(CachedMetaClass, pr.Module, Wrapper)): if sub_modules: yield self, self._sub_modules() - @underscore_memoization + @memoize_default([]) def star_imports(self): modules = [] for i in self.base.imports: if i.is_star_import(): name = i.star_import_name() - modules += imports.ImportWrapper(self._evaluator, name).follow() + new = imports.ImportWrapper(self._evaluator, name).follow() + for module in new: + if isinstance(module, pr.Module): + modules += module.star_imports() + modules += new return modules @memoize_default()