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)