3 files changed, 95 insertions, 0 deletions
diff --git a/todolists/__init__.py b/todolists/__init__.py
new file mode 100644
index 00000000..e69de29b
--- /dev/null
+++ b/todolists/__init__.py
diff --git a/todolists/models.py b/todolists/models.py
new file mode 100644
index 00000000..0a4f445e
--- /dev/null
+++ b/todolists/models.py
@@ -0,0 +1,31 @@
+from django.db import models
+from django.contrib.auth.models import User
+from archlinux.packages.models import Package
+
+class TodolistManager(models.Manager):
+	def get_incomplete(self):
+		results = []
+		for l in self.all().order_by('-date_added'):
+			if TodolistPkg.objects.filter(list=l.id).filter(complete=False).count() > 0:
+				results.append(l)
+		return results
+
+class Todolist(models.Model):
+	id = models.AutoField(primary_key=True)
+	creator = models.ForeignKey(User)
+	name = models.CharField(maxlength=255)
+	description = models.TextField()
+	date_added = models.DateField(auto_now_add=True)
+	objects = TodolistManager()
+	class Meta:
+		db_table = 'todolists'
+
+class TodolistPkg(models.Model):
+	id = models.AutoField(primary_key=True)
+	list = models.ForeignKey(Todolist)
+	pkg = models.ForeignKey(Package)
+	complete = models.BooleanField(default=False)
+	class Meta:
+		db_table = 'todolists_pkgs'
+		unique_together = (('list','pkg'),)
+
diff --git a/todolists/views.py b/todolists/views.py
new file mode 100644
index 00000000..e364f36e
--- /dev/null
+++ b/todolists/views.py
@@ -0,0 +1,64 @@
+from django.http import HttpResponse, HttpResponseRedirect
+from django.shortcuts import get_object_or_404
+from django.contrib.auth.decorators import login_required, user_passes_test
+from django.contrib.auth.models import User
+from archlinux.utils import render_template
+from archlinux.todolists.models import Todolist, TodolistPkg
+from archlinux.packages.models import Package
+
+# FIXME: ugly hackery. http://code.djangoproject.com/ticket/3450
+import django.db
+IntegrityError = django.db.backend.Database.IntegrityError
+
+@login_required
+#@is_maintainer
+def flag(request, listid, pkgid):
+	list = get_object_or_404(Todolist, id=listid)
+	pkg  = get_object_or_404(TodolistPkg, id=pkgid)
+	pkg.complete = not pkg.complete
+	pkg.save()
+	return HttpResponseRedirect('/todo/%s/' % (listid))
+
+@login_required
+def view(request, listid):
+	list = get_object_or_404(Todolist, id=listid)
+	pkgs = TodolistPkg.objects.filter(list=list.id).order_by('pkg')
+	return render_template('todolists/view.html', request, {'list':list,'pkgs':pkgs})
+
+@login_required
+def list(request):
+	lists = Todolist.objects.order_by('-date_added')
+	for l in lists:
+		l.complete = TodolistPkg.objects.filter(list=l.id,complete=False).count() == 0
+	return render_template('todolists/list.html', request, {'lists':lists})
+
+@login_required
+#@is_maintainer
+@user_passes_test(lambda u: u.has_perm('todolists.add_todolist'))
+def add(request):
+	if request.POST:
+		try:
+			m = User.objects.get(username=request.user.username)
+		except User.DoesNotExist:
+			return render_template('error_page.html', request,
+				{'errmsg': 'Cannot find a maintainer record for you!'})
+		# create the list
+		todo = Todolist(
+			creator     = m,
+			name        = request.POST.get('name'),
+			description = request.POST.get('description'))
+		todo.save()
+		# now link in packages
+		for p in request.POST.get('packages').split("\n"):
+			for pkg in Package.objects.filter(pkgname=p.strip()):
+				todopkg = TodolistPkg(
+					list = todo,
+					pkg  = pkg)
+				try:
+					todopkg.save()
+				except IntegrityError, (num, desc):
+					if num == 1062: # duplicate entry aka dupe package on list
+						pass
+		return HttpResponseRedirect('/todo/')
+
+	return render_template('todolists/add.html', request)