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 archweb_dev.lib.utils import render_response
   6 from archweb_dev.todolists.models import Todolist, TodolistPkg
   7 from archweb_dev.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_response(request, 'todolists/view.html', {'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_response(request, 'todolists/list.html', {'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_response(request, 'error_page.html',
  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     return render_response(request, 'todolists/add.html')
  64
  65 # vim: set ts=4 sw=4 et:
  66