Added cancel payment view and payment add for admin

backend/apps/payments/serializers.py

@@ -19,6 +19,21 @@ class PaymentStatusSerializer(serializers.Serializer):
     payment_id = serializers.CharField(required=True)
 
 
+class PaymentIdSerializer(serializers.Serializer):
+    payment_id = serializers.CharField(help_text="Unique identifier for a payment.")
+
+    def validate_payment_id(self, value):
+        """
+        Check that the payment_id exists in the database and is associated with an order.
+        You might want to add more complex validations depending on your business logic.
+        """
+        if not Order.objects.filter(payment_id=value).exists():
+            raise serializers.ValidationError(
+                "No order found with the given payment ID."
+            )
+        return value
+
+
 class PaymentListOutputSerializer(serializers.ModelSerializer):
     buyer = BuyerPaymentSerializer(source="cart.buyer", read_only=True)
 
@@ -32,3 +47,8 @@ class Meta:
             "payment_status",
             "ticket_file",
         ]
+
+
+class PaymentItemInputSerializer(serializers.Serializer):
+    cart_id = serializers.IntegerField(required=True)
+    total = serializers.FloatField(required=True)

backend/apps/payments/urls.py

@@ -2,6 +2,8 @@
 
 from .views import (
     BookingsByOrderIdAPIView,
+    PaymentCancelView,
+    PaymentItemInputView,
     PaymentListView,
     PaymentProcessingView,
     PaymentStatusView,
@@ -17,5 +19,9 @@
         BookingsByOrderIdAPIView.as_view(),
         name="bookings-by-order",
     ),
-    path("", PaymentListView.as_view(), name="payment-list"),
+    path("list/", PaymentListView.as_view(), name="payment-list"),
+    path("add-item/", PaymentItemInputView.as_view(), name="payment-add-item"),
+    path(
+        "cancel/<str:payment_id>/", PaymentCancelView.as_view(), name="payment-cancel"
+    ),
 ]

backend/apps/payments/views.py

@@ -1,4 +1,5 @@
 import json
+import secrets
 
 from apps.amo.views import post_orders
 from apps.booking.models import Booking, Buyer, Cart
@@ -15,6 +16,8 @@
 
 from .models import Order
 from .serializers import (
+    PaymentIdSerializer,
+    PaymentItemInputSerializer,
     PaymentListOutputSerializer,
     PaymentProcessingSerializer,
     PaymentStatusSerializer,
@@ -30,6 +33,72 @@ def get(self, request, *args, **kwargs):
         return self.list(request, *args, **kwargs)
 
 
+class PaymentItemInputView(generics.CreateAPIView):
+    serializer_class = PaymentItemInputSerializer
+
+    def post(self, request, *args, **kwargs):
+        serializer = self.get_serializer(data=request.data)
+        if not serializer.is_valid():
+            return Response(serializer.errors, status=status.HTTP_400_BAD_REQUEST)
+
+        cart_id = request.data.get("cart_id")
+        total = request.data.get("total")
+
+        try:
+            cart = Cart.objects.get(id=cart_id)
+        except Cart.DoesNotExist:
+            return Response(
+                {"error": "Cart not found"}, status=status.HTTP_404_NOT_FOUND
+            )
+
+        payment_id = secrets.token_hex(10)
+        confirmation_token = secrets.token_hex(10)
+
+        order = Order.objects.create(
+            cart=cart,
+            total=total,
+            payment_id=payment_id,
+            confirmation_token=confirmation_token,
+            ticket_file=request.data.get(
+                "ticket_file", "https://kolomnago.ru/ticket_file.pdf"
+            ),
+            qr_code=request.data.get("qr_code", "https://kolomnago.ru/qr_code"),
+            payment_status=request.data.get("payment_status", "pending"),
+        )
+
+        return Response(
+            {
+                "payment_id": payment_id,
+                "confirmation_token": confirmation_token,
+                "message": "Order created successfully",
+            },
+            status=status.HTTP_201_CREATED,
+        )
+
+
+class PaymentCancelView(generics.GenericAPIView):
+    def patch(self, request, *args, **kwargs):
+        payment_id = kwargs.get("payment_id")
+        if payment_id is None:
+            return Response(
+                {"error": "Missing payment ID"}, status=status.HTTP_400_BAD_REQUEST
+            )
+
+        order = Order.objects.filter(payment_id=payment_id).first()
+        if order is None:
+            return Response(
+                {"error": "Order not found"}, status=status.HTTP_404_NOT_FOUND
+            )
+        if order.payment_status == "canceled":
+            return Response(
+                {"error": "Order already canceled"}, status=status.HTTP_409_CONFLICT
+            )
+
+        order.payment_status = "canceled"
+        order.save()
+        return Response({"success": "Payment canceled"}, status=status.HTTP_200_OK)
+
+
 class PaymentStatusView(generics.RetrieveAPIView):
     serializer_class = PaymentStatusSerializer