[#1862] Ajout de managers et d'une représentation unicode à Element
[auf_django_coda.git] / auf / django / coda / models.py
index 57b2d5f..42d15ad 100755 (executable)
@@ -3,6 +3,16 @@
 from django.conf import settings
 from django.db import models
 
 from django.conf import settings
 from django.db import models
 
+class ElementManager(models.Manager):
+
+    def __init__(self, niveau):
+        super(ElementManager, self).__init__()
+        self.niveau = niveau
+
+    def get_query_set(self):
+        qs = super(ElementManager, self).get_query_set()
+        return qs.filter(niveau=self.niveau)
+
 
 class Element(models.Model):
     id = models.CharField(max_length=74, primary_key=True)
 
 class Element(models.Model):
     id = models.CharField(max_length=74, primary_key=True)
@@ -11,10 +21,18 @@ class Element(models.Model):
     date_modification = models.DateField()
     nom = models.CharField(max_length=36)
 
     date_modification = models.DateField()
     nom = models.CharField(max_length=36)
 
+    # Managers
+    objects = models.Manager()
+    comptes_comptables = ElementManager(1)
+    tiers_operations = ElementManager(2)
+    projets_postes = ElementManager(3)
+
     class Meta:
         managed = False
         db_table = 'coda_element'
 
     class Meta:
         managed = False
         db_table = 'coda_element'
 
+    def __unicode__(self):
+        return self.code
 
 class Ecriture(models.Model):
     id = models.CharField(max_length=25, primary_key=True)
 
 class Ecriture(models.Model):
     id = models.CharField(max_length=25, primary_key=True)