Sponsor API 추가

Author: oleveloper

pyconkr/urls.py

@@ -22,16 +22,16 @@
     SpectacularSwaggerView,
 )
 
-import sponsor.routers
+import payment.urls
+import sponsor.urls
 import status.urls
 import ticket.urls
-import payment.urls
 
 urlpatterns = [
     path("api-auth/", include("rest_framework.urls")),
     path("summernote/", include("django_summernote.urls")),
     path("admin/", admin.site.urls),
-    path("sponsors", include(sponsor.routers.get_router().urls)),
+    path("sponsors/", include(sponsor.urls)),
     path("programs/", include("program.urls")),
     path("statuses/", include(status.urls)),
     path("tickets/", include(ticket.urls)),

sponsor/routers.py

@@ -9,7 +9,7 @@ class Meta:
         model = Sponsor
         fields = [
             "name",
-            # "desc",  # 국문/영문 모두 한 필드에 담아 제공하는 것으로 결정 # TODO: 상세 페이지 오픈 후 활성화
+            "desc",  # 국문/영문 모두 한 필드에 담아 제공
             "manager_name",  # 상세에만 포함되는 필드
             "manager_email",  # 상세에만 포함되는 필드
             "manager_tel",  # 상세에만 포함되는 필드
@@ -23,28 +23,35 @@ class Meta:
         ]
 
 
-class SponsorListSerializer(serializers.ModelSerializer):
+class SponsorDetailSerializer(serializers.ModelSerializer):
     class Meta:
         model = Sponsor
         fields = [
             "name",
-            "level",
+            "desc",
             "url",
             "logo_image",
+            "level",
             "id",
         ]
 
 
-class SponsorLevelSerializer(serializers.ModelSerializer):
+class SponsorListSerializer(serializers.ModelSerializer):
+    paid = SerializerMethodField()
+
     class Meta:
-        model = SponsorLevel
+        model = Sponsor
         fields = [
             "name",
-            "price",
-            "desc",
-            "limit",
+            "logo_image",
+            "level",
             "id",
-        ]  # TODO: Add fields to show
+            "paid",
+        ]
+
+    @staticmethod
+    def get_paid(obj: Sponsor):
+        return obj.paid_at is not None
 
 
 class SponsorRemainingAccountSerializer(serializers.ModelSerializer):

sponsor/serializers.py

@@ -1,6 +1,15 @@
-from django.contrib import admin
-from django.urls import include, path
+from django.urls import path
+from rest_framework.routers import DefaultRouter
+
+from sponsor.viewsets import SponsorViewSet
+
+router = DefaultRouter()
+router.register("", SponsorViewSet, basename="sponsor")
 
 urlpatterns = [
-    # path("", ),   # TODO
+    path("list/", SponsorViewSet.as_view({"get": "list"})),
+    path(
+        "list/<int:id>/",
+        SponsorViewSet.as_view({"get": "retrieve", "put": "update"}),
+    ),
 ]

sponsor/urls.py

@@ -2,13 +2,14 @@
 
 from django.db.transaction import atomic
 from django.shortcuts import get_object_or_404
+from rest_framework import status
 from rest_framework.response import Response
 from rest_framework.viewsets import ModelViewSet
 
 from sponsor.models import Sponsor, SponsorLevel
 from sponsor.permissions import IsOwnerOrReadOnly, OwnerOnly
 from sponsor.serializers import (
-    SponsorLevelSerializer,
+    SponsorDetailSerializer,
     SponsorListSerializer,
     SponsorRemainingAccountSerializer,
     SponsorSerializer,
@@ -20,14 +21,14 @@
 class SponsorViewSet(ModelViewSet):
     serializer_class = SponsorSerializer
     permission_classes = [IsOwnerOrReadOnly]  # 본인 소유만 수정 가능
-    http_method_names = ["get", "post"]  # 지금은 조회/등록만 가능 TODO: 추후 수정 기능 추가
+    http_method_names = ["get", "post", "put"]
     validator = SponsorValidater()
 
     def get_queryset(self):
         return Sponsor.objects.all()
 
     def list(self, request, *args, **kwargs):
-        queryset = Sponsor.objects.filter(accepted=True).order_by("name")
+        queryset = Sponsor.objects.filter(accepted=True).order_by("level")
         serializer = SponsorListSerializer(queryset, many=True)
         return Response(serializer.data)
 
@@ -49,15 +50,15 @@ def create(self, request, *args, **kwargs):
         return Response(serializer.data)
 
     def retrieve(self, request, *args, **kwargs):
-        pk = kwargs["pk"]
+        pk = kwargs["id"]
         sponsor_data = get_object_or_404(Sponsor, pk=pk)
 
         # 본인 소유인 경우는 모든 필드
         # 그렇지 않은 경우는 공개 가능한 필드만 응답
         serializer = (
             SponsorSerializer(sponsor_data)
             if self.check_owner_permission(request, sponsor_data)
-            else SponsorListSerializer(sponsor_data)
+            else SponsorDetailSerializer(sponsor_data)
         )
 
         return Response(serializer.data)
@@ -67,17 +68,31 @@ def check_owner_permission(self, request, sponsor_data: Sponsor):
             self=Type[OwnerOnly], request=request, view=self, obj=sponsor_data
         )
 
+    def update(self, request, *args, **kwargs):
+        pk = kwargs["id"]
+        sponsor_data = get_object_or_404(Sponsor, pk=pk)
+        serializer = SponsorSerializer(sponsor_data, data=request.data)
+
+        # if not self.check_owner_permission(request, sponsor_data):
+        #     return Response(None, status.HTTP_401_UNAUTHORIZED)
+
+        if serializer.is_valid():
+            serializer.save()
+            return Response(serializer.data)
 
-class SponsorLevelViewSet(ModelViewSet):
-    serializer_class = SponsorLevelSerializer
+        return Response(serializer.errors, status.HTTP_400_BAD_REQUEST)
+
+
+class SponsorListViewSet(ModelViewSet):
+    serializer_class = SponsorListSerializer
     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)
+        serializer = SponsorListSerializer(queryset, many=True)
 
         return Response(serializer.data)