def create():
if not session.get('user'):
return redirect(url_for('home'))
form = CreateAlbumForm()
if form.validate_on_submit():
title = request.form.get('title')
description = request.form.get('description')
date = datetime.datetime.strptime(request.form.get('date'), '%Y-%m-%d')
new_album = Album(title=title, description=description, date=date)
db.session.add(new_album)
db.session.commit()
return redirect(url_for('home'))
return render_template('create.html', form=form)
def createAlbum():
form = CreateAlbumForm(prefix="createAlbum")
if form.validate_on_submit():
album = Album(name=form.name.data,
description=form.description.data,
creationDate=datetime.utcnow(),
numImages=0,
coverImageId=0)
if form.passwordProtected.data and form.passwordHash.data:
album.passwordHash = getAlbumPasswordHash(album.creationDate, form.passwordHash.data)
db.session.add(album)
db.session.commit()
return admin(createAlbumForm=form)
def create_new_album(username):
form = CreateAlbumForm()
form.country.choices = country_choices
if form.validate_on_submit():
title = form.title.data
country_id = form.country.data
album = Album.add_album(g.user.id, country_id, title)
if album:
country_albums[album.country.iso] += 1
return redirect(
f'/users/{username}/albums/{countries[country_id]["iso"]}/{album.title}'
)
else:
form.title.errors.append('Album title already exists')
return render_template('form.html', form=form)
return render_template('form.html', form=form)
