def restaurant_menu(rest_id):
database = Modify()
restaurant = database.select_rests(rest_id)
items = database.select_items(rest_id=rest_id)
return render_template('restaurant_menu.html',
rest=restaurant[0],
items=items)
def add_new_restaurant():
if request.method == 'POST':
database = Modify()
database.add_new_rest(request.form['newname'])
return redirect(url_for('home'))
else:
return render_template('new_rest.html')
def add_new_item():
if request.method == 'POST':
database = Modify()
database.DBInsertItem(request.form['name'], request.form['price'],
request.form['rest_id'], request.form['desc'])
return redirect(url_for('home'))
else:
return render_template('new_item.html')
def home():
database = Modify()
restaurants = database.DBSelectAllRests()
for restaurant in restaurants:
items = database.DBSelectAllItems(restaurant.id)
restaurant_and_items[restaurant.name] = items
return render_template('menu.html',
Restaurant=restaurants[0],
Items=restaurant_and_items[restaurants[1].name])
def delete_item(item_id):
if request.method == 'POST':
try:
database = Modify()
items = database.DBSelectAllItems()
item = ""
for i in items:
if i.id == item_id:
item = i
break
print('this is the item : {}'.format(item))
result = database.DBDeleteItem(item.name)
print(result)
return redirect(url_for('home'))
except Exception as e:
print('failed')
print(e)
return redirect(url_for('home'))
else:
print('get mothod')
database = Modify()
items = database.DBSelectAllItems(2)
item = ""
for i in items:
if i.id == item_id:
item = i
print(item.name)
restaurant = database.DBSelectRestById(item.restaurant_id)
return render_template('delete_template.html',
item=item,
Restaurant=restaurant)
def delete_item(rest_id, item_id):
if request.method == 'POST':
database = Modify()
olditem = database.select_items(rest_id=rest_id, item_id=item_id)
database.delete_item(item_id)
flash('Item ({}) deleted successfully.'.format(olditem[0].name))
return redirect(url_for('restaurant_menu', rest_id=rest_id))
else:
database = Modify()
item = database.select_items(rest_id=rest_id, item_id=item_id)[0]
restaurant = database.select_rests(rest_id)[0]
return render_template('delete_item.html',
item=item,
restaurant=restaurant)
def delete_rest(rest_id):
if request.method == 'POST':
database = Modify()
restaurant = database.select_rests(rest_id)
database.delete_rest(restaurant[0].id)
flash('restaurant ({}) deleted successfully.'.format(
restaurant[0].name))
return redirect(url_for('home'))
else:
database = Modify()
restaurant = database.select_rests(rest_id)[0]
return render_template('delete_rest.html', restaurant=restaurant)
def edit_rest_name(rest_id):
if request.method == 'POST':
database = Modify()
restaurant = database.select_rests(rest_id)
database.edit_rest(restaurant[0].id, request.form['newname'])
flash('Restaurant ({}) edited to ({})'.format(restaurant[0].name,
request.form['newname']))
return redirect(url_for('home'))
else:
database = Modify()
restaurant = database.select_rests(rest_id)
return render_template('edit_rest_name.html', restaurant=restaurant)
def edit_item(rest_id, item_id):
if request.method == 'POST':
database = Modify()
olditem = database.select_items(rest_id=rest_id, item_id=item_id)
database.edit_item(rest_id, item_id, request.form['newname'],
request.form['price'], request.form['desc'])
flash('item ({}) edited successfully to ({})'.format(
olditem[0].name, request.form['newname']))
return redirect(url_for('restaurant_menu', rest_id=rest_id))
else:
database = Modify()
item = database.select_items(rest_id=rest_id, item_id=item_id)[0]
restaurant = database.select_rests(rest_id)[0]
return render_template('edit_item.html',
item=item,
restaurant=restaurant)
def add_new_item(rest_id):
if request.method == 'POST':
database = Modify()
database.add_new_item(request.form['name'], request.form['price'],
request.form['desc'], rest_id)
flash('Item({}) added successfully.'.format(request.form["name"]))
return redirect(url_for('restaurant_menu', rest_id=rest_id))
else:
database = Modify()
restaurant = database.select_rests(rest_id)[0]
return render_template('add_new_item.html', restaurant=restaurant)
def edit_item(item_id):
if request.method == 'POST':
database = Modify()
items = database.DBSelectAllItems()
item = ''
for i in items:
if i.id == item_id:
item = i
database.DBEditItem(item.name, request.form['newname'])
return redirect(url_for('home'))
else:
database = Modify()
items = database.DBSelectAllItems(2)
item = ""
for i in items:
if i.id == item_id:
item = i
return render_template('edit_template.html', item=item)
def home():
database = Modify()
restaurants = database.select_rests()
return render_template('home.html', restaurants=restaurants)