"""
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)