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 archweb_dev.utils import render_template from archweb_dev.todolists.models import Todolist, TodolistPkg from archweb_dev.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)