filtre nb postes

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

diff --git a/project/rh/management/commands/rh.py b/project/rh/management/commands/rh.py
index c5e6206..47e708d 100644 (file)
--- a/project/rh/management/commands/rh.py
+++ b/project/rh/management/commands/rh.py
@@ -1,7 +1,7 @@
 # -*- encoding: utf-8 -*-
 
 from django.core.management.base import BaseCommand, CommandError
-from project.rh.models import Poste
+from project.rh.models import Poste, Employe
 
 
 def flag_vacant():
@@ -24,10 +24,28 @@ def flag_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] == "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")