fix #1425

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

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

import datetime
from django.db import models
from datamaster_modeles.models import Pays, Implantation
from dae.managers import SecurityManager

GENRE_CHOICES = (
    ('m', 'Homme'),
    ('f', 'Femme'),
)
SITUATION_CHOICES = (
    ('C', 'Célibataire'),
    ('F', 'Fiancé'),
    ('M', 'Marié'),
)

class Employe(models.Model):   
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    prenom = models.CharField(max_length=255)
    nationalite = models.ForeignKey('datamaster_modeles.Pays', to_field='code', related_name='nationalite', db_column='nationalite')
    date_naissance = models.DateField(null=True, blank=True)
    #Infos personnelles
    genre = models.CharField(max_length=1, choices=GENRE_CHOICES, null=True, blank=True)
    situation_famille = models.CharField(max_length=1, choices=SITUATION_CHOICES, null=True, blank=True)
    date_entree = models.DateField(null=True, blank=True) #devrait pas être là
    #Coordonnées
    tel_domicile = models.CharField(max_length=255, null=True, blank=True)
    tel_cellulaire = models.CharField(max_length=255, null=True, blank=True)
    adresse = models.CharField(max_length=255, null=True, blank=True)
    no_rue = models.CharField(max_length=255, null=True, blank=True)
    ville = models.CharField(max_length=255, null=True, blank=True)
    province = models.CharField(max_length=255, null=True, blank=True)
    code_postal = models.CharField(max_length=255, null=True, blank=True)
    pays = models.ForeignKey('datamaster_modeles.Pays', to_field='code', null=True, blank=True, related_name='pays', db_column='pays')
    #Métas
    date_creation = models.DateField(auto_now_add=True)
    date_maj = models.DateField(auto_now=True)
    commentaire = models.TextField(null=True, blank=True)

    def __unicode__(self):
        return u'%s %s' % (self.prenom, self.nom)


TYPE_DOSSIER_CHOICES = (
    ('2', 'Local'),
    ('1', 'Expatrié'),
)

class DossierManager(models.Manager):
    """
    Chargement de tous les objets FK existants sur chaque QuerySet.
    """
    def get_query_set(self):
        fkeys = (
            'employe',
            'poste1',
            'implantation1',
            'poste2',
            'implantation2',
            'service',
            'responsable',
            'remplacement_de',
            'statut',
            'organisme_bstg',
            'classement',
            'type_contrat',
        )
        return super(DossierManager, self).get_query_set().select_related(*fkeys).all()

class Dossier(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    code = models.CharField(max_length=10, unique=True)
    employe = models.ForeignKey('Employe', db_column='employe')
    #Postes
    poste1 = models.ForeignKey('Poste', db_column='poste1', related_name='poste1')
    implantation1 = models.ForeignKey('datamaster_modeles.Implantation', db_column='implantation1', related_name='implantation1', blank=True, null=True)
    complement1 = models.TextField(null=True, blank=True)
    responsable_implantation1 = models.IntegerField()
    poste2 = models.ForeignKey('Poste', db_column='poste2', related_name='poste2', blank=True, null=True)
    implantation2 = models.ForeignKey('datamaster_modeles.Implantation', db_column='implantation2', related_name='implantation2', null=True, blank=True)
    complement2 = models.TextField(null=True, blank=True)
    responsable_implantation2 = models.IntegerField() 
    #Relations
    service = models.ForeignKey('Service', db_column='service', blank=True, null=True)
    responsable = models.ForeignKey('Employe', db_column='responsable', related_name='responsable', blank=True, null=True)
    remplacement_de = models.ForeignKey('Employe', db_column='remplacement_de', related_name='remplacement_de', blank=True, null=True)
    type = models.CharField(max_length=1, choices=TYPE_DOSSIER_CHOICES)
    statut = models.ForeignKey('Statut', db_column='statut', blank=True, null=True)
    organisme_bstg = models.ForeignKey('OrganismeBstg', db_column='organisme_bstg', blank=True, null=True)
    #Rémunération
    classement = models.ForeignKey('Classement', db_column='classement', blank=True, null=True)
    regime_travail = models.IntegerField()
    #Mandat
    mandat_date_debut = models.DateField()
    mandat_date_fin = models.DateField(null=True, blank=True)
    #Contrat
    contrat_date_debut = models.DateField()
    contrat_date_fin = models.DateField()
    type_contrat = models.ForeignKey('TypeContrat', db_column='type_contrat', blank=True, null=True)
    #Meta
    date_creation = models.DateField(auto_now_add=True)
    date_maj = models.DateField(auto_now=True)
    commentaire = models.TextField(null=True, blank=True) 

    # Managers
    objects = DossierManager()
    
    def __unicode__(self):
        return u'%s : %s %s' % (self.employe, self.poste1, self.complement1)

    def get_salaire_display(self):
        """
        Moyen rapide de récupérer le salaire correspodant à un dossier. Par contre,
        toutes les rémuérations n'ont pas de devise associées, c'est pourquoi on récupère
        les anciennes rémunérations pour rechercher si elle existait auparavant.
        """
        remun = self.remuneration_set.all()
        if len(remun) == 0:
            return ""

        devise = None
        for r in remun:
            try:
                if r.devise_id is not None:
                    devise = Devise.objects.get(id=r.devise_id).code
            except:
                pass
        return "%s %s" % (int(remun[0].montant), devise)

    def get_salaire_euro_display(self):
        """
        Moyen rapide de récupérer le salaire correspodant à un dossier. Par contre,
        toutes les rémuérations n'ont pas de devise associées, c'est pourquoi on récupère
        les anciennes rémunérations pour rechercher si elle existait auparavant.
        La valeur est est affichée en Euros en se servant du taux actuel.
        """
        remun = self.remuneration_set.all()
        if len(remun) == 0:
            return ""

        devise = None
        montant_euros = 0
        for r in remun:
            try:
                if r.devise_id is not None:
                    devise = Devise.objects.get(id=r.devise_id)
                    taux = TauxChange.objects.filter(devise=devise).order_by("annee")[0]
            except:
                pass
        montant_euros = remun[0].montant * taux.taux
        return "%s EUR" % (int(montant_euros))

LIEN_PARENTE_CHOICES = (
    ('Conjoint', 'Conjoint'),
    ('Conjointe', 'Conjointe'),
    ('Fille', 'Fille'),
    ('Fils', 'Fils'),
)

class AyantDroit(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    prenom = models.CharField(max_length=255)
    #Relation
    employe = models.ForeignKey('Employe', db_column='employe', related_name='employe')
    lien_parente = models.CharField(max_length=10, choices=LIEN_PARENTE_CHOICES, null=True, blank=True)
    #Méta
    commentaire = models.TextField(null=True, blank=True) 
    actif = models.BooleanField()
    
    
class Remuneration(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    dossier = models.ForeignKey('Dossier', db_column='dossier')
    type = models.ForeignKey('TypeRemuneration', db_column='type')
    type_revalorisation = models.ForeignKey('TypeRevalorisation', db_column='type_revalorisation', null=True, blank=True)
    montant = models.FloatField(null=True, blank=True)
    devise = models.ForeignKey('Devise', to_field='code', db_column='devise', null=True, blank=True)
    date_effective = models.DateField(null=True, blank=True)
    pourcentage = models.IntegerField(null=True, blank=True)
    #Méta
    date_creation = models.DateField(auto_now_add=True)
    user_creation = models.IntegerField(null=True, blank=True) #User ou employé
    desactivation = models.NullBooleanField(null=True, blank=True) #
    date_desactivation = models.DateField(null=True, blank=True)
    user_desactivation = models.IntegerField(null=True, blank=True) #User ou employé
    annulation = models.NullBooleanField(null=True, blank=True)
    date_annulation = models.DateField(null=True, blank=True)
    user_annulation = models.IntegerField(null=True, blank=True) #User ou employé
    
    def __unicode__(self):
        try:
            devise = self.devise.code
        except:
            devise = "???"
        return "%s %s" % (self.montant, devise)

class FamilleEmploi(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    #Méta
    actif = models.BooleanField()

class TypePoste(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    nom_feminin = models.CharField(max_length=255)
    description = models.CharField(max_length=255)
    is_responsable = models.BooleanField()
    famille_emploi = models.ForeignKey('FamilleEmploi', db_column='famille_emploi')
    #Méta
    date_modification = models.DateField(auto_now=True)
    actif = models.BooleanField()

    def __unicode__(self):
        return u'%s' % self.nom
        
    class Meta:
        ordering = ['nom']


TYPE_PAIEMENT_CHOICES = (
    ('Régulier', 'Régulier'),
    ('Ponctuel', 'Ponctuel'),
)

NATURE_REMUNERATION_CHOICES = (
    ('Accessoire', 'Accessoire'),
    ('Charges', 'Charges'),
    ('Indemnité', 'Indemnité'),
    ('RAS', 'RAS'),
    ('Traitement', 'Traitement'),
)

class TypeRemuneration(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    type_paiement = models.CharField(max_length=30, choices=TYPE_PAIEMENT_CHOICES)
    nature_remuneration = models.CharField(max_length=30, choices=NATURE_REMUNERATION_CHOICES)
    #Méta
    actif = models.BooleanField()

    def __unicode__(self):
        return u'%s' % self.nom


class TypeRevalorisation(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    #Méta
    actif = models.BooleanField()
    
PROPORTION_CHOICES = (
    ('0.5', '0.5'),
    ('1', '1'),
)

class PosteManager(SecurityManager):
    """
    Chargement de tous les objets FK existants sur chaque QuerySet.
    """
    prefixe_implantation = 'implantation'

    def get_query_set(self):
        fkeys = (
            'implantation',
            'type_poste',
        )
        return super(PosteManager, self).get_query_set().select_related(*fkeys).all()

class Poste(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    implantation = models.ForeignKey('datamaster_modeles.Implantation',
            db_column='implantation', related_name='+')
    type_poste = models.ForeignKey('TypePoste', db_column='type_poste')
    proportion = models.CharField(max_length=10, choices=PROPORTION_CHOICES)
    #(sert à quoi?) renommer "regime_travail" ou autre?    convertir data en % (data * 100; ex: 1 = 100%)
    #Méta
    date_modification = models.DateField(auto_now=True)
    actif = models.BooleanField()

    # Managers
    objects = PosteManager()

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


class Service(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    #Méta
    actif = models.BooleanField()

    def __unicode__(self):
        return u'%s' % self.nom
        
    class Meta:
        ordering = ['nom']


TYPE_ORGANISME_CHOICES = (
    ('MAD', 'Mise à disposition'),
    ('DET', 'Détachement'),
)

class OrganismeBstg(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    type = models.CharField(max_length=10, choices=TYPE_ORGANISME_CHOICES)
    #Méta
    actif = models.BooleanField()

    def __unicode__(self):
        return u'%s (%s)' % (self.nom, self.type)

    class Meta:
        ordering = ['type', 'nom']


CONTRAT_CATEGORIE_CHOICES= (
    ('A', 'A'),
    ('C', 'C'),
)
class Statut(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    code = models.CharField(max_length=25, unique=True)
    nom = models.CharField(max_length=255)
    type_contrat_categorie = models.CharField(max_length=10, choices=CONTRAT_CATEGORIE_CHOICES)
    #CHOICES A, C (veut dire quoi?) voir TypeContrat.categorie
    #Méta
    actif = models.BooleanField()

    def __unicode__(self):
        return u'%s : %s' % (self.code, self.nom)

TYPE_CLASSEMENT_CHOICES = (
    ('S', 'S'),
    ('T', 'T'),
)
class Classement(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    type = models.CharField(max_length=10, choices=TYPE_CLASSEMENT_CHOICES)
    echelon = models.IntegerField()
    degre = models.IntegerField()
    coefficient = models.FloatField()
    #Méta
    commentaire = models.TextField(null=True, blank=True) 
    date_modification = models.DateField(auto_now=True)
    actif = models.BooleanField()

    def __unicode__(self):
        return u'%s.%s.%s (%s)' % (self.type, self.echelon, self.degre,
                                   self.coefficient)
    class Meta:
        ordering = ['type','echelon','degre','coefficient']

class TauxChange(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    devise = models.ForeignKey('Devise', db_column='devise')
    annee = models.IntegerField()
    taux = models.FloatField()
    #Relations
    implantation = models.ForeignKey('datamaster_modeles.Implantation', db_column='implantation')

class ValeurPointManager(models.Manager):
    """
    Manager qui travaille uniquement sur les valeurs du point de l'année en cours.
    """
    mois = datetime.datetime.now().month
    annee_courante = datetime.datetime.now().year

    # Pour le mois de janvier et décembre on mets les 2 années pour faire la transition
    if mois == 1:
        filtre_annee = (annee_courante-1, annee_courante)
    elif mois == 12:
        filtre_annee = (annee_courante, annee_courante+1)
    else:
        filtre_annee = (annee_courante,)

    def get_query_set(self):
        return super(ValeurPointManager, self).get_query_set().select_related('implantation').filter(annee__in=self.filtre_annee)


class ValeurPoint(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    valeur = models.FloatField()
    implantation = models.ForeignKey('datamaster_modeles.Implantation', db_column='implantation')
    #Méta
    annee = models.IntegerField()

    # Stockage de tous les taux de change pour optimiser la recherche de la devise associée
    annee_courante = datetime.datetime.now().year
    tauxchange = TauxChange.objects.select_related('devise').filter(annee=annee_courante)
    
    def get_tauxchange_courant(self):
        """
        Recherche le taux courant associé à la valeur d'un point.
        Tous les taux de l'année courante sont chargés, pour optimiser un affichage en liste.
        (On pourrait probablement améliorer le manager pour lui greffer le taux courant sous forme de JOIN)
        """
        for tauxchange in self.tauxchange:
            if tauxchange.implantation_id == self.implantation_id:
                return tauxchange
        return None

    def __unicode__(self):
        tx = self.get_tauxchange_courant()
        if tx:
            devise_code = tx.devise.code
        else:
            devise_code = "??"
        return u'%s %s (%s-%s)' % (self.valeur, devise_code, self.implantation_id, self.annee)
        
    class Meta:
        ordering = ['valeur']

    objects = models.Manager()
    actuelles = ValeurPointManager()


class Devise(models.Model):
    id = models.IntegerField(primary_key=True)
    code =  models.CharField(max_length=10, unique=True)
    nom = models.CharField(max_length=255)

    def __unicode__(self):
        return u'%s - %s' % (self.code, self.nom)


class TypeContrat(models.Model):
    #Identification
    id = models.IntegerField(primary_key=True)
    nom = models.CharField(max_length=255)
    nom_long = models.CharField(max_length=255) #description
    categorie = models.CharField(max_length=10, choices=CONTRAT_CATEGORIE_CHOICES)
    #Méta
    actif = models.BooleanField()

    def __unicode__(self):
        return u'%s' % (self.nom)