PEP8

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

# -=- encoding: utf-8 -=-

from django.contrib.auth.models import User
from tinymce import models as tinymce_models
from django.db import models
from django.db.models.signals import pre_save, pre_delete
from django.dispatch import receiver


from south.modelsinspector import add_introspection_rules
add_introspection_rules([], ["^tinymce.models.HTMLField"])

from auf.django.emploi import models as emploi
from recrutement.workflow import  grp_evaluateurs_recrutement

### CONSTANTES
#NOTES
NOTE_MIN = 1
NOTE_RANGE = 1
NOTE_MAX = 11
NOTES = [(i, i) for i in range(NOTE_MIN, NOTE_MAX, NOTE_RANGE)]

#HELP_TEXT
HELP_TEXT_NB_DEPENDANT = "Le nombre de personnes à charge"
HELP_TEXT_TAGS_ACCEPTES = "Pour le texte, les variables disponibles sont : \
                            {{ nom_candidat }} {{ prenom_candidat }} \
                            {{ offre_emploi }} et {{ genre_candidat }} \
                            (Pour Monsieur/Madame). Ces champs seront \
                            automatiquement remplacés par les informations de \
                            chaque candidat."


# Abstracts
class Metadata(models.Model):
    """
    Méta-données AUF.
    Metadata.actif = flag remplaçant la suppression.
    actif == False : objet réputé supprimé.
    """
    actif = models.BooleanField(default=True)
    date_creation = models.DateField(auto_now_add=True, )

    class Meta:
        abstract = True


class Candidat(emploi.Candidat):
    class Meta:
        proxy = True

    def moyenne_notes(self):
        evaluations = CandidatEvaluation.objects.filter(candidat=self)
        notes = [evaluation.note for evaluation in evaluations.all() \
                    if evaluation.note is not None]

        if len(notes) > 0:
            moyenne_votes = float(sum(notes)) / len(notes)
        else:
            moyenne_votes = "Non disponible"
        return moyenne_votes


class OffreEmploi(emploi.OffreEmploi):
    class Meta:
        proxy = True


class CandidatPiece(emploi.CandidatPiece):
    class Meta:
        proxy = True


class OffreEmploiManager(models.Manager):
    def get_query_set(self):
        fkeys = ('region',)
        return super(OffreEmploiManager, self).get_query_set().\
                    select_related(*fkeys).all()


class ProxyOffreEmploi(emploi.OffreEmploi):
    class Meta:
        proxy = True
        verbose_name = u"Offre d'emploi (visualisation)"
        verbose_name_plural = u"Offres d'emploi (visualisation)"

    def __unicode__(self):
        return '%s [%s] - View' % (self.nom, self.id)


class ProxyCandidat(emploi.Candidat):
    class Meta:
        proxy = True
        verbose_name = u"Candidat (visualisation)"
        verbose_name_plural = u"Candidats (visualisation)"

    def __unicode__(self):
        return '%s %s [%s]' % (self.prenom, self.nom, self.id)


class Evaluateur(models.Model):
    user = models.ForeignKey(User, unique=True, verbose_name=u"Évaluateur")

    class Meta:
        verbose_name = u"évaluateur"

    def __unicode__(self):
        return '%s %s' % (self.user.first_name, self.user.last_name)


# Synchro des objets évaluateurs pour mettre les users Django dans le groupe
# évaluateurs, afin d'y mettre des permissions
@receiver(pre_save, sender=Evaluateur)
def sync_add_groupe_evaluateur(sender, **kwargs):
    instance = kwargs['instance']
    if grp_evaluateurs_recrutement not in instance.user.groups.all():
        instance.user.groups.add(grp_evaluateurs_recrutement)
        instance.user.save()


@receiver(pre_delete, sender=Evaluateur)
def sync_delete_groupe_evaluateur(sender, **kwargs):
    instance = kwargs['instance']
    instance.user.groups.remove(grp_evaluateurs_recrutement)
    instance.user.save()


class CandidatEvaluation(models.Model):
    candidat = models.ForeignKey(emploi.Candidat, db_column='candidat',
                related_name='evaluations',)
    evaluateur = models.ForeignKey(Evaluateur, db_column='evaluateur',
                    related_name='+', verbose_name=u'Évaluateur')
    note = models.IntegerField(choices=NOTES, blank=True, null=True)
    commentaire = models.TextField(null=True, blank=True, default='Aucun')
    date = models.DateField(auto_now_add=True,)

    class Meta:
        verbose_name = u'évaluation du candidat'
        verbose_name_plural = u'évaluations des candidats'

    def __unicode__(self):
        return u"Évaluation de %s" % self.candidat

    def get_appreciation(self,):
        if self.note is None:
            return u"non disponible"
        return u"%s %s" % (self.note, self.commentaire,)


class MesCandidatEvaluation(CandidatEvaluation):

    class Meta:
        proxy = True
        verbose_name = u"Mon candidat à évaluer"
        verbose_name_plural = u"Mes candidats à évaluer"


#### TEMPLATE COURRIEL
TEMPLATE_CHOICES = (
    ('SEL', 'Sélectionné'),
    ('REF', 'Refusé'),
)


class CourrielTemplate(models.Model):
    nom_modele = models.CharField(u'nom du modèle', max_length=100)
    sujet = models.CharField(u'sujet du courriel', max_length=100)
    plain_text = models.TextField(u'texte', help_text=HELP_TEXT_TAGS_ACCEPTES)

    def __unicode__(self):
        return self.nom_modele

    class Meta:
        ordering = ('nom_modele',)
        verbose_name = "Modèle de courriel"
        verbose_name_plural = "Modèles de courriel"


class CandidatCourriel(models.Model):
    candidats = models.ManyToManyField(Candidat, verbose_name=u"Candidats")
    template = models.ForeignKey(
        CourrielTemplate, db_column='template', related_name='+',
        verbose_name=u"Modèle de courriel"
    )
    sujet = models.CharField(
        max_length=255, blank=True, help_text=HELP_TEXT_TAGS_ACCEPTES
    )
    plain_text = models.TextField(
        u'texte', blank=True, help_text=HELP_TEXT_TAGS_ACCEPTES
    )
    html = tinymce_models.HTMLField(
        verbose_name=u'texte en HTML', null=True, blank=True,
        help_text=HELP_TEXT_TAGS_ACCEPTES
    )

    def __unicode__(self):
        return self.titre

    class Meta:
        verbose_name = u"modèle de courriel"
        verbose_name_plural = u"modèles de courriel"