Merge branch 'dae' into dev
[auf_rh_dae.git] / project / rh_v1 / models.py
index 54a9a39..f0424ca 100644 (file)
@@ -47,6 +47,27 @@ TYPE_DOSSIER_CHOICES = (
     ('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)
@@ -54,35 +75,38 @@ class Dossier(models.Model):
     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')
+    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')
+    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')
-    responsable = models.ForeignKey('Employe', db_column='responsable', related_name='responsable')
-    remplacement_de = models.ForeignKey('Employe', db_column='remplacement_de', related_name='remplacement_de')
+    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')
-    organisme_bstg = models.ForeignKey('OrganismeBstg', db_column='organisme_bstg')
+    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')
+    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()
+    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')
+    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()
+
 LIEN_PARENTE_CHOICES = (
     ('Conjoint', 'Conjoint'),
     ('Conjointe', 'Conjointe'),
@@ -144,6 +168,9 @@ class TypePoste(models.Model):
 
     def __unicode__(self):
         return u'%s' % self.nom
+        
+    class Meta:
+        ordering = ['nom']
 
 
 TYPE_PAIEMENT_CHOICES = (
@@ -180,6 +207,17 @@ PROPORTION_CHOICES = (
     ('1', '1'),
 )
 
+class PosteManager(models.Manager):
+    """
+    Chargement de tous les objets FK existants sur chaque QuerySet.
+    """
+    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)
@@ -192,8 +230,11 @@ class Poste(models.Model):
     date_modification = models.DateField(auto_now=True)
     actif = models.BooleanField()
 
+    # Managers
+    objects = PosteManager()
+
     def __unicode__(self):
-        return u'%s - %s' % (self.implantation, self.type_poste.nom)
+        return u'%s - %s [%s]' % (self.implantation, self.type_poste.nom, self.id)
 
 
 class Service(models.Model):
@@ -205,6 +246,9 @@ class Service(models.Model):
 
     def __unicode__(self):
         return u'%s' % self.nom
+        
+    class Meta:
+        ordering = ['nom']
 
 
 TYPE_ORGANISME_CHOICES = (
@@ -272,6 +316,9 @@ class ValeurPoint(models.Model):
 
     def __unicode__(self):
         return u'%s (%s-%s)' % (self.valeur, self.implantation_id, self.annee)
+        
+    class Meta:
+        ordering = ['valeur']
 
 
 class TauxChange(models.Model):