mirror of
https://github.com/davidhalter/jedi.git
synced 2025-12-06 14:04:26 +08:00
61 lines
2.0 KiB
Python
61 lines
2.0 KiB
Python
from jedi._compatibility import unicode
|
|
from jedi.evaluate.compiled.context import CompiledObject, CompiledName, \
|
|
CompiledObjectFilter, CompiledContextName, create_from_access_path, \
|
|
create_from_name
|
|
from jedi.evaluate.base_context import ContextWrapper
|
|
from jedi.evaluate.helpers import execute_evaluated
|
|
|
|
|
|
def builtin_from_name(evaluator, string):
|
|
builtins = evaluator.builtins_module
|
|
filter_ = next(builtins.get_filters())
|
|
name, = filter_.get(string)
|
|
context, = name.infer()
|
|
return context
|
|
|
|
|
|
class CompiledValue(ContextWrapper):
|
|
def __init__(self, instance, compiled_obj):
|
|
super(CompiledValue, self).__init__(instance)
|
|
self._compiled_obj = compiled_obj
|
|
|
|
def __getattribute__(self, name):
|
|
if name in ('get_safe_value', 'execute_operation', 'access_handle'):
|
|
return getattr(self._compiled_obj, name)
|
|
return super(CompiledValue, self).__getattribute__(name)
|
|
|
|
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)
|
|
compiled_obj = create_from_access_path(
|
|
evaluator,
|
|
evaluator.compiled_subprocess.create_simple_object(obj)
|
|
)
|
|
instance, = builtin_from_name(evaluator, compiled_obj.name.string_name).execute()
|
|
return CompiledValue(instance, compiled_obj)
|
|
|
|
|
|
def get_special_object(evaluator, identifier):
|
|
return create_from_access_path(
|
|
evaluator,
|
|
evaluator.compiled_subprocess.get_special_object(identifier)
|
|
)
|
|
|
|
|
|
def get_string_context_set(evaluator):
|
|
return execute_evaluated(builtin_from_name(evaluator, u'str'))
|
|
|
|
|
|
def load_module(evaluator, **kwargs):
|
|
access_path = evaluator.compiled_subprocess.load_module(**kwargs)
|
|
if access_path is None:
|
|
return None
|
|
return create_from_access_path(evaluator, access_path)
|