VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2025-12-07 06:24:27 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

get_related_names testing

Browse Source
This commit is contained in:
David Halter
2012-09-10 00:52:01 +02:00
parent 47262ea45e
commit da3c327af7
6 changed files with 69 additions and 10 deletions
Download Patch File Download Diff File Expand all files Collapse all files

24
functions.py
View File

@@ -278,8 +278,13 @@ def goto(source, line, column, source_path):
else:
search_name_new = search_name
scopes = _prepare_goto(source, pos, source_path, f, goto_path)
definitions = evaluate.goto(scopes, search_name_new)
context = f.get_context()
if next(context) in ('class', 'def'):
definitions = set([f.parser.user_scope])
else:
scopes = _prepare_goto(source, pos, source_path, f, goto_path)
definitions = evaluate.goto(scopes, search_name_new)
d = [Definition(d) for d in set(definitions)]
_clear_caches()
return d
@@ -299,8 +304,19 @@ def get_related_names(source, line, column, source_path):
else:
search_name_new = search_name
scopes = _prepare_goto(source, pos, source_path, f, goto_path)
definitions = evaluate.goto(scopes, search_name_new)
context = f.get_context()
if next(context) in ('class', 'def'):
if isinstance(f.parser.user_scope, parsing.Function):
e = evaluate.Function(f.parser.user_scope)
else:
e = evaluate.Class(f.parser.user_scope)
definitions = [e]
elif isinstance(f.parser.user_stmt, parsing.Param):
definitions = [f.parser.user_stmt]
else:
scopes = _prepare_goto(source, pos, source_path, f, goto_path)
definitions = evaluate.goto(scopes, search_name_new)
module = set([d.get_parent_until() for d in definitions])
module.add(f.parser.module)
names = dynamic.get_related_names(definitions, search_name, module)