def edit_restaurant(restaurant_id):
""" Edit a given restaurant. """
restaurant_to_edit = session.query(Restaurant).filter_by(id=restaurant_id).one()
form = RestaurantForm(request.form, restaurant_to_edit)
if form and request.method == "POST":
if form.validate():
form.populate_obj(restaurant_to_edit)
session.add(restaurant_to_edit)
session.commit()
flash("Restaurant edited successfully.")
return redirect(url_for('show_restaurants'))
else:
return render_template('editRestaurant.html', restaurant_id=restaurant_id, form=form)
else:
return render_template('editRestaurant.html', restaurant_id=restaurant_id, form=form)
def new_restaurant():
form = RestaurantForm(request.form)
if form and request.method == 'POST':
if form.validate():
new_rest = Restaurant(name=request.form.get('name'),
address=request.form.get('address'),
city=request.form.get('city'),
url=request.form.get('url'),
phone=request.form.get('phone'),
fax=request.form.get('fax'),
slogan=request.form.get('slogan')
)
session.add(new_rest)
session.commit()
flash('Restaurant %s added to database.' % (new_rest.name,))
return redirect(url_for('show_restaurants'))
else:
return render_template('newRestaurant.html', form=form)
else:
return render_template('newRestaurant.html', form=form)
