def restaurant_add_update_category():
user = get_user()
if (user):
restaurant = Restaurant.get_from_id(user.restaurant_id)
if (restaurant):
name = get_header("name")
id = get_header_int("id")
if (name and len(name) <= 50):
if (is_int(id)):
category = Category.get_from_id(restaurant, id)
if (category):
if (get_header("remove")):
category.remove()
else:
category.name = name
get_session().commit()
return "true"
else:
abort(404)
else:
Category.add(name, restaurant)
return "true"
else:
abort(400)
else:
abort(404)
else:
abort(404)
def restaurant_add_update_product():
user = get_user()
if (user):
restaurant = Restaurant.get_from_id(user.restaurant_id)
if (restaurant):
name = get_header("name")
description = get_header("description")
price = get_header_float("price")
id = get_header_int("id")
if (is_int(id)):
product = Product.get_from_id(id)
if (product):
if (get_header("remove")):
product.remove()
else:
if (name and description and is_float(price)
and len(name) <= 50
and len(description) <= 100):
product.name = name
product.description = description
product.price = price
get_session().commit()
else:
abort(400)
return "true"
else:
abort(404)
else:
categoryId = get_header_int("category")
if (is_int(categoryId) and len(name) <= 50
and len(description) <= 100):
category = Category.get_from_id(restaurant, categoryId)
if (category):
Product.add(name, description, price, restaurant,
category)
return "true"
else:
abort(404)
else:
abort(400)
else:
abort(404)
else:
abort(404)
