Add UpdateUserWalletView and CustomUserByWalletBalanceSerializer for wallet balance updates

billing/views.py

@@ -10,7 +10,7 @@ from rest_framework.response import Response
 from api.mixins import StaffEditorPermissionMixin
 
 from .models import Device, Payment
-from .serializers import PaymentSerializer
+from .serializers import PaymentSerializer, UpdatePaymentSerializer
 
 
 class InsufficientFundsError(Exception):
@@ -75,6 +75,21 @@ class PaymentDetailAPIView(StaffEditorPermissionMixin, generics.RetrieveAPIView)
     lookup_field = "pk"
 
 
+class UpdatePaymentAPIView(StaffEditorPermissionMixin, generics.UpdateAPIView):
+    queryset = Payment.objects.select_related("user").all()
+    serializer_class = UpdatePaymentSerializer
+    lookup_field = "pk"
+
+    def update(self, request, *args, **kwargs):
+        instance = self.get_object()
+        devices = instance.devices.all()
+        serializer = self.get_serializer(instance, data=request.data, partial=False)
+        serializer.is_valid(raise_exception=True)
+        self.perform_update(serializer)
+        devices.update(is_active=True, expiry_date=instance.expires_at)
+        return Response(serializer.data)
+
+
 class VerifyPaymentView(StaffEditorPermissionMixin, generics.UpdateAPIView):
     serializer_class = PaymentSerializer
     queryset = Payment.objects.all()