Make separate methods for goto_import and infer_import

2019-08-21 23:58:26 +02:00
parent 2fb04db0ab
commit bf4d42798b
2 changed files with 11 additions and 2 deletions
@@ -57,7 +57,16 @@ class ModuleCache(object):
 # This memoization is needed, because otherwise we will infinitely loop on
 # certain imports.
@inference_state_method_cache(default=NO_VALUES)
-def infer_import(context, tree_name, is_goto=False):
+def infer_import(context, tree_name):
+    return _infer_import(context, tree_name, is_goto=False)
+
+
+@inference_state_method_cache(default=[])
+def goto_import(context, tree_name):
+    return _infer_import(context, tree_name, is_goto=True)
+
+
+def _infer_import(context, tree_name, is_goto=False):
    module_context = context.get_root_context()
    import_node = search_ancestor(tree_name, 'import_name', 'import_from')
    import_path = import_node.get_path_for_name(tree_name)