def add_title():
"""
Add a title to the database
"""
check_admin()
add_title = True
form = TitleForm()
if form.validate_on_submit():
title = Title(name=form.name.data)
try:
# add title to the database
db.session.add(title)
db.session.commit()
flash('You have successfully added a new title.')
except:
# in case title name already exists
flash('Error: title name already exists.')
# redirect to titles page
return redirect(url_for('admin.list_titles'))
# load title template
return render_template('admin/titles/title.html',
action="Add",
add_title=add_title,
form=form,
title="Add Title")
def edit_title(id):
"""
Edit a title
"""
check_admin()
add_title = False
title = Title.query.get_or_404(id)
form = TitleForm(obj=title)
if form.validate_on_submit():
title.name = form.name.data
db.session.commit()
flash('You have successfully edited the title.')
# redirect to the titles page
return redirect(url_for('admin.list_titles'))
form.name.data = title.name
return render_template('admin/titles/title.html',
action="Edit",
add_title=add_title,
form=form,
title_edit=title,
title="Edit Title")
