diff --git a/apps/wallet/views/application.py b/apps/wallet/views/application.py index 1342e0c..0241939 100755 --- a/apps/wallet/views/application.py +++ b/apps/wallet/views/application.py @@ -354,7 +354,7 @@ class CategoryDepositVerifyAPIView(generics.RetrieveAPIView): category = get_object_or_404(Category.objects.all(), uuid=category_uuid) else: category = None - wallet = Wallet.objects.get(owner_uuid=application.uuid, owner_type=TypeChoices.APPLICATION, category=category) + wallet, created = Wallet.objects.get_or_create(owner_uuid=application.uuid, owner_type=TypeChoices.APPLICATION, category=category) return Transaction.objects.filter(application=application, payer_wallet=wallet).all() def retrieve(self, request, *args, **kwargs): @@ -464,7 +464,7 @@ class CategoryWithdrawVerifyAPIView(generics.RetrieveAPIView): category = get_object_or_404(Category.objects.all(), uuid=category_uuid) else: category = None - wallet = Wallet.objects.get(owner_uuid=application.uuid, owner_type=TypeChoices.APPLICATION, category=category) + wallet, created = Wallet.objects.get_or_create(owner_uuid=application.uuid, owner_type=TypeChoices.APPLICATION, category=category) return Transaction.objects.filter(application=application, payee_wallet=wallet).all() def retrieve(self, request, *args, **kwargs):