Feature: Add SponsorLevelViewSet

Author: oleveloper

sponsor/routers.py

@@ -5,6 +5,7 @@
 
 def get_router():
     router = DefaultRouter()
+    router.register("prospectus", SponsorLevelViewSet, basename="prospectus")
     router.register("", SponsorViewSet, basename="sponsor")
 
     return router

sponsor/serializers.py

@@ -1,6 +1,7 @@
+from rest_framework.fields import SerializerMethodField
 from rest_framework.serializers import ModelSerializer
 
-from sponsor.models import Sponsor
+from sponsor.models import Sponsor, SponsorLevel
 
 
 class SponsorSerializer(ModelSerializer):
@@ -31,3 +32,21 @@ class Meta:
             "logo_image",
             "id",
         ]
+
+
+class SponsorLevelSerializer(ModelSerializer):
+    remaining = SerializerMethodField()
+
+    class Meta:
+        model = SponsorLevel
+        fields = [
+            "name",
+            "price",
+            "desc",
+            "limit",
+            "remaining",
+            "id",
+        ]
+
+    def get_remaining(self, obj):
+        return obj.current_remaining_number

sponsor/viewsets.py

@@ -1,11 +1,10 @@
 from django.shortcuts import get_object_or_404
-from rest_framework.permissions import IsAuthenticatedOrReadOnly
 from rest_framework.response import Response
 from rest_framework.viewsets import ModelViewSet
 
-from sponsor.models import Sponsor
+from sponsor.models import Sponsor, SponsorLevel
 from sponsor.permissions import IsOwnerOrReadOnly, OwnerOnly
-from sponsor.serializers import SponsorListSerializer, SponsorSerializer
+from sponsor.serializers import SponsorListSerializer, SponsorSerializer, SponsorLevelSerializer
 
 
 class SponsorViewSet(ModelViewSet):
@@ -45,3 +44,17 @@ def check_owner_permission(self, request, sponsor_data: Sponsor):
         return OwnerOnly.has_object_permission(
             self=OwnerOnly, request=request, view=self, obj=sponsor_data
         )
+
+
+class SponsorLevelViewSet(ModelViewSet):
+    serializer_class = SponsorLevelSerializer
+    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 = SponsorLevelSerializer(queryset, many=True)
+
+        return Response(serializer.data)