swag-shop/app/api/routes/cart_routes.py

from flask import jsonify, abort, request
from flask_jwt_extended import jwt_required, get_jwt_identity

from app.doc.cart_swag import (
    show_cart_swagger,
    add_to_cart_swagger,
    remove_from_cart_swagger,
    update_count_in_cart_swagger,
    purchase_swagger,
)

from flasgger import swag_from

from app.api import bp_cart

from app.services.cart_service import CartService


@bp_cart.route("", methods=["GET"])
@jwt_required()
@swag_from(show_cart_swagger)
def show_cart():
    user_id = get_jwt_identity()

    result, status_code = CartService.show_cart(user_id)

    return result, status_code


@bp_cart.route("/add/<int:product_id>", methods=["PUT"])
@jwt_required()
@swag_from(add_to_cart_swagger)
def add_to_cart(product_id: int):
    user_id = get_jwt_identity()
    count = request.args.get("count", default=1, type=int)

    if count < 1:
        return abort(400)

    result, status_code = CartService.add_to_cart(user_id, product_id, count)

    return result, status_code


@bp_cart.route("/remove/<int:product_id>", methods=["DELETE"])
@jwt_required()
@swag_from(remove_from_cart_swagger)
def remove_from_cart(product_id: int):
    user_id = get_jwt_identity()

    result, status_code = CartService.delete_from_cart(user_id, product_id)

    return result, status_code


@bp_cart.route("/update/<int:product_id>", methods=["PUT"])
@jwt_required()
@swag_from(update_count_in_cart_swagger)
def update_count_in_cart(product_id: int):
    user_id = get_jwt_identity()
    count = request.args.get("count", type=int)

    if not count:
        return abort(400)

    result, status_code = CartService.update_count(user_id, product_id, count)

    return result, status_code


@bp_cart.route("/purchase", methods=["GET"])
@jwt_required()
@swag_from(purchase_swagger)
def purchase():
    user_id = get_jwt_identity()

    result, status_code = CartService.purchase(user_id)

    return result, status_code