create user if not available in getting user balance

This commit is contained in:
Sayyid Hamid Mahdavi 2025-06-05 13:42:33 +03:30
parent e7a6b632ab
commit 94b3d521d4

View file

@ -9,7 +9,7 @@ from apps.users.models import User
from apps.wallet.constans import TypeChoices, StateChoices from apps.wallet.constans import TypeChoices, StateChoices
from apps.wallet.models import Transaction from apps.wallet.models import Transaction
from apps.wallet.serializers import ApplicationDepositSerializer, ApplicationWithdrawSerializer, \ from apps.wallet.serializers import ApplicationDepositSerializer, ApplicationWithdrawSerializer, \
ApplicationTransactionSerializer, ApplicationBalanceSerializer ApplicationTransactionSerializer, ApplicationBalanceSerializer, MyWalletSerializer
def get_application(request): def get_application(request):
@ -37,12 +37,12 @@ class UserBalanceAPIView(generics.RetrieveAPIView):
API for getting specified user or application balance. API for getting specified user or application balance.
''' '''
permission_classes = [IsAuthenticatedOrTokenHasScope] permission_classes = [IsAuthenticatedOrTokenHasScope]
serializer_class = ApplicationBalanceSerializer serializer_class = MyWalletSerializer
required_scopes = ['wallet.application:get_user_balance'] required_scopes = ['wallet.application:get_user_balance']
def get_object(self): def get_object(self):
user_uuid = self.kwargs.get('uuid') 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) # user = get_object_or_404(User.objects.all(), uuid=user_uuid)
return user return user