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 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 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)