80 lines
1.9 KiB
Python
80 lines
1.9 KiB
Python
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
|