project.py

from flask import Flask, render_template, request, redirect, url_for, jsonify
from sqlalchemy import create_engine
from sqlalchemy.orm import sessionmaker
from database_setup import Base, Restaurant, MenuItem

app = Flask(__name__)

engine = create_engine('sqlite:///restaurantmenu.db')
Base.metadata.bind = engine


@app.route('/restaurants/<int:restaurant_id>/menu/JSON')
def restaurantMenuJSON(restaurant_id):
	DBSession = sessionmaker(bind=engine)
	session = DBSession()
	items = session.query(MenuItem).filter_by(restaurant_id=restaurant_id).all()
	return jsonify(MenuItems=[i.serialize for i in items])


@app.route('/restaurants/restaurant-id=<int:restaurant_id>/menu-id=<int:menu_id>/JSON')
def menuJSON(restaurant_id, menu_id):
	DBSession = sessionmaker(bind=engine)
	session = DBSession()
	menu = session.query(MenuItem).filter_by(id=menu_id).one()
	return jsonify(MenuItem=menu.serialize)

@app.route('/')
@app.route('/restaurants/<int:restaurant_id>/')
def RestaurantMenu(restaurant_id):
	DBSession = sessionmaker(bind=engine)
	session = DBSession()
	restaurant = session.query(Restaurant).get(restaurant_id)
	items = session.query(MenuItem).filter_by(restaurant_id=restaurant_id)
	return render_template('menu.html', restaurant=restaurant, items=items)


@app.route('/restaurants/<int:restaurant_id>/new/', methods=['GET', 'POST'])
def newMenuItem(restaurant_id):
	if request.method == 'POST':
		DBSession = sessionmaker(bind=engine)
		session = DBSession()
		newItem = MenuItem(name=request.form['name'], restaurant_id=restaurant_id)
		session.add(newItem)
		session.commit()
		return redirect(url_for('RestaurantMenu', restaurant_id=restaurant_id))
	else:
		return render_template('newmenuitem.html', restaurant_id=restaurant_id)


@app.route('/restaurants/<int:restaurant_id>/<int:menu_id>/edit/', methods=['GET', 'POST'])
def editMenuItem(restaurant_id, menu_id):
	DBSession = sessionmaker(bind=engine)
	session = DBSession()
	editedItem = session.query(MenuItem).filter_by(id=menu_id).one()
	if request.method == "POST":
		if request.form['name']:
			editedItem.name = request.form['name']
		session.add(editedItem)
		session.commit()
		return redirect(url_for('RestaurantMenu', restaurant_id=restaurant_id))
	else:
		return render_template('editmenuitem.html', restaurant_id=restaurant_id, menu_id=menu_id, item=editedItem)


@app.route('/restaurants/<int:restaurant_id>/<int:menu_id>/delete/', methods=['GET', 'POST'])
def deleteMenuItem(restaurant_id, menu_id):
	DBSession = sessionmaker(bind=engine)
	session = DBSession()
	item = session.query(MenuItem).filter_by(id=menu_id).one()
	if request.method == "POST":
		session.delete(item)
		session.commit()
		return redirect(url_for('RestaurantMenu', restaurant_id=restaurant_id))
	else:
		return render_template('deletemenuitem.html', restaurant_id=restaurant_id, menu_id=menu_id, item=item)


if __name__ == '__main__':
	app.debug = True
	app.run(host='0.0.0.0', port=5000)