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' })
