18b11765dd652f6a870dbc5ab3dd034b11fee22f
[auf_rh_dae.git] / project / dae / forms.py
1 # -*- encoding: utf-8 -*-
2
3 from django import forms
4
5 from datamaster_modeles import models as ref
6
7 from dae import models as dae
8 from rh_v1 import models as rh
9
10
11 class PosteForm(forms.ModelForm):
12 """ Formulaire des postes. """
13 class Meta:
14 model = dae.Poste
15
16 fields = ('poste', 'implantation', 'type_poste', 'service', 'nom',
17 'responsable', 'statut_residence', 'mise_a_disposition',
18 'appel', 'date_debut', 'date_fin', 'actif',
19 'regime_travail', 'regime_travail_nb_heure_semaine',
20 'classement_min', 'classement_max',
21 'valeur_point_min', 'valeur_point_max',
22 'salaire_min', 'salaire_max', 'indemn_min', 'indemn_max',
23 'autre_min', 'autre_max', 'devise_comparaison',
24 'comp_locale_min', 'comp_locale_max',
25 'comp_universite_min', 'comp_universite_max',
26 'comp_fonctionpub_min', 'comp_fonctionpub_max',
27 'comp_ong_min', 'comp_ong_max',
28 'comp_autre_min', 'comp_autre_max',)
29 widgets = dict(statut_residence=forms.RadioSelect(),
30 appel=forms.RadioSelect(),
31 nom=forms.TextInput(attrs={'size': 60 }),
32 )
33
34 responsable=forms.ModelChoiceField(
35 queryset=rh.Poste.objects.select_related(depth=1))
36
37 # La liste des choix est laissée vide. Voir __init__ pour la raison.
38 poste = forms.ChoiceField(label="Nouveau poste, ou évolution du poste", choices=(), required=False)
39
40 def __init__(self, *args, **kwargs):
41 """ Mise à jour dynamique du contenu du menu des postes.
42
43 Si on ne met le menu à jour de cette façon, à chaque instantiation du
44 formulaire, son contenu est mis en cache par le système et il ne
45 reflète pas les changements apportés par les ajouts, modifications,
46 etc...
47
48 Aussi, dans ce cas-ci, on ne peut pas utiliser un ModelChoiceField
49 car le "id" de chaque choix est spécial (voir _poste_choices).
50
51 """
52 super(PosteForm, self).__init__(*args, **kwargs)
53 self.fields['poste'].choices = self._poste_choices()
54
55 def _poste_choices(self):
56 """ Menu déroulant pour les postes.
57
58 Constitué des postes de dae et des postes de rh_v1 qui n'ont pas
59 d'équivalent dans dae.
60
61 """
62 dae_ = dae.Poste.objects.filter(id_rh__isnull=True)
63 copies = dae.Poste.objects.exclude(id_rh__isnull=True)
64 id_copies = [p.id_rh_id for p in copies.all()]
65 rhv1 = rh.Poste.objects.exclude(id__in=id_copies)
66 # Optimisation de la requête
67 rhv1 = rhv1.select_related(depth=1)
68
69 return [('', 'Nouveau poste')]+ \
70 sorted([('dae-%s' % p.id, unicode(p)) for p in dae_ | copies] +
71 [('rh-%s' % p.id, unicode(p)) for p in rhv1],
72 key=lambda t: t[1])
73
74 def save(self, *args, **kwargs):
75 kwargs2 = kwargs.copy()
76 kwargs2['commit'] = False
77 poste = super(PosteForm, self).save(*args, **kwargs2)
78 # id_rh
79 if 'commit' not in kwargs or kwargs['commit']:
80 poste.save()
81 return poste
82
83
84 class ChoosePosteForm(forms.ModelForm):
85 class Meta:
86 model = dae.Poste
87 fields = ('poste',)
88
89 # La liste des choix est laissée vide. Voir PosteForm.__init__.
90 poste = forms.ChoiceField(choices=(), required=False)
91
92 def __init__(self, *args, **kwargs):
93 super(ChoosePosteForm, self).__init__(*args, **kwargs)
94 self.fields['poste'].choices = self._poste_choices()
95
96 def _poste_choices(self):
97 """ Menu déroulant pour les postes. """
98 dae_ = dae.Poste.objects.filter(id_rh__isnull=True)
99 copies = dae.Poste.objects.exclude(id_rh__isnull=True)
100 id_copies = [p.id_rh_id for p in copies.all()]
101
102 return [('', '----------')]+ \
103 sorted([('dae-%s' % p.id, unicode(p)) for p in dae_ | copies],
104 key=lambda t: t[1])
105
106
107 class PosteFinancementForm(forms.ModelForm):
108 """ Formulaire des sources de financement pour un poste. """
109 class Meta:
110 model = dae.PosteFinancement
111 fields = ('type', 'montant', 'pourcentage', 'commentaire')
112
113
114 class EmployeForm(forms.ModelForm):
115 """ Formulaire des employés. """
116 class Meta:
117 model = dae.Employe
118 fields = ('employe', 'nom', 'prenom', 'genre')
119
120 # La liste des choix est laissée vide. Voir Poste.__init__ pour la raison.
121 employe = forms.ChoiceField(choices=(), required=False)
122
123 def __init__(self, *args, **kwargs):
124 """ Mise à jour dynamique du contenu du menu des employés. """
125 super(EmployeForm, self).__init__(*args, **kwargs)
126 self.fields['employe'].choices = self._employe_choices()
127
128 def _employe_choices(self):
129 """ Menu déroulant pour les employés. """
130 dae_ = dae.Employe.objects.filter(id_rh__isnull=True)
131 copies = dae.Employe.objects.exclude(id_rh__isnull=True)
132 id_copies = [p.id_rh_id for p in copies.all()]
133 rhv1 = rh.Employe.objects.exclude(id__in=id_copies)
134
135 return [('', 'Nouvel employé')] + \
136 sorted([('dae-%s' % p.id, unicode(p)) for p in dae_ | copies] +
137 [('rh-%s' % p.id, unicode(p)) for p in rhv1],
138 key=lambda t: t[1])
139
140
141 class DossierForm(forms.ModelForm):
142 """ Formulaire des dossiers. """
143 class Meta:
144 model = dae.Dossier