from django.apps import apps
from django.core.exceptions import FieldDoesNotExist, ImproperlyConfigured
from django.db.models import Manager

from .conf import get_available_languages, get_default_language
from .fields import TranslationField, translated_field_factory
from .manager import MultilingualManager, transform_translatable_fields
from .utils import get_model_field


def get_i18n_field(Model):
    """
    Return the i18n field if the model has it, else None.
    """
    try:
        i18n_field = Model._meta.get_field("i18n")
    except FieldDoesNotExist:
        return

    if not isinstance(i18n_field, TranslationField):
        return

    return i18n_field


def get_translated_models(app_name):
    """
    Return models having a i18n = TranslationField() for given app_name.
    """
    app = apps.get_app_config(app_name)
    for model in app.get_models():
        i18n_field = get_i18n_field(model)
        if i18n_field is not None:
            yield model


def get_i18n_field_param(Model, i18n_field, param_name):
    """
    Return i18n_param from Model.i18n_field_params dict if exists
    or get param from i18n_field
    """
    if hasattr(Model, "i18n_field_params") and param_name in Model.i18n_field_params:
        return Model.i18n_field_params[param_name]
    return getattr(i18n_field, param_name)


def translate_model(Model):
    i18n_field = get_i18n_field(Model)

    if i18n_field is None:
        return

    if not i18n_field.virtual_fields:
        # This mode is required for the migration process:
        # It needs to have a stage where we do have the TranslationField,
        # but not the virtual fields (which would collide with the
        # django-modeltranslation `<field>_<lang>`-fields), to be able to
        # copy the values from the `<field>_<lang>`-fields into `i18n.<field>_<lang>`.
        return

    validate(Model)

    add_manager(Model)
    fields_to_translate = get_i18n_field_param(Model, i18n_field, "fields")
    required_languages = get_i18n_field_param(Model, i18n_field, "required_languages")
    add_virtual_fields(Model, fields_to_translate, required_languages)
    patch_constructor(Model)

    translate_meta_ordering(Model)


def check_languages(languages, model):
    for lang in languages:
        if lang not in get_available_languages():
            raise ImproperlyConfigured(
                'Language "{}" is in required_languages on Model "{}" but '
                "not in settings.MODELTRANS_AVAILABLE_LANGUAGES.".format(lang, model.__name__)
            )


def validate(Model):
    """
    Perform validation of the arguments to TranslationField
    """
    i18n_field = Model._meta.get_field("i18n")
    for field in i18n_field.fields:
        try:
            Model._meta.get_field(field)
        except FieldDoesNotExist:
            raise ImproperlyConfigured(
                'Argument "fields" to TranslationField contains an item "{}", '
                "which is not a field (missing a comma?).".format(field)
            )

    if i18n_field.fallback_language_field:
        field = get_model_field(Model, i18n_field.fallback_language_field)
        if field is None:
            raise ImproperlyConfigured(
                'Argument "fallback_language_field" to TranslationField is "{}", '
                "which is not an existing field.".format(i18n_field.fallback_language_field)
            )

    if i18n_field.required_languages:
        required_languages = i18n_field.required_languages
        allowed_types = (tuple, list, set)
        if isinstance(required_languages, allowed_types):
            check_languages(required_languages, Model)
        elif isinstance(required_languages, dict):
            for fieldname, languages in required_languages.items():
                if fieldname not in i18n_field.fields:
                    raise ImproperlyConfigured(
                        'Fieldname "{}" in required_languages which is not '
                        'defined as translatable in "{}.i18n".'.format(field, Model.__name__)
                    )
                if not isinstance(languages, allowed_types):
                    raise ImproperlyConfigured(
                        'required_languages["{}"] must be a tuple, list or set'.format(fieldname)
                    )
                check_languages(languages, Model)
        else:
            raise ImproperlyConfigured('"required_languages" must be a tuple, list, set or dict')


def raise_if_field_exists(Model, field_name):
    if not hasattr(Model, field_name):
        return

    try:
        Model._meta.get_field(field_name)
    except FieldDoesNotExist:
        return

    raise ImproperlyConfigured(
        'Error adding translation field. Model "{}" already contains '
        'a field named "{}".'.format(Model._meta.object_name, field_name)
    )


def add_virtual_fields(Model, fields, required_languages):
    """
    Adds newly created translation fields to the given translation options.
    """
    # Add virtual/proxy fields to assign values to and get values from.
    for field_name in fields:
        # if required_languages is dict, it contains an iterable of required languages
        # for each field.
        if isinstance(required_languages, dict):
            field_required_languages = required_languages.get(field_name, ())
        else:
            field_required_languages = required_languages

        original_field = Model._meta.get_field(field_name)

        # first, add a `<original_field_name>_i18n` virtual field to get the currently
        # active translation for a field
        field = translated_field_factory(
            original_field=original_field, blank=True, null=True, editable=False  # disable in admin
        )

        raise_if_field_exists(Model, field.get_field_name())
        field.contribute_to_class(Model, field.get_field_name())

        # add a virtual field pointing to the original field with name
        # <original_field_name>_<LANGUAGE_CODE>
        field = translated_field_factory(
            original_field=original_field,
            language=get_default_language(),
            blank=True,
            null=True,
            editable=False,
        )
        raise_if_field_exists(Model, field.get_field_name())
        field.contribute_to_class(Model, field.get_field_name())

        # now, for each language, add a virtual field to get the tranlation for
        # that specific langauge
        # <original_field_name>_<language>
        for language in get_available_languages(include_default=False):
            blank_allowed = language not in field_required_languages
            field = translated_field_factory(
                original_field=original_field,
                language=language,
                blank=blank_allowed,
                null=blank_allowed,
            )
            raise_if_field_exists(Model, field.get_field_name())
            field.contribute_to_class(Model, field.get_field_name())


def has_custom_queryset(manager):
    """
    Check whether manager (or its parents) has declared some custom get_queryset method.
    """
    return getattr(manager, "get_queryset", None) != getattr(Manager, "get_queryset", None)


def add_manager(model):
    """
    Monkey patches the original model to use MultilingualManager instead of
    default managers (not only ``objects``, but also every manager defined and inherited).

    Custom managers are merged with MultilingualManager.
    """
    if model._meta.abstract:
        return

    def patch_manager_class(manager):
        if isinstance(manager, MultilingualManager):
            return
        if manager.__class__ is Manager:
            manager.__class__ = MultilingualManager
        else:

            class NewMultilingualManager(MultilingualManager, manager.__class__):
                use_for_related_fields = getattr(
                    manager.__class__, "use_for_related_fields", not has_custom_queryset(manager)
                )
                _old_module = manager.__module__
                _old_class = manager.__class__.__name__

                def deconstruct(self):
                    return (
                        False,  # as_manager
                        "%s.%s" % (self._old_module, self._old_class),  # manager_class
                        None,  # qs_class
                        self._constructor_args[0],  # args
                        self._constructor_args[1],  # kwargs
                    )

            manager.__class__ = NewMultilingualManager

    # Make all managers local for this model to fix patching parent model managers
    model._meta.local_managers = model._meta.managers

    managers = model._meta.local_managers
    for current_manager in managers:
        prev_class = current_manager.__class__
        patch_manager_class(current_manager)
        if model._default_manager.__class__ is prev_class:
            # Normally model._default_manager is a reference to one of model's managers
            # (and would be patched by the way).
            # However, in some rare situations (mostly proxy models)
            # model._default_manager is not the same instance as one of managers, but it
            # share the same class.
            model._default_manager.__class__ = current_manager.__class__
    patch_manager_class(model._base_manager)
    if hasattr(model._meta, "_expire_cache"):
        model._meta._expire_cache()


def patch_constructor(model):
    """
    Monkey patches the original model to rewrite fields names in __init__
    """
    old_init = model.__init__

    def patched_init(self, *args, **kwargs):
        old_init(self, *args, **transform_translatable_fields(self.__class__, kwargs))

    model.__init__ = patched_init


def translate_meta_ordering(Model):
    """
    If a model has ``Meta.ordering`` defined, we check if
    one of it's fields is a translated field. If that's the case,
    add the expression to get the value from the i18n-field.
    """
    if len(Model._meta.ordering) == 0:
        return
    queryset = Model.objects.get_queryset()

    Model._meta.ordering = queryset._rewrite_ordering(Model._meta.ordering)

from django.apps import apps
from django.core.exceptions import FieldDoesNotExist, ImproperlyConfigured
from django.db.models import Manager

from .conf import get_available_languages, get_default_language
from .fields import TranslationField, translated_field_factory
from .manager import MultilingualManager, transform_translatable_fields
from .utils import get_model_field


def get_i18n_field(Model):
    """
    Return the i18n field if the model has it, else None.
    """
    try:
        i18n_field = Model._meta.get_field("i18n")
    except FieldDoesNotExist:
        return

    if not isinstance(i18n_field, TranslationField):
        return

    return i18n_field


def get_translated_models(app_name):
    """
    Return models having a i18n = TranslationField() for given app_name.
    """
    app = apps.get_app_config(app_name)
    for model in app.get_models():
        i18n_field = get_i18n_field(model)
        if i18n_field is not None:
            yield model


def get_i18n_field_param(Model, i18n_field, param_name):
    """
    Return i18n_param from Model.i18n_field_params dict if exists
    or get param from i18n_field
    """
    if hasattr(Model, "i18n_field_params") and param_name in Model.i18n_field_params:
        return Model.i18n_field_params[param_name]
    return getattr(i18n_field, param_name)


def translate_model(Model):
    i18n_field = get_i18n_field(Model)

    if i18n_field is None:
        return

    if not i18n_field.virtual_fields:
        # This mode is required for the migration process:
        # It needs to have a stage where we do have the TranslationField,
        # but not the virtual fields (which would collide with the
        # django-modeltranslation `<field>_<lang>`-fields), to be able to
        # copy the values from the `<field>_<lang>`-fields into `i18n.<field>_<lang>`.
        return

    validate(Model)

    add_manager(Model)
    fields_to_translate = get_i18n_field_param(Model, i18n_field, "fields")
    required_languages = get_i18n_field_param(Model, i18n_field, "required_languages")
    add_virtual_fields(Model, fields_to_translate, required_languages)
    patch_constructor(Model)

    translate_meta_ordering(Model)


def check_languages(languages, model):
    for lang in languages:
        if lang not in get_available_languages():
            raise ImproperlyConfigured(
                'Language "{}" is in required_languages on Model "{}" but '
                "not in settings.MODELTRANS_AVAILABLE_LANGUAGES.".format(lang, model.__name__)
            )


def validate(Model):
    """
    Perform validation of the arguments to TranslationField
    """
    i18n_field = Model._meta.get_field("i18n")
    for field in i18n_field.fields:
        try:
            Model._meta.get_field(field)
        except FieldDoesNotExist:
            raise ImproperlyConfigured(
                'Argument "fields" to TranslationField contains an item "{}", '
                "which is not a field (missing a comma?).".format(field)
            )

    if i18n_field.fallback_language_field:
        field = get_model_field(Model, i18n_field.fallback_language_field)
        if field is None:
            raise ImproperlyConfigured(
                'Argument "fallback_language_field" to TranslationField is "{}", '
                "which is not an existing field.".format(i18n_field.fallback_language_field)
            )

    if i18n_field.required_languages:
        required_languages = i18n_field.required_languages
        allowed_types = (tuple, list, set)
        if isinstance(required_languages, allowed_types):
            check_languages(required_languages, Model)
        elif isinstance(required_languages, dict):
            for fieldname, languages in required_languages.items():
                if fieldname not in i18n_field.fields:
                    raise ImproperlyConfigured(
                        'Fieldname "{}" in required_languages which is not '
                        'defined as translatable in "{}.i18n".'.format(field, Model.__name__)
                    )
                if not isinstance(languages, allowed_types):
                    raise ImproperlyConfigured(
                        'required_languages["{}"] must be a tuple, list or set'.format(fieldname)
                    )
                check_languages(languages, Model)
        else:
            raise ImproperlyConfigured('"required_languages" must be a tuple, list, set or dict')


def raise_if_field_exists(Model, field_name):
    if not hasattr(Model, field_name):
        return

    try:
        Model._meta.get_field(field_name)
    except FieldDoesNotExist:
        return

    raise ImproperlyConfigured(
        'Error adding translation field. Model "{}" already contains '
        'a field named "{}".'.format(Model._meta.object_name, field_name)
    )


def add_virtual_fields(Model, fields, required_languages):
    """
    Adds newly created translation fields to the given translation options.
    """
    # Add virtual/proxy fields to assign values to and get values from.
    for field_name in fields:
        # if required_languages is dict, it contains an iterable of required languages
        # for each field.
        if isinstance(required_languages, dict):
            field_required_languages = required_languages.get(field_name, ())
        else:
            field_required_languages = required_languages

        original_field = Model._meta.get_field(field_name)

        # first, add a `<original_field_name>_i18n` virtual field to get the currently
        # active translation for a field
        field = translated_field_factory(
            original_field=original_field, blank=True, null=True, editable=False  # disable in admin
        )

        raise_if_field_exists(Model, field.get_field_name())
        field.contribute_to_class(Model, field.get_field_name())

        # add a virtual field pointing to the original field with name
        # <original_field_name>_<LANGUAGE_CODE>
        field = translated_field_factory(
            original_field=original_field,
            language=get_default_language(),
            blank=True,
            null=True,
            editable=False,
        )
        raise_if_field_exists(Model, field.get_field_name())
        field.contribute_to_class(Model, field.get_field_name())

        # now, for each language, add a virtual field to get the tranlation for
        # that specific langauge
        # <original_field_name>_<language>
        for language in get_available_languages(include_default=False):
            blank_allowed = language not in field_required_languages
            field = translated_field_factory(
                original_field=original_field,
                language=language,
                blank=blank_allowed,
                null=blank_allowed,
            )
            raise_if_field_exists(Model, field.get_field_name())
            field.contribute_to_class(Model, field.get_field_name())


def has_custom_queryset(manager):
    """
    Check whether manager (or its parents) has declared some custom get_queryset method.
    """
    return getattr(manager, "get_queryset", None) != getattr(Manager, "get_queryset", None)


def add_manager(model):
    """
    Monkey patches the original model to use MultilingualManager instead of
    default managers (not only ``objects``, but also every manager defined and inherited).

    Custom managers are merged with MultilingualManager.
    """
    if model._meta.abstract:
        return

    def patch_manager_class(manager):
        if isinstance(manager, MultilingualManager):
            return
        if manager.__class__ is Manager:
            manager.__class__ = MultilingualManager
        else:

            class NewMultilingualManager(MultilingualManager, manager.__class__):
                use_for_related_fields = getattr(
                    manager.__class__, "use_for_related_fields", not has_custom_queryset(manager)
                )
                _old_module = manager.__module__
                _old_class = manager.__class__.__name__

                def deconstruct(self):
                    return (
                        False,  # as_manager
                        "%s.%s" % (self._old_module, self._old_class),  # manager_class
                        None,  # qs_class
                        self._constructor_args[0],  # args
                        self._constructor_args[1],  # kwargs
                    )

            manager.__class__ = NewMultilingualManager

    # Make all managers local for this model to fix patching parent model managers
    model._meta.local_managers = model._meta.managers

    managers = model._meta.local_managers
    for current_manager in managers:
        prev_class = current_manager.__class__
        patch_manager_class(current_manager)
        if model._default_manager.__class__ is prev_class:
            # Normally model._default_manager is a reference to one of model's managers
            # (and would be patched by the way).
            # However, in some rare situations (mostly proxy models)
            # model._default_manager is not the same instance as one of managers, but it
            # share the same class.
            model._default_manager.__class__ = current_manager.__class__
    patch_manager_class(model._base_manager)
    if hasattr(model._meta, "_expire_cache"):
        model._meta._expire_cache()


def patch_constructor(model):
    """
    Monkey patches the original model to rewrite fields names in __init__
    """
    old_init = model.__init__

    def patched_init(self, *args, **kwargs):
        old_init(self, *args, **transform_translatable_fields(self.__class__, kwargs))

    model.__init__ = patched_init


def translate_meta_ordering(Model):
    """
    If a model has ``Meta.ordering`` defined, we check if
    one of it's fields is a translated field. If that's the case,
    add the expression to get the value from the i18n-field.
    """
    if len(Model._meta.ordering) == 0:
        return
    queryset = Model.objects.get_queryset()

    Model._meta.ordering = queryset._rewrite_ordering(Model._meta.ordering)

Return the i18n field if the model has it, else None. Return models having a i18n = TranslationField() for given app_name. Return i18n_param from Model.i18n_field_params dict if exists     or get param from i18n_field # This mode is required for the migration process: # It needs to have a stage where we do have the TranslationField, # but not the virtual fields (which would collide with the # django-modeltranslation `<field>_<lang>`-fields), to be able to # copy the values from the `<field>_<lang>`-fields into `i18n.<field>_<lang>`. Perform validation of the arguments to TranslationField Adds newly created translation fields to the given translation options. # Add virtual/proxy fields to assign values to and get values from. # if required_languages is dict, it contains an iterable of required languages # for each field. # first, add a `<original_field_name>_i18n` virtual field to get the currently # active translation for a field # disable in admin # add a virtual field pointing to the original field with name # <original_field_name>_<LANGUAGE_CODE> # now, for each language, add a virtual field to get the tranlation for # that specific langauge # <original_field_name>_<language> Check whether manager (or its parents) has declared some custom get_queryset method. Monkey patches the original model to use MultilingualManager instead of     default managers (not only ``objects``, but also every manager defined and inherited).      Custom managers are merged with MultilingualManager. # as_manager # manager_class # qs_class # args # kwargs # Make all managers local for this model to fix patching parent model managers # Normally model._default_manager is a reference to one of model's managers # (and would be patched by the way). # However, in some rare situations (mostly proxy models) # model._default_manager is not the same instance as one of managers, but it # share the same class. Monkey patches the original model to rewrite fields names in __init__ If a model has ``Meta.ordering`` defined, we check if     one of it's fields is a translated field. If that's the case,     add the expression to get the value from the i18n-field.