def put(self, dish_id):
data = Dish.parser.parse_args()
dish = DishModel.find_by_id(dish_id)
if dish is None:
dish = DishModel(dish_id, **data)
else:
dish.name = data['name']
dish.price = data['price']
dish.dish_type = data['dish_type']
dish.save_to_db()
return dish.json()
def post(self, dish_id):
if DishModel.find_by_id(dish_id):
return {
'message': "A dish with id {} already exists".format(dish_id)
}
data = Dish.parser.parse_args()
dish = DishModel(dish_id, **data)
try:
dish.save_to_db()
except:
return {'message': "An error occurred inserting the item."}, 500
return dish.json(), 201
def delete(self, dish_id):
dish = DishModel.find_by_id(dish_id)
if dish:
dish.delete_from_db()
return {'message': 'Item deleted'}
def get(self, dish_id):
dish = DishModel.find_by_id(dish_id)
if dish:
return dish.json()
return {'message': "Dish not found"}, 404