def create_album_view(request, user_name):
if request.method == 'POST':
form = AlbumForm(request.POST)
if form.is_valid():
album = form.save(commit=False)
album.owner = request.user
album.save()
album.title = form.cleaned_data.get('title')
photos = form.cleaned_data.get('photos')
for photo in photos:
album.photos.add(photo)
album.save()
return redirect(album_view, request.user.username, album.pk)
else:
context = {'album_form': NewAlbumForm()}
return render(request, 'create_album.html', context)
def edit_album_view(request, user_name, album_id):
try:
album = Album.objects.get(pk=album_id)
except Album.AlbumNotFound:
raise Http404
if request.method == 'POST':
form = AlbumForm(request.POST, instance=album)
if form.is_valid():
update_album = form.save(commit=False)
update_album.title = form.cleaned_data.get('title')
photos = form.cleaned_data.get('photos')
for photo in photos:
album.photos.add(photo)
update_album.save()
return redirect(album_view, request.user.username, album.pk)
else:
album_form = AlbumForm(instance=album)
context = {'album_form': album_form, 'album': album}
return render(request, 'edit_album.html', context)