def edit_category(request, category_id):
category = get_object_or_404(Category, pk = category_id)
if request.method == 'POST':
form = CategoryForm(request.POST, instance = category)
if form.is_valid():
form.save()
return HttpResponseRedirect(reverse('lists:category_index'))
else:
form = CategoryForm(instance = category)
return render(request, 'categories/edit.html', {'edit_category': category, 'form': form})
def new_category(request):
if request.method == 'POST':
form = CategoryForm(request.POST)
if form.is_valid():
cd = form.cleaned_data
Category.objects.create_category(cd['name'])
return HttpResponseRedirect(reverse('lists:category_index'))
else:
form = CategoryForm()
return render(request, 'categories/new.html', {'form': form})
