🔨Refactor: validation로직 추가

oz-main-project-team-02 · Nov 4, 2024 · 166bdd0 · 166bdd0
1 parent af172c8
commit 166bdd0
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/wallets/views/wallet_view.py b/src/wallets/views/wallet_view.py
@@ -114,6 +114,9 @@ def post(self, request, *args, **kwargs):
         serializer.is_valid(raise_exception=True)
         coin_amount = serializer.validated_data["coin"]
 
+        if wallet.coin < coin_amount:
+            return Response({"error": "잔액이 부족합니다."}, status=status.HTTP_400_BAD_REQUEST)
+
         # F expression으로 잔액에서 차감
         wallet.coin = F("coin") - coin_amount
         wallet.save()