todolists/models.py

   1 from django.db import models
   2 from django.contrib.auth.models import User
   3 from archweb_dev.packages.models import Package
   4
   5 class TodolistManager(models.Manager):
   6     def get_incomplete(self):
   7         results = []
   8         for l in self.all().order_by('-date_added'):
   9             if TodolistPkg.objects.filter(list=l.id).filter(complete=False).count() > 0:
  10                 results.append(l)
  11         return results
  12
  13 class Todolist(models.Model):
  14     id = models.AutoField(primary_key=True)
  15     creator = models.ForeignKey(User)
  16     name = models.CharField(maxlength=255)
  17     description = models.TextField()
  18     date_added = models.DateField(auto_now_add=True)
  19     objects = TodolistManager()
  20     class Meta:
  21         db_table = 'todolists'
  22
  23 class TodolistPkg(models.Model):
  24     id = models.AutoField(primary_key=True)
  25     list = models.ForeignKey(Todolist)
  26     pkg = models.ForeignKey(Package)
  27     complete = models.BooleanField(default=False)
  28     class Meta:
  29         db_table = 'todolists_pkgs'
  30         unique_together = (('list','pkg'),)
  31
  32 # vim: set ts=4 sw=4 et:
  33