diff options
Diffstat (limited to 'todolists')
| -rw-r--r-- | todolists/views.py | 65 |
1 files changed, 41 insertions, 24 deletions
diff --git a/todolists/views.py b/todolists/views.py index 82f2d94a..d76c24b3 100644 --- a/todolists/views.py +++ b/todolists/views.py @@ -1,6 +1,7 @@ import django.newforms as forms from django.http import HttpResponse, HttpResponseRedirect +from django.template import RequestContext from django.shortcuts import get_object_or_404 from django.contrib.auth.decorators import permission_required from django.contrib.auth.models import User @@ -12,6 +13,25 @@ from archweb_dev.main.models import Arch, Repo import django.db IntegrityError = django.db.backend.Database.IntegrityError +class TodoListForm(forms.Form): + name = forms.CharField(max_length=255, + widget=forms.TextInput(attrs={'size': '30'})) + description = forms.CharField(required=False, + widget=forms.Textarea(attrs={'rows': '4', 'cols': '60'})) + packages = forms.CharField(required=False, + help_text='(one per line)', + widget=forms.Textarea(attrs={'rows': '20', 'cols': '60'})) + + def clean_packages(self): + packages = [] + for p in self.clean_data['packages'].split("\n"): + for pkg in Package.objects.filter( + pkgname=p.strip()).order_by('arch').distinct(): + packages .append(pkg) + + return packages + + def flag(request, listid, pkgid): list = get_object_or_404(Todolist, id=listid) pkg = get_object_or_404(TodolistPkg, id=pkgid) @@ -37,31 +57,28 @@ def list(request): @permission_required('todolists.add_todolist') def add(request): if request.POST: - try: - m = User.objects.get(username=request.user.username) - except User.DoesNotExist: - return render_response(request, 'error_page.html', - {'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()).order_by('arch'): - todopkg = TodolistPkg( - list = todo, - pkg = pkg) - try: - todopkg.save() - except IntegrityError, (num, desc): - if num != 1062: # duplicate entry aka dupe package on list - raise - return HttpResponseRedirect('/todo/') - return render_response(request, 'todolists/add.html') + form = TodoListForm(request.POST) + if form.is_valid(): + todo = Todolist( + creator = request.user, + name = form.clean_data['name'], + description = form.clean_data['description']) + todo.save() + # now link in packages + for pkg in form.clean_data['packages']: + todopkg = TodolistPkg(list = todo, pkg = pkg) + todopkg.save() + return HttpResponseRedirect('/todo/') + else: + form = TodoListForm() + + page_dict = { + 'title': 'Add To-do List', + 'form': form, + 'submit_text': 'Create List' + } + return render_response(request, 'general_form.html', page_dict) # vim: set ts=4 sw=4 et: |