Add UpdateUserWalletView and CustomUserByWalletBalanceSerializer for wallet balance updates

api/serializers.py

@@ -73,6 +73,15 @@ class CustomReadOnlyUserByIDCardSerializer(serializers.ModelSerializer):
         fields = ("id_card",)
 
 
+class CustomUserByWalletBalanceSerializer(serializers.ModelSerializer):
+    """serializer for the user object"""
+
+    class Meta:  # type: ignore
+        model = User
+        # fields = "__all__"
+        fields = ("wallet_balance",)
+
+
 class UserSerializer(serializers.ModelSerializer):
     """serializer for the user object"""
 

api/urls.py

@@ -17,6 +17,7 @@ from .views import (
     ListCreateIslandView,
     RetrieveUpdateDestroyIslandView,
     filter_user,
+    UpdateUserWalletView,
 )
 
 
@@ -29,6 +30,9 @@ urlpatterns = [
     path("tokens/", KnoxTokenListApiView.as_view(), name="knox_tokens"),
     # path("auth/", CustomAuthToken.as_view()),
     path("users/", ListUserView.as_view(), name="users"),
+    path(
+        "update-wallet/<int:pk>/", UpdateUserWalletView.as_view(), name="update-wallet"
+    ),
     path("users/<int:pk>/", UserDetailAPIView.as_view(), name="user-detail"),
     path("users/filter/", filter_user, name="filter-users"),
     path("healthcheck/", healthcheck, name="healthcheck"),

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