From 94b3d521d4d2566b16b8dc54a9f7ca87d59a1246 Mon Sep 17 00:00:00 2001
From: Sayyid Hamid Mahdavi <sayyidhamidmahdavi@gmail.com>
Date: Thu, 5 Jun 2025 13:42:33 +0330
Subject: [PATCH] create user if not available in getting user balance

---
 apps/wallet/views/application.py | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/apps/wallet/views/application.py b/apps/wallet/views/application.py
index 74821b6..4f75456 100755
--- a/apps/wallet/views/application.py
+++ b/apps/wallet/views/application.py
@@ -9,7 +9,7 @@ from apps.users.models import User
 from apps.wallet.constans import TypeChoices, StateChoices
 from apps.wallet.models import Transaction
 from apps.wallet.serializers import ApplicationDepositSerializer, ApplicationWithdrawSerializer, \
-    ApplicationTransactionSerializer, ApplicationBalanceSerializer
+    ApplicationTransactionSerializer, ApplicationBalanceSerializer, MyWalletSerializer
 
 
 def get_application(request):
@@ -37,12 +37,12 @@ class UserBalanceAPIView(generics.RetrieveAPIView):
     API for getting specified user or application balance.
     '''
     permission_classes = [IsAuthenticatedOrTokenHasScope]
-    serializer_class = ApplicationBalanceSerializer
+    serializer_class = MyWalletSerializer
     required_scopes = ['wallet.application:get_user_balance']
 
     def get_object(self):
         user_uuid = self.kwargs.get('uuid')
-        user = User.objects.get_or_create(uuid=user_uuid)
+        user, created = User.objects.get_or_create(uuid=user_uuid)
         # user = get_object_or_404(User.objects.all(), uuid=user_uuid)
 
         return user