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)