VimPlug/jedi
1
0
Fork 1
mirror of https://github.com/davidhalter/jedi.git synced 2026-02-16 02:02:04 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

Deprecate Evaluator.wrap.

Browse Source
This commit is contained in:
Dave Halter
2016-11-01 23:38:06 +01:00
parent 2eb701d2d2
commit f57455f0ad
5 changed files with 74 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

41
jedi/api/__init__.py
View File

@@ -132,15 +132,19 @@ class Script(object):
debug.speed('init')
@cache.memoize_method
def _get_module(self):
def _get_module_node(self):
cache.invalidate_star_import_cache(self._path)
parser = FastParser(self._grammar, self._source, self.path)
save_parser(self.path, parser, pickling=False)
module = self._evaluator.wrap(parser.module, parent_context=None)
imports.add_module(self._evaluator, unicode(module.name), module)
return parser.module
@cache.memoize_method
def _get_module(self):
module = er.ModuleContext(self._evaluator, self._get_module_node())
imports.add_module(self._evaluator, unicode(module.name), module)
return module
@property
def source_path(self):
"""
@@ -183,13 +187,14 @@ class Script(object):
:rtype: list of :class:`classes.Definition`
"""
leaf = self._get_module().name_for_position(self._pos)
module_node = self._get_module_node()
leaf = module_node.name_for_position(self._pos)
if leaf is None:
leaf = self._get_module().get_leaf_for_position(self._pos)
leaf = module_node.get_leaf_for_position(self._pos)
if leaf is None:
return []
context = self._evaluator.create_context(leaf)
context = self._evaluator.create_context(self._get_module(), leaf)
definitions = helpers.evaluate_goto_definition(self._evaluator, context, leaf)
names = [s.name for s in definitions]
@@ -229,7 +234,7 @@ class Script(object):
"""
Used for goto_assignments and usages.
"""
name = self._get_module().name_for_position(self._pos)
name = self._get_module_node().name_for_position(self._pos)
if name is None:
return []
return list(self._evaluator.goto(name))
@@ -248,7 +253,8 @@ class Script(object):
temp, settings.dynamic_flow_information = \
settings.dynamic_flow_information, False
try:
user_stmt = self._get_module().get_statement_for_position(self._pos)
module_node = self._get_module_node()
user_stmt = module_node.get_statement_for_position(self._pos)
definitions = self._goto()
if not definitions and isinstance(user_stmt, tree.Import):
# For not defined imports (goto doesn't find something, we take
@@ -270,7 +276,7 @@ class Script(object):
definitions)
module = set([d.get_parent_until() for d in definitions])
module.add(self._get_module())
module.add(module_node)
names = usages.usages(self._evaluator, definitions, module)
for d in set(definitions):
@@ -297,7 +303,7 @@ class Script(object):
:rtype: list of :class:`classes.CallSignature`
"""
call_signature_details = \
helpers.get_call_signature_details(self._get_module(), self._pos)
helpers.get_call_signature_details(self._get_module_node(), self._pos)
if call_signature_details is None:
return []
@@ -320,9 +326,10 @@ class Script(object):
def _analysis(self):
self._evaluator.is_analysis = True
self._evaluator.analysis_modules = [self._get_module()]
module_node = self._get_module_node()
self._evaluator.analysis_modules = [module_node]
try:
for node in self._get_module().nodes_to_execute():
for node in module_node.nodes_to_execute():
if node.type in ('funcdef', 'classdef'):
if node.type == 'classdef':
continue
@@ -387,11 +394,9 @@ class Interpreter(Script):
super(Interpreter, self).__init__(source, **kwds)
self.namespaces = namespaces
parser_module = super(Interpreter, self)._get_module()
self._module = interpreter.MixedModule(self._evaluator, parser_module, self.namespaces)
def _get_module(self):
return self._module
def _get_module_node(self):
parser_module = super(Interpreter, self)._get_module_node()
return interpreter.MixedModule(self._evaluator, parser_module, self.namespaces)
def defined_names(source, path=None, encoding='utf-8'):
@@ -437,7 +442,7 @@ def names(source=None, path=None, encoding='utf-8', all_scopes=False,
# Set line/column to a random position, because they don't matter.
script = Script(source, line=1, column=0, path=path, encoding=encoding)
defs = [classes.Definition(script._evaluator, name_part)
for name_part in get_module_names(script._get_module(), all_scopes)]
for name_part in get_module_names(script._get_module().module_node, all_scopes)]
return sorted(filter(def_ref_filter, defs), key=lambda x: (x.line, x.column))

25
jedi/api/completion.py
View File

@@ -47,11 +47,11 @@ def filter_names(evaluator, completion_names, stack, like_name):
yield new
def get_user_scope(module, position):
def get_user_scope(module_context, position):
"""
Returns the scope in which the user resides. This includes flows.
"""
user_stmt = module.get_statement_for_position(position)
user_stmt = module_context.module_node.get_statement_for_position(position)
if user_stmt is None:
def scan(scope):
for s in scope.children:
@@ -62,20 +62,20 @@ def get_user_scope(module, position):
return scan(s)
return None
return scan(module) or module
return scan(module) or module_context
else:
return user_stmt.get_parent_scope(include_flows=True)
class Completion:
def __init__(self, evaluator, module_node, code_lines, position, call_signatures_method):
def __init__(self, evaluator, module, code_lines, position, call_signatures_method):
self._evaluator = evaluator
self._module_node = module_node
self._module = evaluator.wrap(module_node, parent_context=None)
self._module_context = module
self._module_node = module.module_node
self._code_lines = code_lines
# The first step of completions is to get the name
self._like_name = helpers.get_on_completion_name(module_node, code_lines, position)
self._like_name = helpers.get_on_completion_name(self._module_node, code_lines, position)
# The actual cursor position is not what we need to calculate
# everything. We want the start of the name we're on.
self._position = position[0], position[1] - len(self._like_name)
@@ -172,10 +172,10 @@ class Completion:
yield keywords.keyword(self._evaluator, k).name
def _global_completions(self):
scope = get_user_scope(self._module_node, self._position)
scope = get_user_scope(self._module_context, self._position)
if not scope.is_scope(): # Might be a flow (if/while/etc).
scope = scope.get_parent_scope()
scope = self._evaluator.create_context(scope)
scope = self._evaluator.create_context(self._module_context, scope)
debug.dbg('global completion scope: %s', scope)
filters = get_global_filters(
self._evaluator,
@@ -189,8 +189,9 @@ class Completion:
return completion_names
def _trailer_completions(self, atom_expr):
user_scope = get_user_scope(self._module_node, self._position)
contexts = self._evaluator.eval_element(self._evaluator.create_context(atom_expr), atom_expr)
user_scope = get_user_scope(self._module_context, self._position)
evaluation_context = self._evaluator.create_context(self._module_context, atom_expr)
contexts = self._evaluator.eval_element(evaluation_context, atom_expr)
completion_names = []
debug.dbg('trailer completion contexts: %s', contexts)
for context in contexts:
@@ -215,7 +216,7 @@ class Completion:
def _get_importer_names(self, names, level=0, only_modules=True):
names = [str(n) for n in names]
i = imports.Importer(self._evaluator, names, self._module, level)
i = imports.Importer(self._evaluator, names, self._module_context, level)
return i.completion_names(self._evaluator, only_modules=only_modules)
def _get_class_context_completions(self, is_function=True):