[#1862] Ajout de managers et d'une représentation unicode à Element
authorEric Mc Sween <eric.mcsween@auf.org>
Wed, 5 Oct 2011 21:10:43 +0000 (17:10 -0400)
committerEric Mc Sween <eric.mcsween@auf.org>
Wed, 5 Oct 2011 21:10:43 +0000 (17:10 -0400)
auf/django/coda/models.py
setup.py

index 57b2d5f..42d15ad 100755 (executable)
@@ -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)
index 8ee645a..c11900e 100644 (file)
--- a/setup.py
+++ b/setup.py
@@ -5,7 +5,7 @@ import sys
 from setuptools import setup, find_packages
 
 name = 'auf.django.coda'
-version = '0.3'
+version = '0.4'
 
 setup(name=name,
       version=version,