Feat: 연도 필터 추가

Author: jungmir

sponsor/migrations/0008_merge_20240806_2206.py

@@ -0,0 +1,13 @@
+# Generated by Django 4.1.5 on 2024-08-06 13:06
+
+from django.db import migrations
+
+
+class Migration(migrations.Migration):
+
+    dependencies = [
+        ("sponsor", "0006_sponsorlevel_year"),
+        ("sponsor", "0007_remove_sponsorbenefit_offer_and_more"),
+    ]
+
+    operations = []

sponsor/viewsets.py

@@ -25,14 +25,13 @@
 from sponsor.validators import SponsorValidater
 
 
-
 class SponsorBenefitViewSet(ModelViewSet):
     lookup_field = "id"
     http_method_names = ["get", "post", "put", "delete"]
     serializer_class = SponsorBenefitSerializer
 
     def get_queryset(self):
-        return SponsorBenefit.objects.all()
+        return SponsorBenefit.objects.filter(level__year=self.request.version).all()
 
 
 class SponsorLevelViewSet(ModelViewSet):
@@ -71,6 +70,7 @@ def create_or_update_benefits(self, request):
         serializer.save()
         return Response(serializer.data)
 
+
 class SponsorViewSet(
     mixins.CreateModelMixin,
     mixins.RetrieveModelMixin,
@@ -84,7 +84,12 @@ class SponsorViewSet(
     validator = SponsorValidater()
 
     def get_queryset(self):
-        return super().get_queryset().filter(paid_at__isnull=False, level__year=self.request.version).order_by("level__order", "paid_at")
+        return (
+            super()
+            .get_queryset()
+            .filter(paid_at__isnull=False, level__year=self.request.version)
+            .order_by("level__order", "paid_at")
+        )
 
     def get_serializer_class(self):
         if self.action == "list":