group as cron

[auf_rh_dae.git] / project / rh / management / commands / rh.py

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

from django.core.management.base import BaseCommand, CommandError
from project.rh.models import Poste, Employe


def flag_vacant():
    """
    Mettre a jour le booleen de poste pour savoir si le poste
    est vacant.
    """
    for p in Poste.objects.all():
        test = p.vacant
        employes = p.occupe_par()
        if len(employes) > 0:
            vacant = False
        else:
            vacant = True

        if p.actif is False:
            vacant = None

        if test != vacant:
            p.vacant = vacant
            p.save()       

def flag_nb_postes():
    """
    """
    for e in Employe.objects.all():
        total = 0
        test = e.nb_postes
        for d in e.rh_dossiers.all():
            if d.date_fin is None:
                total +=1
        if test != total:
            e.nb_postes = total
            e.save()       


class Command(BaseCommand):

    def handle(self, *args, **options):

        if args[0] == "cron":
            flag_vacant()            
            flag_nb_postes()            

        if args[0] == "vacant":
            flag_vacant()            
            self.stdout.write("Flag vacant mis à jour\n")

        if args[0] == "nb_postes":
            flag_nb_postes()            
            self.stdout.write("Flag nb_postes mis à jour\n")