forked from VimPlug/jedi
48 lines
1.5 KiB
Python
48 lines
1.5 KiB
Python
"""
|
|
TODO Some parts of this module are still not well documented.
|
|
"""
|
|
|
|
from jedi.inference.value import ModuleValue
|
|
from jedi.inference import compiled
|
|
from jedi.inference.compiled import mixed
|
|
from jedi.inference.compiled.access import create_access_path
|
|
from jedi.inference.base_value import ValueWrapper
|
|
|
|
|
|
def _create(inference_state, obj):
|
|
return compiled.create_from_access_path(
|
|
inference_state, create_access_path(inference_state, obj)
|
|
)
|
|
|
|
|
|
class NamespaceObject(object):
|
|
def __init__(self, dct):
|
|
self.__dict__ = dct
|
|
|
|
|
|
class MixedModuleValue(ValueWrapper):
|
|
type = 'mixed_module'
|
|
|
|
def __init__(self, inference_state, tree_module, namespaces, file_io, code_lines):
|
|
module_value = ModuleValue(
|
|
inference_state, tree_module,
|
|
file_io=file_io,
|
|
string_names=('__main__',),
|
|
code_lines=code_lines
|
|
)
|
|
super(MixedModuleValue, self).__init__(module_value)
|
|
self._namespace_objects = [NamespaceObject(n) for n in namespaces]
|
|
|
|
def get_filters(self, *args, **kwargs):
|
|
for filter in self._wrapped_value.get_filters(*args, **kwargs):
|
|
yield filter
|
|
|
|
for namespace_obj in self._namespace_objects:
|
|
compiled_object = _create(self.inference_state, namespace_obj)
|
|
mixed_object = mixed.MixedObject(
|
|
compiled_object=compiled_object,
|
|
tree_value=self._wrapped_value
|
|
)
|
|
for filter in mixed_object.get_filters(*args, **kwargs):
|
|
yield filter
|