Add UpdateUserWalletView and CustomUserByWalletBalanceSerializer for wallet balance updates

api/views.py

@@ -11,7 +11,11 @@ from rest_framework.response import Response
 from rest_framework import status
 from rest_framework.exceptions import ValidationError
 from rest_framework.decorators import api_view, permission_classes
-from api.serializers import AtollSerializer, IslandSerializer
+from api.serializers import (
+    AtollSerializer,
+    IslandSerializer,
+    CustomUserByWalletBalanceSerializer,
+)
 
 # knox imports
 from knox.views import LoginView as KnoxLoginView
@@ -44,6 +48,25 @@ class ErrorMessages:
     INVALID_ACCOUNT = "Please enter a valid account number."
 
 
+class UpdateUserWalletView(generics.UpdateAPIView):
+    # Create user API view
+    serializer_class = CustomUserByWalletBalanceSerializer
+    permission_classes = (permissions.IsAuthenticated,)
+    queryset = User.objects.all()
+
+    def update(self, request, *args, **kwargs):
+        wallet_balance = request.data.get("wallet_balance")
+        if not wallet_balance:
+            return Response(
+                {"message": "wallet_balance is required."},
+                status=status.HTTP_400_BAD_REQUEST,
+            )
+        user = self.get_object()
+        user.wallet_balance = wallet_balance
+        user.save()
+        return Response({"message": "Wallet balance updated successfully."})
+
+
 class CreateUserView(generics.CreateAPIView):
     # Create user API view
     serializer_class = UserSerializer