示例#1
0
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)
示例#2
0
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)