mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 22:14:27 +08:00
92 lines
2.6 KiB
Python
92 lines
2.6 KiB
Python
import parsing
|
|
|
|
def get_names_for_scope(scope):
|
|
""" Get all completions possible for the current scope. """
|
|
comp = []
|
|
start_scope = scope
|
|
while scope:
|
|
# class variables/functions are only availabe
|
|
if not isinstance(scope, parsing.Class) or scope == start_scope:
|
|
comp += scope.get_set_vars()
|
|
scope = scope.parent
|
|
return comp
|
|
|
|
|
|
def follow_path(scope, path):
|
|
"""
|
|
Follow a path of python names.
|
|
recursive!
|
|
:returns: list(Scope)
|
|
"""
|
|
comp = get_names_for_scope(scope)
|
|
print path, comp
|
|
|
|
path = list(path)
|
|
name = path.pop(0)
|
|
scopes = []
|
|
|
|
# make the full comparison, because the names have to match exactly
|
|
comp = [c for c in comp if [name] == list(c.names)]
|
|
# TODO differentiate between the different comp input (some are overridden)
|
|
for c in comp:
|
|
p_class = c.parent.__class__
|
|
if p_class == parsing.Class or p_class == parsing.Scope:
|
|
scopes.append(c.parent)
|
|
#elif p_class == parsing.Function:
|
|
elif p_class == parsing.Statement:
|
|
print 'state', c.parent.token_list, c.parent.get_assignment_calls()
|
|
pass
|
|
else:
|
|
print 'error follow_path:', p_class, repr(c.parent)
|
|
|
|
if path:
|
|
new_scopes = []
|
|
for s in tuple(scopes):
|
|
new_scopes += follow_path(s, tuple(path))
|
|
scopes = new_scopes
|
|
return set(scopes)
|
|
|
|
def _parseassignment(self):
|
|
""" TODO remove or replace, at the moment not used """
|
|
assign = ''
|
|
token_type, tok, indent = self.next()
|
|
if token_type == tokenize.STRING or tok == 'str':
|
|
return '""'
|
|
elif tok == '(' or tok == 'tuple':
|
|
return '()'
|
|
elif tok == '[' or tok == 'list':
|
|
return '[]'
|
|
elif tok == '{' or tok == 'dict':
|
|
return '{}'
|
|
elif token_type == tokenize.NUMBER:
|
|
return '0'
|
|
elif tok == 'open' or tok == 'file':
|
|
return 'file'
|
|
elif tok == 'None':
|
|
return '_PyCmplNoType()'
|
|
elif tok == 'type':
|
|
return 'type(_PyCmplNoType)' # only for method resolution
|
|
else:
|
|
assign += tok
|
|
level = 0
|
|
while True:
|
|
token_type, tok, indent = self.next()
|
|
if tok in ('(', '{', '['):
|
|
level += 1
|
|
elif tok in (']', '}', ')'):
|
|
level -= 1
|
|
if level == 0:
|
|
break
|
|
elif level == 0:
|
|
if tok in (';', '\n'):
|
|
break
|
|
assign += tok
|
|
return "%s" % assign
|
|
|
|
def dbg(*args):
|
|
if debug_function:
|
|
debug_function(*args)
|
|
|
|
|
|
debug_function = None
|