def change_dish(id, do):
id = int(id)
if do == "add":
s = Dish()
else:
s = mydb.session.query(Dish).filter(Dish.id_dish == id).one_or_none()
form = DishForm(request.form, obj=s)
if do == "delete":
mydb.session.delete(s)
mydb.session.flush()
return redirect(url_for('dish'))
if form.button_save.data:
form.populate_obj(s)
mydb.session.add(s)
if s.id_dish != id:
return redirect(url_for('dish', id=s.id_dish))
return render_template('change_dish.html', form=form)
