def category_edit(request, category_id): c = None if category_id: c = get_object_or_404(Category, pk=category_id) if request.method == 'POST': form = CategoryForm(request.POST, instance=c) if form.is_valid(): form.save() if '_addanother' in request.POST: return HttpResponseRedirect(reverse('category_add')) else: return HttpResponseRedirect(reverse('categories')) else: form = CategoryForm(instance=c) return render(request, 'category_form.html', { 'form': form, 'action': 'edit' })
def category_add(request): if request.method == 'POST': form = CategoryForm(request.POST) if form.is_valid(): form.save() if '_addanother' in request.POST: form = CategoryForm() else: return HttpResponseRedirect(reverse('categories')) else: form = CategoryForm() return render(request, 'category_form.html', { 'form': form, 'action': 'add' })