Merge pull request #45 from golony6449/golony/feature/sponsor

update: available 필드 추가

Author: golony6449

sponsor/serializers.py

@@ -1,10 +1,11 @@
 from rest_framework.fields import SerializerMethodField
-from rest_framework.serializers import ModelSerializer
+
+import rest_framework.serializers as serializers
 
 from sponsor.models import Sponsor, SponsorLevel
 
 
-class SponsorSerializer(ModelSerializer):
+class SponsorSerializer(serializers.ModelSerializer):
     class Meta:
         model = Sponsor
         fields = [
@@ -23,7 +24,7 @@ class Meta:
         ]
 
 
-class SponsorListSerializer(ModelSerializer):
+class SponsorListSerializer(serializers.ModelSerializer):
     class Meta:
         model = Sponsor
         fields = [
@@ -35,7 +36,7 @@ class Meta:
         ]
 
 
-class SponsorLevelSerializer(ModelSerializer):
+class SponsorLevelSerializer(serializers.ModelSerializer):
     class Meta:
         model = SponsorLevel
         fields = [
@@ -47,8 +48,9 @@ class Meta:
         ]  # TODO: Add fields to show
 
 
-class SponsorRemainingAccountSerializer(ModelSerializer):
+class SponsorRemainingAccountSerializer(serializers.ModelSerializer):
     remaining = SerializerMethodField()
+    available = SerializerMethodField()
 
     class Meta:
         model = SponsorLevel
@@ -58,8 +60,13 @@ class Meta:
             "limit",
             "remaining",
             "id",
+            "available",
         ]
 
     @staticmethod
     def get_remaining(obj):
         return obj.current_remaining_number
+
+    @staticmethod
+    def get_available(obj: SponsorLevel):
+        return True if obj.current_remaining_number < obj.limit else False

sponsor/viewsets.py

@@ -75,14 +75,14 @@ def list(self, request, *args, **kwargs):
 
 
 class SponsorRemainingAccountViewSet(ModelViewSet):
-    serializer_class = SponsorLevelSerializer
+    serializer_class = SponsorRemainingAccountSerializer
     http_method_names = ["get"]
 
     def get_queryset(self):
         return SponsorLevel.objects.all()
 
     def list(self, request, *args, **kwargs):
         queryset = SponsorLevel.objects.all().order_by("-price")
-        serializer = SponsorRemainingAccountSerializer(queryset, many=True)
+        serializer = self.get_serializer(queryset, many=True)
 
         return Response(serializer.data)