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)