def edit_category(request, category_id): category = get_object_or_404(Category, pk=category_id) if category.owner != request.user: return render_to_response('gallery_error.html', {'message' : _("You don't have enough privileges to edit this category")}, context_instance=RequestContext(request)) if category.name == category.owner.username: return render_to_response('gallery_error.html', {'message' : _("You can't edit this category")}, context_instance=RequestContext(request)) if request.method == 'GET': return render_to_response('edit_category.html', {'form' : CategoryForm(instance=category), 'category_id' : category_id}, context_instance=RequestContext(request)) form = CategoryForm(request.POST) if form.is_valid(): category_data = form.cleaned_data category.category_parent = category_data['category_parent'] category.name = category_data['name'] category.save() return redirect('/') return render_to_response('edit_category.html', {'form' : form, 'category_id' : category_id}, context_instance=RequestContext(request))
def add_category(request): if request.method == 'GET': return render_to_response('add_category.html', {'form' : CategoryForm(user=request.user)}, context_instance=RequestContext(request)) form = CategoryForm(request.POST) if form.is_valid(): category_data = form.cleaned_data category_data['owner'] = request.user category = Category(**category_data) category.save() return redirect('/') return render_to_response('add_category.html', {'form' : form}, context_instance=RequestContext(request))