poste import + liens dans RH
[auf_rh_dae.git] / project / dae / managers.py
index 5ae8204..dabe062 100644 (file)
@@ -3,7 +3,8 @@
 from django.db import models
 from django.db.models import Q
 from utils import get_employe_from_user
-from workflow import MAP_GROUPE_ETATS_A_FAIRE, DOSSIER_ETAT_FINALISE
+from workflow import MAP_GROUPE_ETATS_A_FAIRE, DOSSIER_ETAT_FINALISE, \
+                        POSTE_ETAT_FINALISE
 from workflow import grp_drh, \
                      grp_drh2, \
                      grp_accior, \
@@ -60,9 +61,23 @@ class DossierFinaliseManager(models.Manager):
     def get_query_set(self):
         from dae.models import ImportDossier
         dae_importees = [str(i.dae_id) for i in ImportDossier.objects.all()]
-        qs = super(DossierFinaliseManager,
-                self).get_query_set().filter(etat=DOSSIER_ETAT_FINALISE,).extra(where=['id\
-                    NOT IN (%s)' % ", ".join(dae_importees)])
+        qs = super(DossierFinaliseManager, self).get_query_set().filter(etat=DOSSIER_ETAT_FINALISE,)
+        if len(dae_importees) > 0:
+            qs = qs.extra(where=['id NOT IN (%s)' % ", ".join(dae_importees)])
+        return qs
+
+
+class PosteFinaliseManager(models.Manager):
+    """
+    Dossiers finalisés, qui ne sont pas encore importés
+    """
+    def get_query_set(self):
+        from dae.models import ImportPoste
+        dae_importees = [str(i.dae_id) for i in ImportPoste.objects.all()]
+
+        qs = super(PosteFinaliseManager, self).get_query_set().filter(etat=POSTE_ETAT_FINALISE,)
+        if len(dae_importees) > 0:
+            qs = qs.extra(where=['id NOT IN (%s)' % ", ".join(dae_importees)])
         return qs