Source code for lotus.forms.translated

from django import forms
from django.utils.translation import gettext_lazy as _




class TranslatedModelChoiceField(forms.ModelChoiceField):
    """
    Customize ModelChoiceField for model which inherit from ``Translated``
    model to display object language in brackets.
    """


    def label_from_instance(self, obj):
        return "{title} [{lang}]".format(
            title=str(obj),
            lang=obj.get_language_display(),
        )






class TranslatedModelMultipleChoiceField(forms.ModelMultipleChoiceField):
    """
    Customize ModelMultipleChoiceField for model which inherit from ``Translated``
    model to display object language in brackets.
    """


    def label_from_instance(self, obj):
        return "{title} [{lang}]".format(
            title=str(obj),
            lang=obj.get_language_display(),
        )






class TranslateToLangForm(forms.Form):
    """
    A very basic non model form just for listing available language choices.
    """
    language = forms.ChoiceField(
        label=_("Available languages"),
        choices=[],
    )

    def __init__(self, *args, **kwargs):
        available_languages = kwargs.pop("available_languages", [])
        super().__init__(*args, **kwargs)

        self.fields["language"].choices = available_languages