X-Git-Url: http://git.auf.org/?p=auf_django_coda.git;a=blobdiff_plain;f=auf%2Fdjango%2Fcoda%2Fmodels.py;h=42d15adc3c6533b9d7c80c0e7efe6e8685f2420f;hp=57b2d5ff92872a792338600d8c2843c66bd0d9c1;hb=d17f22539231066bd7a8123e18cb0c866164fcdd;hpb=d7957f8d389e5e9667af69cf330f133b773787e2

diff --git a/auf/django/coda/models.py b/auf/django/coda/models.py
index 57b2d5f..42d15ad 100755
--- a/auf/django/coda/models.py
+++ b/auf/django/coda/models.py
@@ -3,6 +3,16 @@
 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)
@@ -11,10 +21,18 @@ class Element(models.Model):
     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'
 
+    def __unicode__(self):
+        return self.code
 
 class Ecriture(models.Model):
     id = models.CharField(max_length=25, primary_key=True)