def deleteRestaurant(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 delete this restaurant!"
if request.method == 'GET':
return render_template('deleterestaurant.html', restaurant=restaurant)
else:
rest_crud.deleteRestaurant(restaurant_id)
flash('Restaurant ' + restaurant.name + ' Successfully Deleted')
return redirect(url_for('showRestaurants'))
def deleteMenuItem(restaurant_id, menu_item_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 delete an item from this restaurant!"
menu_item = rest_crud.getMenuItem(menu_item_id)
if request.method == 'GET':
return render_template('deletemenuitem.html', restaurant_id=restaurant_id, menu_item=menu_item)
else:
rest_crud.deleteMenuItem(menu_item_id)
flash("Menu Item " + menu_item.name + " Successfully Deleted")
return redirect(url_for('showRestaurant', restaurant_id=restaurant_id))
def editMenuItem(restaurant_id, menu_item_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 an item from this restaurant!"
menu_item = rest_crud.getMenuItem(menu_item_id)
if request.method == 'GET':
return render_template('editmenuitem.html', restaurant_id=restaurant_id, item=menu_item)
else:
rest_crud.editMenuItem(menu_item_id=menu_item_id,
name=request.form['name'],
description=request.form['description'],
course=request.form['course'],
price=request.form['price'])
flash("Menu Item " + request.form['name'] + " Successfully Edited")
return redirect(url_for('showRestaurant', restaurant_id=restaurant_id))
def newMenuItem(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 add an item to this restaurant!"
if request.method == 'GET':
return render_template('newmenuitem.html', restaurant=restaurant)
else:
menu_item = rest_crud.newMenuItem(name=request.form['name'],
description=request.form['description'],
course=request.form['course'],
price=request.form['price'],
restaurant_id=restaurant_id,
user_id=login_session['user_id'])
flash("New Menu Item " + menu_item.name + " Created")
return redirect(url_for('showRestaurant', restaurant_id=restaurant_id))
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'))
