update: 구좌수 검증 추가

Author: golony6449

sponsor/validators.py

@@ -0,0 +1,16 @@
+from collections import OrderedDict
+
+from sponsor.models import Sponsor
+
+
+class SponsorValidater:
+    def assert_create(self, sponsor: OrderedDict):
+        target = [
+            self.check_remain_slot(sponsor)
+        ]
+
+    def check_remain_slot(self, sponsor: OrderedDict):
+        target_level = sponsor.get("level")
+
+        if target_level.limit <= len(Sponsor.objects.filter(level=target_level, accepted=True)):
+            raise RuntimeError("ERROR: 남은 슬롯 없음")

sponsor/viewsets.py

@@ -1,5 +1,6 @@
 from typing import Type
 
+from django.db.transaction import atomic
 from django.shortcuts import get_object_or_404
 from rest_framework.response import Response
 from rest_framework.viewsets import ModelViewSet
@@ -12,12 +13,14 @@
     SponsorRemainingAccountSerializer,
     SponsorSerializer,
 )
+from sponsor.validators import SponsorValidater
 
 
 class SponsorViewSet(ModelViewSet):
     serializer_class = SponsorSerializer
     permission_classes = [IsOwnerOrReadOnly]  # 본인 소유만 수정가능
     http_method_names = ["get", "post"]  # 지금은 조회/등록만 가능 TODO: 추후 수정기능 추가
+    validator = SponsorValidater()
 
     def get_queryset(self):
         return Sponsor.objects.all()
@@ -27,10 +30,14 @@ def list(self, request, *args, **kwargs):
         serializer = SponsorListSerializer(queryset, many=True)
         return Response(serializer.data)
 
+    @atomic
     def create(self, request, *args, **kwargs):
         serializer = self.get_serializer(data=request.data)
         serializer.is_valid(raise_exception=True)
-        serializer.save()
+        self.validator.assert_create(serializer.validated_data)
+
+        new_sponsor = serializer.save()
+
         return Response(serializer.data)
 
     def retrieve(self, request, *args, **kwargs):