1622: Trop de requêtes lors de l'ajout d'une offre d'emploi
[auf_rh_dae.git] / project / recrutement / admin.py
1 # -*- encoding: utf-8 -*-
2
3 from django.core.urlresolvers import reverse
4 from django.http import HttpResponseRedirect
5 from django.contrib import admin
6 from django.shortcuts import get_object_or_404
7
8 from reversion.admin import VersionAdmin
9 from datamaster_modeles.models import Employe, Implantation, Region
10
11 from recrutement.models import *
12
13 class OffreEmploiAdmin(VersionAdmin):
14 date_hierarchy = 'date_creation'
15 list_display = ('nom', 'resume', 'date_limite', '_candidatsList', 'region')
16
17 def _candidatsList(self, obj):
18 return "<a href='%s?offre_emploi__id__exact=%s'>Voir les candidats \
19 </a>" % (reverse('admin:recrutement_candidat_changelist'), obj.id)
20 _candidatsList.allow_tags = True
21 _candidatsList.short_description = "Liste des candidats"
22
23 def get_form(self, request, obj=None, **kwargs):
24 form = super(OffreEmploiAdmin, self).get_form(request, obj, **kwargs)
25 if form.declared_fields.has_key('region'):
26 region_field = form.declared_fields['region']
27 else:
28 region_field = form.base_fields['region']
29
30 user = UserProfile.objects.get(user=request.user)
31 region_ids = [g.id for g in user.regions.all()]
32 region_field.queryset = Region.objects.filter(id__in=region_ids)
33 return form
34
35 def queryset(self, request):
36 """
37 Spécifie un queryset limité, autrement Django exécute un
38 select_related() sans paramètre, ce qui a pour effet de charger tous
39 les objets FK, sans limite de profondeur. Dès qu'on arrive, dans les
40 modèles de Region, il existe plusieurs boucles, ce qui conduit à la
41 génération d'une requête infinie.
42 """
43 qs = self.model._default_manager.get_query_set()
44 try:
45 user = UserProfile.objects.get(user=request.user)
46 except UserProfile.DoesNotExist:
47 return qs.none()
48
49 region_ids = [g.id for g in user.regions.all()]
50 return qs.select_related('offre_emploi').filter(region__in=region_ids)
51
52 class CandidatPieceInline(admin.TabularInline):
53 model = CandidatPiece
54 extra = 1
55
56 class EvaluateurInline(admin.TabularInline):
57 model = UserProfile.candidats.through
58 extra = 1
59
60 class CandidatAdmin(VersionAdmin):
61 date_hierarchy = 'date_creation'
62 list_display = ('nom', 'prenom', 'offre_emploi','statut', '_actions',
63 'evaluer_candidat', )
64 list_filter = ('offre_emploi', )
65 fieldsets = (
66 ("Offre d'emploi", {
67 'fields': ('offre_emploi', )
68 }),
69 ('Informations personnelles', {
70 'fields': ('prenom','nom','genre', 'nationalite', 'date_naissance',
71 'situation_famille', 'nombre_dependant',)
72 }),
73 ('Adresse', {
74 'fields': ('adresse', 'ville', 'etat_province', 'pays', )
75 }),
76 ('Informations professionnelles', {
77 'fields': ('niveau_diplome','employeur_actuel',
78 'poste_actuel', 'domaine_professionnel',)
79 }),
80 ('Options avancées', {
81 'classes': ('collapse',),
82 'fields': ('actif', 'statut', )
83 }),
84 )
85 inlines = [
86 CandidatPieceInline,
87 EvaluateurInline,
88 ]
89
90 # Affecter un évaluateurs à des candidats
91 actions = ['affecter_candidats_evaluateur']
92 def affecter_candidats_evaluateur(modeladmin, obj, candidats):
93
94 selected = obj.POST.getlist(admin.ACTION_CHECKBOX_NAME)
95
96 return HttpResponseRedirect(reverse('affecter_evaluateurs_candidats')+
97 "?ids=%s" % (",".join(selected)))
98 affecter_candidats_evaluateur.short_description = "Affecter evaluateur"
99
100 def evaluer_candidat(self, obj):
101 return "<a href='%s?id=%s'>Évaluer le candidat \
102 </a>" % (reverse('evaluer_candidat'), obj.id)
103 evaluer_candidat.allow_tags = True
104 evaluer_candidat.short_description = "Évaluer"
105
106 def get_form(self, request, obj=None, **kwargs):
107 form = super(CandidatAdmin, self).get_form(request, obj, **kwargs)
108 if form.declared_fields.has_key('offre_emploi'):
109 region_field = form.declared_fields['offre_emploi']
110 else:
111 region_field = form.base_fields['offre_emploi']
112
113 try:
114 user = UserProfile.objects.get(user=request.user)
115 except UserProfile.DoesNotExist:
116 region_field.queryset = region_field.queryset.none()
117 return form
118
119 region_ids = [g.id for g in user.regions.all()]
120 region_field.queryset = Region.objects.filter(id__in=region_ids)
121 return form
122
123 def queryset(self, obj):
124 """
125 Spécifie un queryset limité, autrement Django exécute un
126 select_related() sans paramètre, ce qui a pour effet de charger tous
127 les objets FK, sans limite de profondeur. Dès qu'on arrive, dans les
128 modèles de Region, il existe plusieurs boucles, ce qui conduit à la
129 génération d'une requête infinie.
130 """
131
132 qs = self.model._default_manager.get_query_set()
133 try:
134 user = UserProfile.objects.get(user=obj.user)
135 except UserProfile.DoesNotExist:
136 return qs.none()
137
138 ids = [c.id for c in user.candidats.all()]
139 return qs.select_related('candidats').filter(id__in=ids)
140
141 def _actions(self, obj):
142 return "<a href='%s?id=%s'>Voir l'offre d'emploi</a> \
143 " % (reverse('admin:recrutement_offreemploi_changelist'),
144 obj.offre_emploi.id)
145 _actions.allow_tags = True
146 _actions.short_description = "Offre d'emploi"
147
148 class CandidatPieceAdmin(admin.ModelAdmin):
149 list_display = ('nom', 'candidat', )
150
151 def queryset(self, request):
152 """
153 Spécifie un queryset limité, autrement Django exécute un
154 select_related() sans paramètre, ce qui a pour effet de charger tous
155 les objets FK, sans limite de profondeur. Dès qu'on arrive, dans les
156 modèles de Region, il existe plusieurs boucles, ce qui conduit à la
157 génération d'une requête infinie.
158 """
159 qs = self.model._default_manager.get_query_set()
160 return qs.select_related('candidat')
161
162 class EvaluateurAdmin(VersionAdmin):
163 fieldsets = (
164 (None, {'fields': ('user', 'regions')}),
165 (None, {'fields': ('candidats',)}),
166 )
167
168 class CandidatEvaluationAdmin(VersionAdmin):
169 list_display = ('candidat', 'evaluateur', 'note', 'commentaire', 'date', )
170
171 def queryset(self, obj):
172 """
173 Spécifie un queryset limité, autrement Django exécute un
174 select_related() sans paramètre, ce qui a pour effet de charger tous
175 les objets FK, sans limite de profondeur. Dès qu'on arrive, dans les
176 modèles de Region, il existe plusieurs boucles, ce qui conduit à la
177 génération d'une requête infinie.
178 """
179 qs = self.model._default_manager.get_query_set()
180 return qs.select_related('offre_emploi')
181
182 admin.site.register(OffreEmploi, OffreEmploiAdmin)
183 admin.site.register(Candidat, CandidatAdmin)
184 admin.site.register(CandidatPiece, CandidatPieceAdmin)
185 admin.site.register(UserProfile, EvaluateurAdmin)
186 admin.site.register(CandidatEvaluation, CandidatEvaluationAdmin)