1 files changed, 64 insertions, 0 deletions
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)