todolists/models.py

   1 from django.db import models
   2 from django.contrib.auth.models import User
   3 from archlinux.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