[ChickIN] -5 <orderview>

# orders/views.py
import json, uuid
from json             import JSONDecodeError
from enum             import Enum

from django.http      import JsonResponse
from django.views     import View
from django.db        import transaction

from users.utils   import login_required
from carts.models  import Cart
from orders.models import Order, OrderItem, OrderOption

class OrderStatusEnum(Enum):
    PAID            = 1
    PREPARING       = 2
    SHIPPEND        = 3
    DELIVERD        = 4
    ORDER_CANCELLED = 5

class OrderView(View): # 장바구니에 담은 것들을 구매
    @login_required
    def post(self, request):
        try:
            data     = json.loads(request.body)
            user     = request.user
            cart_ids = data['cart_ids'] # cart_id를 리스트 형식으로 여러개를 담아서 cart_ids로 보내준다.
            carts    = Cart.objects.filter(id__in=cart_ids, user=user)

            if not carts.exists():
                return JsonResponse({'message':'NOT_FOUND'}, status=404)

            with transaction.atomic():
                order = Order.objects.create(
                    order_number    = uuid.uuid4(),
                    address         = user.address,
                    user            = user,
                    order_status_id = OrderStatusEnum.PREPARING.value
                )
                for cart in carts:
                    order_item = OrderItem.objects.create(
                        order      = order,
                        product    = cart.product,
                        quantity   = cart.quantity 
                    )

                    cart_options  = cart.cartoption_set.all()
                    order_options = [OrderOption(
                        option     = cart_option.option,
                        order_item = order_item
                    ) for cart_option in cart_options]

                    OrderOption.objects.bulk_create(order_options)

                carts.delete()
                response = {
                    "message" : "created",
                    "data"    : {
                        "order_id"     : order.id,
                        "order_number" : order.order_number
                    }
                }
            return JsonResponse(response, status=201)

        except transaction.TransactionManagementError:
            return JsonResponse({'message':'TransactionManagementError'}, status=400)  
            
    @login_required
    def get(self, request):
        user          = request.user
        orders        = Order.objects.filter(user=user)
        
        order_list = [{
            'order_id'   : order.id,
            'user'       : user.username,
            'address'    : order.address,
            'order_items': [{
                'product_id'    : order_item.product.id,
                'quantity'      : order_item.quantity,
                'product_name'  : order_item.product.name,
                'product_image' : order_item.product.thumbnail,
                'price'         : int((order_item.quantity)*(order_item.product.price)),
                'product_options': [{
                    'option_name'  : order_option.option.name,
                    'option_price' : order_option.option.price
                } for order_option in order_item.orderoption_set.all()] 
            }for order_item in order.orderitem_set.all()]
        } for order in orders]

        return JsonResponse({'order_list':order_list}, status=200)

    @login_required
    def patch(self, request, order_id):
        try:
            data   = json.loads(request.body)
            user   = request.user
            status = data['status']

            order_status = {
                'paid'            : OrderStatusEnum.PAID.value,
                'preparing'       : OrderStatusEnum.PREPARING.value,
                'shippend'        : OrderStatusEnum.SHIPPEND.value,
                'deliverd'        : OrderStatusEnum.DELIVERD.value,
                'order_cancelled' : OrderStatusEnum.ORDER_CANCELLED.value
            }

            order = Order.objects.get(user=user, id=order_id)

            order.order_status_id = order_status[status]
            order.save()

            return JsonResponse({'message':'SUCCESS'}, status=200)

        except Order.DoesNotExist:
            return JsonResponse({'message':'NOT_FOUND'}, status=404)

        except KeyError:
            return JsonResponse({'message':'KEY_ERROR'}, status=400)


class OrderNowView(View): # 바로구매
    @login_required
    def post(self, request):
        try:
            data  = json.loads(request.body)
            user  = request.user

            with transaction.atomic():
                order = Order.objects.create(
                    order_number    = uuid.uuid4(),
                    address         = user.address,
                    user            = user,
                    order_status_id = OrderStatusEnum.PREPARING.value
                )

                order_item  = OrderItem.objects.create(
                    order      = order,
                    product_id = data['product_id'],
                    quantity   = data['quantity']
                )

                if data.get('option_ids'):
                    order_options = [OrderOption(
                        option_id  = option_id,
                        order_item = order_item
                    ) for option_id in data['option_ids']]
                    
                    OrderOption.objects.bulk_create(order_options)

                response = {
                    "message" : "created",
                    "data"    : {
                        "order_id"     : order.id,
                        "order_number" : order.order_number
                    }
                }
            return JsonResponse(response, status=201)

        except transaction.TransactionManagementError:
            return JsonResponse({'message':'TransactionManagementError'}, status=400) 

        except JSONDecodeError:
            return JsonResponse({'message':'JSONDecodeError'}, status=400)

# orders/urls.py
from django.urls import path

from orders.views import OrderView, OrderNowView

urlpatterns = [
    path('', OrderView.as_view()),
    path('/<int:order_id>', OrderView.as_view()), # patch함수는 path로 order_id를 같이 보내야 한다.
    path('/now', OrderNowView.as_view())
]