def new_collection(request):
if request.method == 'POST':
form = CollectionForm(request.POST)
if form.is_valid():
new_collection = form.save(commit=False)
new_collection.user = request.user
new_collection.save()
return redirect('dashboard')
else:
form = CollectionForm()
return render(request, 'collect/new_collection.html', {
'form': form,
})
def edit_collection(request, collection_id):
c = get_object_or_404(Collection, pk=collection_id, user=request.user)
if request.method == 'POST':
form = CollectionForm(request.POST, instance=c)
if form.is_valid():
new_collection = form.save(commit=False)
new_collection.save()
return redirect('dashboard')
else:
form = CollectionForm(instance=c)
return render(request, 'collect/edit_collection.html', {
'collection': c,
'form': form,
})
