Source code for atom.ext.crispy_forms.forms

from __future__ import absolute_import
from crispy_forms.helper import FormHelper
from crispy_forms.layout import Reset, Submit
from django.forms import BaseFormSet
from django.forms.models import BaseInlineFormSet
from django.utils.translation import ugettext as _
from django.utils.translation import ugettext_lazy as _l


class FormsetHelper(FormHelper):
    form_tag = False
    form_method = 'post'


class TableFormSetHelper(FormsetHelper):
    template = 'bootstrap/table_inline_formset.html'


class TableFormSetMixin(object):
    def __init__(self, *args, **kwargs):
        super(InlineTableFormSet, self).__init__(*args, **kwargs)
        self.helper = TableFormSetHelper()


class InlineTableFormSet(TableFormSetMixin, BaseInlineFormSet):
    pass


class TableFormSet(TableFormSetMixin, BaseFormSet):
    pass


class BaseTableFormSet(BaseInlineFormSet):
    def __init__(self, *args, **kwargs):
        super(BaseTableFormSet, self).__init__(*args, **kwargs)
        self.helper = TableFormSetHelper()


class HelperMixin(object):

    form_helper_cls = FormHelper

    def __init__(self, *args, **kwargs):
        super(HelperMixin, self).__init__(*args, **kwargs)
        self.helper = getattr(self, 'helper', self.form_helper_cls(self))


class SingleButtonMixin(HelperMixin):
    """Dynamically add crispy button to form layout

    Example
    -------
    Usage of mixins is obvious::

        from django import forms
        from atom.ext.crispy_forms.forms import SingleButtonMixin

        class PersonModelForm(SingleButtonMixin, forms.ModelForm):

            class Meta:
                model = Person

    Extends
    -------
    HelperMixin
    """

    @property
    def action_text(self):
        """A text of added action submit button.

        In standards it detects when forms save or update objects

        Returns
        -------
        string -- A text used in button
        """
        return _('Update') if hasattr(self, 'instance') and self.instance.pk else _('Save')

    def __init__(self, *args, **kwargs):
        super(SingleButtonMixin, self).__init__(*args, **kwargs)
        self.helper.add_input(Submit('action', self.action_text, css_class="btn-primary"))


class FormHorizontalMixin(HelperMixin):
    def __init__(self, *args, **kwargs):
        super(FormHorizontalMixin, self).__init__(*args, **kwargs)
        self.helper.form_class = 'form-horizontal'
        self.helper.label_class = 'col-lg-3'
        self.helper.field_class = 'col-lg-9'