forked from VimPlug/jedi
48 lines
1.6 KiB
Python
48 lines
1.6 KiB
Python
"""
|
|
TODO Some parts of this module are still not well documented.
|
|
"""
|
|
import copy
|
|
|
|
from jedi.cache import underscore_memoization
|
|
from jedi.evaluate import helpers
|
|
from jedi.evaluate.representation import ModuleContext
|
|
from jedi.evaluate.compiled import mixed
|
|
from jedi.evaluate.context import Context
|
|
|
|
|
|
class MixedModuleContext(Context):
|
|
resets_positions = True
|
|
type = 'mixed_module'
|
|
|
|
def __init__(self, evaluator, tree_module, namespaces):
|
|
self.evaluator = evaluator
|
|
self._namespaces = namespaces
|
|
|
|
self._namespace_objects = [type('jedi_namespace', (), n) for n in namespaces]
|
|
self._module_context = ModuleContext(evaluator, tree_module)
|
|
self.tree_node = tree_module
|
|
|
|
def get_node(self):
|
|
return self.tree_node
|
|
|
|
def names_dicts(self, search_global):
|
|
for names_dict in self._module_context.names_dicts(search_global):
|
|
yield names_dict
|
|
|
|
for namespace_obj in self._namespace_objects:
|
|
m = mixed.MixedObject(self.evaluator, namespace_obj, self.tree_node.name)
|
|
for names_dict in m.names_dicts(False):
|
|
yield names_dict
|
|
|
|
def get_filters(self, *args, **kwargs):
|
|
for filter in self._module_context.get_filters(*args, **kwargs):
|
|
yield filter
|
|
|
|
for namespace_obj in self._namespace_objects:
|
|
m = mixed.MixedObject(self.evaluator, namespace_obj, self.tree_node.name)
|
|
for filter in m.get_filters(*args, **kwargs):
|
|
yield filter
|
|
|
|
def __getattr__(self, name):
|
|
return getattr(self._module_context, name)
|