VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-09 23:34:45 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Make separate methods for goto_import and infer_import

Browse Source
This commit is contained in:
Dave Halter
2019-08-21 23:58:26 +02:00
parent 2fb04db0ab
commit bf4d42798b
2 changed files with 11 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
jedi/inference/__init__.py
View File

@@ -308,7 +308,7 @@ class InferenceState(object):
elif type_ == 'param':
return [ParamName(context, name)]
elif type_ in ('import_from', 'import_name'):
module_names = imports.infer_import(context, name, is_goto=True)
module_names = imports.goto_import(context, name)
return module_names
else:
return [TreeNameDefinition(context, name)]

11
jedi/inference/imports.py
View File

@@ -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)