[#3165] Remplacé les filtres et protection par région pour des filtres et protections...

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

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

from datetime import date

from ajax_select.fields import AutoCompleteSelectField
from auf.django.references import models as ref
from django import forms
from django.db.models import Min

from project.groups import get_employe_from_user, is_user_dans_region
from project.rh import models as rh


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 = rh.Dossier


class ContratForm(forms.ModelForm, FormDate):

    class Model:
        model = rh.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 = rh.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 = rh.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 = rh.Employe.objects.actifs()
    employe = forms.ModelChoiceField(queryset=employes_actifs)

    class Meta:
        model = rh.ResponsableImplantation


class MasseSalarialeForm(forms.Form):
    zone_administrative = forms.ModelChoiceField(
        label=u'Région', queryset=ref.ZoneAdministrative.objects.all(),
        required=False
    )
    implantation = forms.ModelChoiceField(
        label=u'Implantation', queryset=ref.Implantation.objects.all(),
        required=False
    )

    def __init__(self, user, *args, **kwargs):
        super(MasseSalarialeForm, self).__init__(*args, **kwargs)
        min_date = rh.Remuneration.objects \
                .aggregate(Min('date_debut'))['date_debut__min']
        years = range(
            date.today().year,
            min_date.year if min_date else 1997,
            -1
        )
        self.fields['annee'] = forms.TypedChoiceField(
            label='Année', choices=zip(years, years), coerce=int,
            required=False
        )
        if is_user_dans_region(user):
            employe = get_employe_from_user(user)
            self.fields['zone_administrative'].queryset = \
                    ref.ZoneAdministrative.objects.filter(
                        code=employe.implantation.zone_administrative.code
                    )
            self.fields['implantation'].queryset = \
                    ref.Implantation.objects.filter(
                        zone_administrative=(
                            employe.implantation.zone_administrative
                        )
                    )