réparé le formulaire de poste

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

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

from django import forms

from datamaster_modeles import models as ref

from dae import models as dae
from rh_v1 import models as rh


class PosteForm(forms.ModelForm):
    """ Formulaire des postes. """
    class Meta:
        model = dae.Poste

        fields = ('poste', 'implantation', 'type_poste', 'service', 'nom',
                  'responsable', 'statut_residence', 'mise_a_disposition',
                  'appel', 'date_debut', 'date_fin', 'actif',
                  'regime_travail', 'regime_travail_nb_heure_semaine',
                  'classement_min', 'classement_max',
                  'valeur_point_min', 'valeur_point_max',
                  'salaire_min', 'salaire_max', 'indemn_min', 'indemn_max',
                  'autre_min', 'autre_max', 'devise_comparaison',
                  'comp_locale_min', 'comp_locale_max',
                  'comp_universite_min', 'comp_universite_max',
                  'comp_fonctionpub_min', 'comp_fonctionpub_max',
                  'comp_ong_min', 'comp_ong_max',
                  'comp_autre_min', 'comp_autre_max',)
        widgets = dict(statut_residence=forms.RadioSelect(),
                       appel=forms.RadioSelect(),
                       nom=forms.TextInput(attrs={'size': 60 }),
)

    responsable=forms.ModelChoiceField(
            queryset=rh.Poste.objects.select_related(depth=1))

    # La liste des choix est laissée vide. Voir __init__ pour la raison.
    poste = forms.ChoiceField(label="Nouveau poste, ou évolution du poste",
                              choices=(), required=False)

    def __init__(self, *args, **kwargs):
        """ Mise à jour dynamique du contenu du menu des postes.

        Si on ne met le menu à jour de cette façon, à chaque instantiation du
        formulaire, son contenu est mis en cache par le système et il ne
        reflète pas les changements apportés par les ajouts, modifications,
        etc...

        Aussi, dans ce cas-ci, on ne peut pas utiliser un ModelChoiceField
        car le "id" de chaque choix est spécial (voir _poste_choices).

        """
        super(PosteForm, self).__init__(*args, **kwargs)
        self.fields['poste'].choices = self._poste_choices()

    def _poste_choices(self):
        """ Menu déroulant pour les postes.

        Constitué des postes de dae et des postes de rh_v1 qui n'ont pas
        d'équivalent dans dae.

        """
        dae_ = dae.Poste.objects.filter(id_rh__isnull=True)
        copies = dae.Poste.objects.exclude(id_rh__isnull=True)
        id_copies = [p.id_rh_id for p in copies.all()]
        rhv1 = rh.Poste.objects.exclude(id__in=id_copies)
        # Optimisation de la requête
        rhv1 = rhv1.select_related(depth=1)

        return [('', 'Nouveau poste')]+ \
               sorted([('dae-%s' % p.id, unicode(p)) for p in dae_ | copies] +
                      [('rh-%s' % p.id, unicode(p)) for p in rhv1],
                      key=lambda t: t[1])

    def save(self, *args, **kwargs):
        kwargs2 = kwargs.copy()
        kwargs2['commit'] = False
        poste = super(PosteForm, self).save(*args, **kwargs2)
        # id_rh
        if 'commit' not in kwargs or kwargs['commit']:
            poste.save()
        return poste


class ChoosePosteForm(forms.ModelForm):
    class Meta:
        model = dae.Poste
        fields = ('poste',)

    # La liste des choix est laissée vide. Voir PosteForm.__init__.
    poste = forms.ChoiceField(choices=(), required=False)

    def __init__(self, *args, **kwargs):
        super(ChoosePosteForm, self).__init__(*args, **kwargs)
        self.fields['poste'].choices = self._poste_choices()

    def _poste_choices(self):
        """ Menu déroulant pour les postes. """
        dae_ = dae.Poste.objects.filter(id_rh__isnull=True)
        copies = dae.Poste.objects.exclude(id_rh__isnull=True)
        id_copies = [p.id_rh_id for p in copies.all()]

        return [('', '----------')]+ \
               sorted([('dae-%s' % p.id, unicode(p)) for p in dae_ | copies],
                      key=lambda t: t[1])


class PosteFinancementForm(forms.ModelForm):
    """ Formulaire des sources de financement pour un poste. """
    class Meta:
        model = dae.PosteFinancement
        fields = ('type', 'montant', 'pourcentage', 'commentaire')


class EmployeForm(forms.ModelForm):
    """ Formulaire des employés. """
    class Meta:
        model = dae.Employe
        fields = ('employe', 'nom', 'prenom', 'genre')

    # La liste des choix est laissée vide. Voir Poste.__init__ pour la raison.
    employe = forms.ChoiceField(choices=(), required=False)

    def __init__(self, *args, **kwargs):
        """ Mise à jour dynamique du contenu du menu des employés. """
        super(EmployeForm, self).__init__(*args, **kwargs)
        self.fields['employe'].choices = self._employe_choices()

    def _employe_choices(self):
        """ Menu déroulant pour les employés. """
        dae_ = dae.Employe.objects.filter(id_rh__isnull=True)
        copies = dae.Employe.objects.exclude(id_rh__isnull=True)
        id_copies = [p.id_rh_id for p in copies.all()]
        rhv1 = rh.Employe.objects.exclude(id__in=id_copies)

        return [('', 'Nouvel employé')] + \
               sorted([('dae-%s' % p.id, unicode(p)) for p in dae_ | copies] +
                      [('rh-%s' % p.id, unicode(p)) for p in rhv1],
                      key=lambda t: t[1])


class DossierForm(forms.ModelForm):
    """ Formulaire des dossiers. """
    class Meta:
        model = dae.Dossier