PEP8

[auf_rh_dae.git] / project / rh / forms.py

# -*- encoding: utf-8 -*-

from django import forms
from ajax_select.fields import AutoCompleteSelectField
from project.rh.models import Dossier, Contrat, AyantDroit, Employe, \
        ResponsableImplantation


class AjaxSelect(object):

    class Media:
        css = {
            'all': (
                'jquery-autocomplete/jquery.autocomplete.css',
                'css/select.css',
            )
        }
        js = (
            'js/jquery-1.5.1.min.js',
            'jquery-autocomplete/jquery.autocomplete.js',
        )


class FormDate(object):

    def clean_date_fin(self):
        date_fin = self.cleaned_data['date_fin']
        if date_fin is None:
            return date_fin
        date_debut = self.cleaned_data['date_debut']
        if date_fin < date_debut:
            raise forms.ValidationError(
                u"La date de fin est antérieure à la date de début"
            )
        return date_fin


class DossierForm(forms.ModelForm, FormDate):

    class Model:
        model = Dossier


class ContratForm(forms.ModelForm, FormDate):

    class Model:
        model = Contrat


class AyantDroitForm(forms.ModelForm, AjaxSelect):

    # ne fonctionne pas dans un inline

    def __init__(self, *args, **kwargs):
        super(AyantDroitForm, self).__init__(*args, **kwargs)
        self.fields['date_naissance'].widget = forms.widgets.DateInput()

    class Meta:
        model = AyantDroit


class EmployeAdminForm(forms.ModelForm, AjaxSelect):

    nationalite = AutoCompleteSelectField(
        'pays', help_text="Taper le nom ou le code du pays", required=False
    )
    pays = AutoCompleteSelectField(
        'pays', help_text="Taper le nom ou le code du pays", required=False
    )

    class Meta:
        model = Employe

    def __init__(self, *args, **kwargs):
        super(EmployeAdminForm, self).__init__(*args, **kwargs)
        self.fields['date_naissance'].widget = forms.widgets.DateInput()


class ResponsableInlineForm(forms.ModelForm):
    employes_actifs = Employe.objects.actifs()
    employe = forms.ModelChoiceField(queryset=employes_actifs)

    class Meta:
        model = ResponsableImplantation