refact ergo + page perso (vide)

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

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

import os
from django import forms
from django.contrib import admin
from django.forms.models import inlineformset_factory
from datetime import timedelta
from django.forms.widgets import CheckboxSelectMultiple
from django.contrib.admin import widgets as admin_widgets   
from form_utils.forms import BetterModelForm
from django.forms import ModelForm, ModelChoiceField, HiddenInput, CharField
from django.forms.models import BaseInlineFormSet 
from django.core.mail import send_mail

from tinymce.widgets import TinyMCE
from captcha.fields import CaptchaField

from recrutement import models as recr
from auf.django.emploi import forms as emploi
from project.rh import models as rh

################################################################################
# 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())

    def __init__(self, *args, **kwargs):
        self.offres_emploi = kwargs.pop('offres_emploi')
        super(EvaluateurForm, self).__init__(*args, **kwargs)

    def save(self):
        candidats = recr.Candidat.objects.\
                            filter(offre_emploi__in=self.offres_emploi)
        for candidat in candidats:
            for evaluateur in self.cleaned_data.get('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):
    poste = ModelChoiceField(queryset=rh.Poste.objects.all())

    class Meta:
        model = recr.OffreEmploi  

    def save(self, *args, **kwargs):
        kwargs2 = kwargs.copy()
        kwargs2['commit'] = False
        offre = super(OffreEmploiForm, self).save(*args, **kwargs2)
        offre.poste = self.cleaned_data.get("poste")
        offre.poste_nom = self.cleaned_data.get("poste").nom
        if 'commit' not in kwargs or kwargs['commit']:
            offre.save()
        return offre

    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.")

        """if date_limite < datetime.date.today() or \
            debut_affectation < datetime.date.today():
            raise forms.ValidationError("La date limite et/ou la date \
                d'affection doivent être supérieures à la date d'aujourdhui.")
        """
        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', 'html')