def editCategory(name):
""" Edit a category """
# user must be authenticated
if 'user_id' not in session:
return redirect('/login')
category = db_session.query(Category).filter_by(name=name).first()
if category is None:
abort(404)
if category.user_id != session['user_id']:
abort(401)
form = CategoryForm(obj=category)
if form.validate_on_submit():
form.populate_obj(category)
db_session.add(category)
try:
db_session.commit()
except exc.IntegrityError:
# category name should be unique
db_session.rollback()
form.name.errors.append("Category already exists.")
return render_template('edit_category.html',
category=category, form=form)
flash("Category %s edited." % category.name)
return redirect(url_for('category', name=category.name))
return render_template('edit_category.html', category=category, form=form)
def editCategory(name):
if "user_id" not in session:
return redirect("/login")
category = db_session.query(Category).filter_by(name=name).one()
if category is None:
abort(404)
if category.user_id != session["user_id"]:
abort(401)
form = CategoryForm(obj=category)
if form.validate_on_submit():
form.populate_obj(category)
db_session.add(category)
db_session.commit()
flash("Category %s edited." % category.name)
return redirect(url_for("category", name=category.name))
return render_template("editCategory.html", category=category, form=form)
