forked from VimPlug/jedi
04b7c99753
This definitely reduces debug output and it might be slightly faster, because some values are never asked for
65 lines
2.3 KiB
Python
65 lines
2.3 KiB
Python
from jedi._compatibility import unicode
|
|
from jedi.evaluate.compiled.context import CompiledObject, CompiledName, \
|
|
CompiledObjectFilter, CompiledContextName, create_from_access_path
|
|
from jedi.evaluate.base_context import ContextWrapper, LazyContextWrapper
|
|
|
|
|
|
def builtin_from_name(evaluator, string):
|
|
typing_builtins_module = evaluator.builtins_module
|
|
if string in ('None', 'True', 'False'):
|
|
builtins, = typing_builtins_module.non_stub_context_set
|
|
filter_ = next(builtins.get_filters())
|
|
else:
|
|
filter_ = next(typing_builtins_module.get_filters())
|
|
name, = filter_.get(string)
|
|
context, = name.infer()
|
|
return context
|
|
|
|
|
|
class CompiledValue(LazyContextWrapper):
|
|
def __init__(self, compiled_obj):
|
|
self.evaluator = compiled_obj.evaluator
|
|
self._compiled_obj = compiled_obj
|
|
|
|
def __getattribute__(self, name):
|
|
if name in ('get_safe_value', 'execute_operation', 'access_handle',
|
|
'negate', 'py__bool__', 'is_compiled'):
|
|
return getattr(self._compiled_obj, name)
|
|
return super(CompiledValue, self).__getattribute__(name)
|
|
|
|
def _get_wrapped_context(self):
|
|
instance, = builtin_from_name(
|
|
self.evaluator, self._compiled_obj.name.string_name).execute_evaluated()
|
|
return instance
|
|
|
|
def __repr__(self):
|
|
return '<%s: %s>' % (self.__class__.__name__, self._compiled_obj)
|
|
|
|
|
|
def create_simple_object(evaluator, obj):
|
|
"""
|
|
Only allows creations of objects that are easily picklable across Python
|
|
versions.
|
|
"""
|
|
assert type(obj) in (int, float, str, bytes, unicode, slice, complex, bool), obj
|
|
compiled_obj = create_from_access_path(
|
|
evaluator,
|
|
evaluator.compiled_subprocess.create_simple_object(obj)
|
|
)
|
|
return CompiledValue(compiled_obj)
|
|
|
|
|
|
def get_string_context_set(evaluator):
|
|
return builtin_from_name(evaluator, u'str').execute_evaluated()
|
|
|
|
|
|
def load_module(evaluator, dotted_name, **kwargs):
|
|
# Temporary, some tensorflow builtins cannot be loaded, so it's tried again
|
|
# and again and it's really slow.
|
|
if dotted_name.startswith('tensorflow.'):
|
|
return None
|
|
access_path = evaluator.compiled_subprocess.load_module(dotted_name=dotted_name, **kwargs)
|
|
if access_path is None:
|
|
return None
|
|
return create_from_access_path(evaluator, access_path)
|