todolists/views.py

   1 from django.http import HttpResponse, HttpResponseRedirect
   2 from django.shortcuts import get_object_or_404
   3 from django.contrib.auth.decorators import login_required, user_passes_test
   4 from django.contrib.auth.models import User
   5 from archlinux.utils import render_template
   6 from archlinux.todolists.models import Todolist, TodolistPkg
   7 from archlinux.packages.models import Package
   8
   9 # FIXME: ugly hackery. http://code.djangoproject.com/ticket/3450
  10 import django.db
  11 IntegrityError = django.db.backend.Database.IntegrityError
  12
  13 @login_required
  14 #@is_maintainer
  15 def flag(request, listid, pkgid):
  16         list = get_object_or_404(Todolist, id=listid)
  17         pkg  = get_object_or_404(TodolistPkg, id=pkgid)
  18         pkg.complete = not pkg.complete
  19         pkg.save()
  20         return HttpResponseRedirect('/todo/%s/' % (listid))
  21
  22 @login_required
  23 def view(request, listid):
  24         list = get_object_or_404(Todolist, id=listid)
  25         pkgs = TodolistPkg.objects.filter(list=list.id).order_by('pkg')
  26         return render_template('todolists/view.html', request, {'list':list,'pkgs':pkgs})
  27
  28 @login_required
  29 def list(request):
  30         lists = Todolist.objects.order_by('-date_added')
  31         for l in lists:
  32                 l.complete = TodolistPkg.objects.filter(list=l.id,complete=False).count() == 0
  33         return render_template('todolists/list.html', request, {'lists':lists})
  34
  35 @login_required
  36 #@is_maintainer
  37 @user_passes_test(lambda u: u.has_perm('todolists.add_todolist'))
  38 def add(request):
  39         if request.POST:
  40                 try:
  41                         m = User.objects.get(username=request.user.username)
  42                 except User.DoesNotExist:
  43                         return render_template('error_page.html', request,
  44                                 {'errmsg': 'Cannot find a maintainer record for you!'})
  45                 # create the list
  46                 todo = Todolist(
  47                         creator     = m,
  48                         name        = request.POST.get('name'),
  49                         description = request.POST.get('description'))
  50                 todo.save()
  51                 # now link in packages
  52                 for p in request.POST.get('packages').split("\n"):
  53                         for pkg in Package.objects.filter(pkgname=p.strip()):
  54                                 todopkg = TodolistPkg(
  55                                         list = todo,
  56                                         pkg  = pkg)
  57                                 try:
  58                                         todopkg.save()
  59                                 except IntegrityError, (num, desc):
  60                                         if num == 1062: # duplicate entry aka dupe package on list
  61                                                 pass
  62                 return HttpResponseRedirect('/todo/')
  63
  64         return render_template('todolists/add.html', request)