Get rid of get_parent_until.

2017-02-03 09:59:32 +01:00
parent b3cb7b5490
commit 445bf6c419
9 changed files with 15 additions and 30 deletions
--- a/jedi/evaluate/imports.py
+++ b/jedi/evaluate/imports.py
@@ -36,7 +36,7 @@ from jedi.evaluate.filters import AbstractNameDefinition
@memoize_default(default=set())
 def infer_import(context, tree_name, is_goto=False):
    module_context = context.get_root_context()
-    import_node = tree_name.get_parent_until(tree.Import)
+    import_node = tree.search_ancestor(tree_name, ('import_name', 'import_from'))
    import_path = import_node.path_for_name(tree_name)
    from_import_name = None
    evaluator = context.evaluator