""" 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 import compiled 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 get_filters(self, *args, **kwargs): for filter in self._module_context.get_filters(*args, **kwargs): yield filter for namespace_obj in self._namespace_objects: compiled_object = compiled.create(self.evaluator, namespace_obj) mixed_object = mixed.MixedObject( self.evaluator, parent_context=self, compiled_object=compiled_object, tree_name=self.tree_node.name ) for filter in mixed_object.get_filters(*args, **kwargs): yield filter def __getattr__(self, name): return getattr(self._module_context, name)