VimPlug/jedi-fork
1
0
Fork 0
forked from VimPlug/jedi
Code Pull Requests Activity

Use SimpleParamName everywhere it's needed

Browse Source
This commit is contained in:
Dave Halter
2019-09-02 09:29:43 +02:00
parent edb17b8e7c
commit b4dc95553f
4 changed files with 16 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
jedi/inference/usages.py
View File

@@ -1,5 +1,7 @@
from parso.python.tree import search_ancestor
from jedi.inference import imports
from jedi.inference.names import TreeNameDefinition
from jedi.inference.names import TreeNameDefinition, SimpleParamName
def _resolve_names(definition_names, avoid_names=()):
@@ -27,8 +29,14 @@ def _dictionarize(names):
def _find_names(module_context, tree_name):
context = module_context.create_context(tree_name)
name = TreeNameDefinition(context, tree_name)
definition = tree_name.get_definition()
if definition and definition.type == 'param' and definition.name == tree_name:
funcdef = search_ancestor(definition, 'funcdef', 'lambdef')
func = module_context.create_value(funcdef)
name = SimpleParamName(func, tree_name)
else:
context = module_context.create_context(tree_name)
name = TreeNameDefinition(context, tree_name)
found_names = set(name.goto())
found_names.add(name)
return _dictionarize(_resolve_names(found_names))