1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
from django.contrib.auth.models import User
from django.contrib.sites.models import Site
from django.db import models
from django.db.models import Q
from django.db.models.signals import pre_save
from main.models import Arch, Repo, Package
from main.utils import set_created_field
class TodolistManager(models.Manager):
def incomplete(self):
not_done = (Q(todolistpackage__status=TodolistPackage.INCOMPLETE) |
Q(todolistpackage__status=TodolistPackage.IN_PROGRESS))
return self.order_by().filter(not_done).distinct()
class Todolist(models.Model):
slug = models.SlugField(max_length=255, unique=True)
old_id = models.IntegerField(null=True, unique=True)
name = models.CharField(max_length=255)
description = models.TextField()
creator = models.ForeignKey(User, on_delete=models.PROTECT,
related_name="created_todolists")
created = models.DateTimeField(db_index=True)
last_modified = models.DateTimeField(editable=False)
raw = models.TextField(blank=True)
objects = TodolistManager()
class Meta:
get_latest_by = 'created'
def __unicode__(self):
return self.name
def get_absolute_url(self):
return '/todo/%s/' % self.slug
def get_full_url(self, proto='https'):
'''get a URL suitable for things like email including the domain'''
domain = Site.objects.get_current().domain
return '%s://%s%s' % (proto, domain, self.get_absolute_url())
def packages(self):
if not hasattr(self, '_packages'):
self._packages = self.todolistpackage_set.select_related(
'pkg', 'repo', 'arch').order_by('pkgname', 'arch')
return self._packages
class TodolistPackage(models.Model):
INCOMPLETE = 0
COMPLETE = 1
IN_PROGRESS = 2
STATUS_CHOICES = (
(INCOMPLETE, 'Incomplete'),
(COMPLETE, 'Complete'),
(IN_PROGRESS, 'In-progress'),
)
todolist = models.ForeignKey(Todolist)
pkg = models.ForeignKey(Package, null=True, on_delete=models.SET_NULL)
pkgname = models.CharField(max_length=255)
pkgbase = models.CharField(max_length=255)
arch = models.ForeignKey(Arch)
repo = models.ForeignKey(Repo)
created = models.DateTimeField()
status = models.SmallIntegerField(default=0, choices=STATUS_CHOICES)
user = models.ForeignKey(User, null=True, on_delete=models.SET_NULL)
comments = models.TextField(null=True, blank=True)
class Meta:
unique_together = (('todolist','pkgname', 'arch'),)
def __unicode__(self):
return self.pkgname
def status_css_class(self):
return self.get_status_display().lower().replace('-', '')
pre_save.connect(set_created_field, sender=Todolist,
dispatch_uid="todolists.models")
pre_save.connect(set_created_field, sender=TodolistPackage,
dispatch_uid="todolists.models")
# vim: set ts=4 sw=4 et:
|
