Retrait du filtrage par région des évaluateurs

[auf_rh_dae.git] / project / recrutement / admin.py

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

from django.core.urlresolvers import reverse
from django.http import HttpResponseRedirect
from django.contrib import admin
from django.shortcuts import get_object_or_404

from reversion.admin import VersionAdmin
from datamaster_modeles.models import Employe, Implantation, Region

from recrutement.models import *

class OffreEmploiAdmin(VersionAdmin):
    date_hierarchy = 'date_creation'
    list_display = ('nom', 'resume', 'date_limite', '_candidatsList', 'region')
        
    def _candidatsList(self, obj):     
        return "<a href='%s?offre_emploi__id__exact=%s'>Voir les candidats \
            </a>" % (reverse('admin:recrutement_candidat_changelist'), obj.id)
    _candidatsList.allow_tags = True 
    _candidatsList.short_description = "Liste des candidats"   

    def get_form(self, request, obj=None, **kwargs):
        """
        Affiche la liste des régions, seulement si le user
        connecté possède un UserProfile
        """
        form = super(OffreEmploiAdmin, self).get_form(request, obj, **kwargs)
        if form.declared_fields.has_key('region'):
            region_field = form.declared_fields['region']
        else:
            region_field = form.base_fields['region']

        try:
            user = UserProfile.objects.get(user=request.user)
        except UserProfile.DoesNotExist:
            region_field.queryset = Region.objects.none()
            return form

        region_ids = [g.id for g in user.regions.all()]
        region_field.queryset = Region.objects.filter(id__in=region_ids)
        return form

    def queryset(self, request):
        """ 
        N'affiche la liste de candidats que si le user connecté 
        possède un UserProfile
        """
        qs = self.model._default_manager.get_query_set()
        try:
            user = UserProfile.objects.get(user=request.user)
        except UserProfile.DoesNotExist:
            return qs.none()
        
        region_ids = [g.id for g in user.regions.all()]
        return qs.select_related('offre_emploi').filter(region__in=region_ids)

class ProxyCandidatPiece(CandidatPiece):
    """
    Ce proxy sert uniquement dans l'admin à disposer d'un libellé
    plus ergonomique.
    """
    class Meta:
        proxy = True
        verbose_name = "pièce jointe"
        verbose_name_plural = "pièces jointes"

class CandidatPieceInline(admin.TabularInline):
    model = ProxyCandidatPiece
    fields = ('candidat', 'nom', 'path', )
    extra = 1

class ProxyEvaluateur(UserProfile.candidats.through):
    """
    Ce proxy sert uniquement dans l'admin à disposer d'un libellé
    plus ergonomique.
    """
    class Meta:
        proxy = True
        verbose_name = "évaluateur"

class EvaluateurInline(admin.TabularInline):
    model = ProxyEvaluateur
    extra = 1

class CandidatAdmin(VersionAdmin):
    date_hierarchy = 'date_creation'
    list_display = ('nom', 'prenom', 'offre_emploi','statut', '_actions', 
                    'evaluer_candidat', )
    list_filter = ('offre_emploi', )
    fieldsets = (
        ("Offre d'emploi", {
            'fields': ('offre_emploi', )
        }),
        ('Informations personnelles', {
            'fields': ('prenom','nom','genre', 'nationalite', 'date_naissance',
                        'situation_famille', 'nombre_dependant',)        
        }),
        ('Adresse', {
            'fields': ('adresse', 'ville', 'etat_province', 'pays', )
        }),
        ('Informations professionnelles', {
            'fields': ('niveau_diplome','employeur_actuel', 
                        'poste_actuel', 'domaine_professionnel',)
        }),  
        ('Options avancées', {
            'classes': ('collapse',),
            'fields': ('actif', 'statut', )
        }),
    )
    inlines = [
        CandidatPieceInline,
        EvaluateurInline,
    ]

    # Affecter un évaluateurs à des candidats
    actions = ['affecter_candidats_evaluateur']
    def affecter_candidats_evaluateur(modeladmin, obj, candidats):
        
        selected = obj.POST.getlist(admin.ACTION_CHECKBOX_NAME)

        return HttpResponseRedirect(reverse('affecter_evaluateurs_candidats')+
                "?ids=%s" % (",".join(selected)))
    affecter_candidats_evaluateur.short_description = u'Affecter évaluateur'

    # Évaluer un candidat
    def evaluer_candidat(self, obj):
        return "<a href='%s?id=%s'>Évaluer le candidat \
                </a>" % (reverse('evaluer_candidat'), obj.id)
    evaluer_candidat.allow_tags = True    
    evaluer_candidat.short_description = 'Évaluer'

    def get_form(self, request, obj=None, **kwargs):
        """
        Affiche la liste des offres d'emploi, seulement si le user
        connecté possède un UserProfile
        """
        form = super(CandidatAdmin, self).get_form(request, obj, **kwargs)
        if form.declared_fields.has_key('offre_emploi'):
            region_field = form.declared_fields['offre_emploi']
        else:
            region_field = form.base_fields['offre_emploi']

        try:
            user = UserProfile.objects.get(user=request.user)
        except UserProfile.DoesNotExist:
            region_field.queryset = region_field.queryset.none()
            return form  

        region_ids = [g.id for g in user.regions.all()]
        region_field.queryset = OffreEmploi.objects.filter(region__in=region_ids)
        return form

    def queryset(self, obj):
        """
        Spécifie un queryset limité, autrement Django exécute un 
        select_related() sans paramètre, ce qui a pour effet de charger tous 
        les objets FK, sans limite de profondeur. Dès qu'on arrive, dans les 
        modèles de Region, il existe plusieurs boucles, ce qui conduit à la 
        génération d'une requête infinie.
        
        """

        qs = self.model._default_manager.get_query_set()
        try:
            user = UserProfile.objects.get(user=obj.user)
        except UserProfile.DoesNotExist:
            return qs.none()      
    
        ids = [c.id for c in user.candidats.all()]
        return qs.select_related('candidats').filter(id__in=ids)

    def _actions(self, obj):
        return "<a href='%s?id=%s'>Voir l'offre d'emploi</a> \
                " % (reverse('admin:recrutement_offreemploi_changelist'), 
                obj.offre_emploi.id)
    _actions.allow_tags = True
    _actions.short_description = "Offre d'emploi"

class CandidatPieceAdmin(admin.ModelAdmin):
    list_display = ('nom', 'candidat', )

    def queryset(self, request):
        """
        Spécifie un queryset limité, autrement Django exécute un 
        select_related() sans paramètre, ce qui a pour effet de charger tous 
        les objets FK, sans limite de profondeur. Dès qu'on arrive, dans les 
        modèles de Region, il existe plusieurs boucles, ce qui conduit à la 
        génération d'une requête infinie.
        Affiche la liste de candidats que si le user connecté 
        possède un UserProfile
        """
        qs = self.model._default_manager.get_query_set()
        return qs.select_related('candidat')

class EvaluateurAdmin(VersionAdmin):
    fieldsets = (
        (None, {'fields': ('user', 'regions')}),
        (None, {'fields': ('candidats',)}),
    )

    def get_form(self, request, obj=None, **kwargs):
        """
        Affiche la liste des users, regions et candidats, seulement si le user
        connecté possède un UserProfile
        """
        form = super(EvaluateurAdmin, self).get_form(request, obj, **kwargs)
        if form.declared_fields.has_key('user'):
            permission_field = form.declared_fields['user']
        else:
            permission_field = form.base_fields['user']

        if form.declared_fields.has_key('regions'):
            region_field = form.declared_fields['regions']
        else:
            region_field = form.base_fields['regions']

        if form.declared_fields.has_key('candidats'):
            candidats_field = form.declared_fields['candidats']
        else:
            candidats_field = form.base_fields['candidats']

        try:
            user = UserProfile.objects.get(user=request.user)
        except UserProfile.DoesNotExist:
            region_field.queryset = Region.objects.none()
            permission_field.queryset = UserProfile.objects.none()
            candidats_field.queryset = Candidat.objects.none()
            return form

        return form

    def queryset(self, request):
        """ 
        Affiche la liste d'évaluateur que si le user connecté 
        possède un UserProfile
        """
        qs = self.model._default_manager.get_query_set()
        try:
            user = UserProfile.objects.get(user=request.user)
        except UserProfile.DoesNotExist:
            return qs.none()
        
        return qs

class CandidatEvaluationAdmin(VersionAdmin):
    list_display = ('candidat', 'evaluateur', 'note', 'commentaire', 'date', )

    def get_form(self, request, obj=None, **kwargs):
        """
        Affiche la liste des candidats et évaluateur, seulement si le user
        connecté possède un UserProfile
        """
        form = super(CandidatEvaluationAdmin, self).get_form(request, obj,
                **kwargs)
        if form.declared_fields.has_key('evaluateur'):
            evaluateur_field = form.declared_fields['evaluateur']
        else:
            evaluateur_field = form.base_fields['evaluateur']

        if form.declared_fields.has_key('candidat'):
            candidats_field = form.declared_fields['candidat']
        else:
            candidats_field = form.base_fields['candidat']

        try:
            user = UserProfile.objects.get(user=request.user)
        except UserProfile.DoesNotExist:
            evaluateur_field.queryset = UserProfile.objects.none()
            candidats_field.queryset = Candidat.objects.none()
            return form

        return form

    def queryset(self, request):
        """ 
        Affiche la liste d'évaluation que si le user connecté 
        possède un UserProfile
        """
        qs = self.model._default_manager.get_query_set()
        try:
            user = UserProfile.objects.get(user=request.user)
        except UserProfile.DoesNotExist:
            return qs.none()

        return qs.select_related('candidat')

admin.site.register(OffreEmploi, OffreEmploiAdmin)
admin.site.register(Candidat, CandidatAdmin)
admin.site.register(UserProfile, EvaluateurAdmin)
admin.site.register(CandidatEvaluation, CandidatEvaluationAdmin)