def showRestaurant(restaurant_id):
restaurant = rest_crud.getRestaurant(restaurant_id)
creator = rest_crud.getUserById(restaurant.user_id)
menu_items = rest_crud.getRestaurantItems(restaurant_id)
sections= OrderedDict()
sections['Appetizers'] = [item for item in menu_items if item.course == "Appetizer"]
sections['Entrees'] = [item for item in menu_items if item.course == "Entree"]
sections['Desserts'] = [item for item in menu_items if item.course == "Dessert"]
sections['Beverages'] = [item for item in menu_items if item.course == "Beverage"]
if 'username' not in login_session or login_session['user_id'] != restaurant.user_id:
print "user not logged in"
return render_template('menu.html', restaurant=restaurant,
sections=sections,
creator=creator)
else:
return render_template('editrestaurant.html', restaurant=restaurant,
sections=sections,
creator=creator)
def editRestaurant(restaurant_id):
if 'username' not in login_session:
return redirect(url_for('showLogin'))
restaurant = rest_crud.getRestaurant(restaurant_id)
if login_session['user_id'] != restaurant.user_id:
return "you are not allowed to edit this restaurant!"
if request.method == 'GET':
menu_items = rest_crud.getRestaurantItems(restaurant_id)
sections= OrderedDict()
sections['Appetizers'] = [item for item in menu_items if item.course == "Appetizer"]
sections['Entrees'] = [item for item in menu_items if item.course == "Entree"]
sections['Desserts'] = [item for item in menu_items if item.course == "Dessert"]
sections['Beverages'] = [item for item in menu_items if item.course == "Beverage"]
return render_template('editrestaurant.html', restaurant=restaurant,
sections=sections)
else:
# edit on this level only for restaurant name
# menu items edits are handled by editMenuItem function
rest_crud.editRestaurant(restaurant_id, request.form['rest_name'])
flash("Restaurant " + restaurant.name + " Successfully Edited")
return redirect(url_for('showRestaurants'))
def getRestaurantJSON(restaurant_id):
all_items = rest_crud.getRestaurantItems(restaurant_id)
return jsonify(MenuItems=[item.serialize for item in all_items])
