from rest_framework.views import APIView
from rest_framework.generics import ListAPIView
from rest_framework.response import Response
from rest_framework import status
from package_details.models.package_details import PackageDetails
from package_details.serializer.checkout_serializer import CheckoutSerializer
from django.shortcuts import get_object_or_404

class CheckoutView(APIView):
    def get(self, request, pk):
        category = get_object_or_404(PackageDetails, pk=pk)
        serializer = CheckoutSerializer(category)
        return Response(serializer.data)
    
    def delete(self, request, pk):
        category = get_object_or_404(PackageDetails, pk=pk)
        category.delete()
        return Response({"success": "Checkout deleted successfully"}, status=status.HTTP_200_OK)
    
class CheckoutListView(ListAPIView):
    queryset = PackageDetails.objects.all().select_related('discount')
    serializer_class = CheckoutSerializer