Fix django ajax selects

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

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

from django import forms
from django.forms import ModelForm

from auf.django.emploi import forms as emploi

from project.recrutement import models as recr

########################################################################
# EVALUATION
########################################################################
class CandidatEvaluationForm(ModelForm):
    def __init__(self, *args, **kwargs):   
        self.candidat = kwargs.pop('candidat')    
        self.evaluateur = kwargs.pop('evaluateur') 
        super(CandidatEvaluationForm, self).__init__(*args, **kwargs)

    def save(self):
        super(CandidatEvaluationForm, self).save()

    class Meta:
        fields = ('note', 'commentaire')
        model = recr.CandidatEvaluation  

class EvaluateurForm(forms.Form):
    evaluateurs = forms.ModelMultipleChoiceField(queryset=
                    recr.Evaluateur.objects.all(), required=False)


    def __init__(self, *args, **kwargs):
        self.offres_emploi = kwargs.pop('offres_emploi')
        evaluateurs = [e.id for e in self.get_evaluateurs()]
        initial = getattr(kwargs, 'initial', {})
        initial['evaluateurs'] = evaluateurs
        kwargs['initial'] = initial
        super(EvaluateurForm, self).__init__(*args, **kwargs)


    def get_evaluateurs(self):
        return [e.evaluateur for e in \
                recr.CandidatEvaluation.objects.filter(candidat__offre_emploi=self.offres_emploi)]

    def save(self):
        candidats = recr.Candidat.objects.\
                            filter(offre_emploi=self.offres_emploi)

        anciens_evaluateurs = self.get_evaluateurs()
        nouveaux__evaluateurs = self.cleaned_data.get('evaluateurs', [])

        # suppression des évaluations du type
        evaluations = recr.CandidatEvaluation.objects.filter(candidat__offre_emploi=self.offres_emploi)
        for e in evaluations:
            if e.evaluateur not in nouveaux__evaluateurs:
                e.delete()

        # on prépopulent les évaluations si elles ne l'ont pas déja été
        for candidat in candidats:
            for evaluateur in self.cleaned_data.get('evaluateurs', []):
                if evaluateur not in anciens_evaluateurs:
                    candidat_evaluation = recr.CandidatEvaluation()
                    candidat_evaluation.candidat = candidat
                    candidat_evaluation.evaluateur = evaluateur
                    candidat_evaluation.save()
        

########################################################################
# OFFRE EMPLOI
########################################################################
class CandidatPieceForm(emploi.CandidatPieceForm):
    pass

class PostulerOffreEmploiForm(emploi.PostulerOffreEmploiForm):
    pass

class OffreEmploiForm(ModelForm):
    def clean(self):
        cleaned_data = self.cleaned_data
        date_limite = cleaned_data.get("date_limite")
        debut_affectation = cleaned_data.get("debut_affectation")

        if date_limite and debut_affectation:
            if date_limite > debut_affectation:
                raise forms.ValidationError("La date limite ne peut pas être \
                        supérieure à la date d'affection.")
        return cleaned_data

########################################################################
# TEMPLATE COURRIEL
########################################################################
class CandidatCourrielTemplateForm(ModelForm):
    def get_template(self):
        return self.data['template']

    class Meta:
        model = recr.CandidatCourriel
        fields = ('template', )

class CandidatCourrielForm(ModelForm):
    def __init__(self, *args, **kwargs):
        self.candidats = kwargs.pop('candidats')
        self.template = kwargs.pop('template')
        super(CandidatCourrielForm, self).__init__(*args, **kwargs)

    def save(self):
        super(CandidatCourrielForm, self).save()

    class Meta:
        model = recr.CandidatCourriel
        fields = ('sujet', 'plain_text', )