Correction au UserProfile, remplacer foreign key par User, au lieu de auth.User

[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', 'region', '_candidatsList')
        
    # Afficher la liste des candidats pour l'offre d'emploi
    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 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',
                    'voir_offre_emploi', 'evaluer_candidat', #'note_evaluateur',
                    'calculer_moyenne', 'afficher_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',)        
        }),
        ('Coordonnées', {
            'fields': ('telephone', 'email', 'adresse', 'ville', 
                        'etat_province', 'code_postal', 'pays', )
        }),
        ('Informations professionnelles', {
            'fields': ('niveau_diplome','employeur_actuel', 
                        'poste_actuel', 'domaine_professionnel',)
        }),  
        ('Options avancées', {
            'classes': ('collapse',),
            'fields': ('actif', 'statut', )
        }),
    )
    inlines = [
        CandidatPieceInline,
        EvaluateurInline,
    ]

    actions = ['affecter_candidats_evaluateur', ] #'envoyer_courriel_candidats']
    # Affecter un évaluateurs à des candidats
    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'

    # Affecter un évaluateurs à des candidats
    def envoyer_courriel_candidats(modeladmin, obj, candidats):   
        selected = obj.POST.getlist(admin.ACTION_CHECKBOX_NAME)

        return HttpResponseRedirect(reverse('envoyer_courriel_candidats')+
                "?ids=%s" % (",".join(selected)))
    envoyer_courriel_candidats.short_description = u'Envoyer courriel'

    # É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'

    # Afficher un candidat
    def afficher_candidat(self, obj):
        return "<a href='%s?id=%s'>Voir le candidat \
                </a>" % (reverse('afficher_candidat'), obj.id)
    afficher_candidat.allow_tags = True    
    afficher_candidat.short_description = u'Afficher les détails'

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

    """    # Afficher note poster par l'évaluateur
    def note_evaluateur(self, obj):
        import pdb;pdb.set_trace()
        evaluateur = UserProfile.objects.get(user=obj.user)
        import pdb;pdb.set_trace()
        note_eval = CandidatEvaluation.objects.filter(evaluateur=evaluateur, 
                        candidat=obj.user)
        return "blabla"
    note_evaluateur.allow_tags = True
    note_evaluateur.short_description = "Votre note"        
    """

    # Calculer la moyenne des notes
    def calculer_moyenne(self, obj):
        evaluations = CandidatEvaluation.objects.filter(candidat=obj)
        offre_emploi = obj.offre_emploi

        notes = [evaluation.note for evaluation in evaluations.all() if evaluation.note is not None]
 
        if len(notes) > 0 and offre_emploi.date_limite <= datetime.date.today():
            moyenne_votes = float(sum(notes)) / len(notes)
        else:
            moyenne_votes = "Non disponible"
        return moyenne_votes
    calculer_moyenne.allow_tags = True
    calculer_moyenne.short_description = "Moyenne des notes"

    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)

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)