VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2026-05-19 14:59:41 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

improve follow_path params

Browse Source
This commit is contained in:
Dave Halter
2013-12-28 20:56:09 +01:00
parent 75a1b6f8cb
commit 37e157d441
1 changed files with 22 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files
jedi/evaluate/__init__.py
+22 -18
View File
@@ -588,7 +588,7 @@ class Evaluator(object):
current = next(path) current = next(path)
if isinstance(current, pr.Array): if isinstance(current, pr.Array):
result = [er.Array(self, current)] types = [er.Array(self, current)]
else: else:
if isinstance(current, pr.NamePart): if isinstance(current, pr.NamePart):
# This is the first global lookup. # This is the first global lookup.
@@ -599,13 +599,17 @@ class Evaluator(object):
scopes = self.find_name(builtin.Builtin.scope, current.type_as_string()) scopes = self.find_name(builtin.Builtin.scope, current.type_as_string())
# Make instances of those number/string objects. # Make instances of those number/string objects.
scopes = [er.Instance(self, s, (current.value,)) for s in scopes] scopes = [er.Instance(self, s, (current.value,)) for s in scopes]
result = imports.strip_imports(self, scopes) types = imports.strip_imports(self, scopes)
return self.follow_path(path, result, scope, position=position) return self.follow_path(path, types, scope, position=position)
def follow_path(self, path, types, call_scope, position=None): def follow_path(self, path, types, call_scope, position=None):
""" """
In each result, `path` must be followed. Copies the path iterator. Follows a path like::
self.follow_path(iter(['Foo', 'bar']), [a_type], from_somewhere)
to follow a call like ``module.a_type.Foo.bar`` (in ``from_somewhere``).
""" """
results_new = [] results_new = []
iter_paths = itertools.tee(path, len(types)) iter_paths = itertools.tee(path, len(types))
@@ -619,46 +623,46 @@ class Evaluator(object):
return types return types
return results_new return results_new
def _follow_path(self, path, scope, call_scope, position=None): def _follow_path(self, path, type, scope, position=None):
""" """
Uses a generator and tries to complete the path, e.g.:: Uses a generator and tries to complete the path, e.g.::
foo.bar.baz foo.bar.baz
`_follow_path` is only responsible for completing `.bar.baz`, the rest is `_follow_path` is only responsible for completing `.bar.baz`, the rest
done in the `follow_call` function. is done in the `follow_call` function.
""" """
# current is either an Array or a Scope. # current is either an Array or a Scope.
try: try:
current = next(path) current = next(path)
except StopIteration: except StopIteration:
return None return None
debug.dbg('_follow_path: %s in scope %s' % (current, scope)) debug.dbg('_follow_path: %s in scope %s' % (current, type))
result = [] result = []
if isinstance(current, pr.Array): if isinstance(current, pr.Array):
# This must be an execution, either () or []. # This must be an execution, either () or [].
if current.type == pr.Array.LIST: if current.type == pr.Array.LIST:
if hasattr(scope, 'get_index_types'): if hasattr(type, 'get_index_types'):
result = scope.get_index_types(current) result = type.get_index_types(current)
elif current.type not in [pr.Array.DICT]: elif current.type not in [pr.Array.DICT]:
# Scope must be a class or func - make an instance or execution. # Scope must be a class or func - make an instance or execution.
debug.dbg('exe', scope) debug.dbg('exe', type)
result = self.execute(scope, current) result = self.execute(type, current)
else: else:
# Curly braces are not allowed, because they make no sense. # Curly braces are not allowed, because they make no sense.
debug.warning('strange function call with {}', current, scope) debug.warning('strange function call with {}', current, type)
else: else:
# The function must not be decorated with something else. # The function must not be decorated with something else.
if scope.isinstance(er.Function): if type.isinstance(er.Function):
scope = scope.get_magic_method_scope() type = type.get_magic_method_scope()
else: else:
# This is the typical lookup while chaining things. # This is the typical lookup while chaining things.
if filter_private_variable(scope, call_scope, current): if filter_private_variable(type, scope, current):
return [] return []
result = imports.strip_imports(self, self.find_name(scope, current, result = imports.strip_imports(self, self.find_name(type, current,
position=position)) position=position))
return self.follow_path(path, set(result), call_scope, position=position) return self.follow_path(path, set(result), scope, position=position)
def execute(self, scope, params, evaluate_generator=False): def execute(self, scope, params, evaluate_generator=False):
return er.Execution(self, scope, params).get_return_types(evaluate_generator) return er.Execution(self, scope, params).get_return_types(evaluate_generator)