Remove the ImportWrapper and replace it with something simpler.

2016-12-17 16:08:37 +01:00
parent 173c939956
commit 57857b6332
7 changed files with 53 additions and 74 deletions
--- a/jedi/api/usages.py
+++ b/jedi/api/usages.py
@@ -59,6 +59,10 @@ def resolve_potential_imports(evaluator, definitions):
        if isinstance(d, TreeNameDefinition):
            imp_or_stmt = d.tree_name.get_definition()
            if isinstance(imp_or_stmt, tree.Import):
-                s = imports.ImportWrapper(d.parent_context, d.tree_name)
-                new |= resolve_potential_imports(evaluator, set(s.follow(is_goto=True)))
+                new |= resolve_potential_imports(
+                    evaluator,
+                    set(imports.infer_import(
+                        d.parent_context, d.tree_name, is_goto=True
+                    ))
+                )
    return set(definitions) | new