Feature: Add sponser viewset

Author: oleveloper

pyconkr/urls.py

@@ -16,9 +16,11 @@
 from django.contrib import admin
 from django.urls import include, path
 
+import sponsor.routers
+
 urlpatterns = [
     path("api-auth/", include("rest_framework.urls")),
     path("summernote/", include("django_summernote.urls")),
     path("admin/", admin.site.urls),
-    path("sponsors/", include("sponsor.urls")),
+    path("sponsors/", include(sponsor.routers.get_router().urls)),
 ]

sponsor/routers.py

@@ -0,0 +1,10 @@
+from rest_framework.routers import DefaultRouter
+
+from sponsor.viewsets import *
+
+
+def get_router():
+    router = DefaultRouter()
+    router.register("", SponsorViewSet, basename="sponsor")
+
+    return router

sponsor/serializers.py

@@ -0,0 +1,23 @@
+from rest_framework.serializers import ModelSerializer
+
+from sponsor.models import Sponsor
+
+
+class SponsorSerializer(ModelSerializer):
+    class Meta:
+        model = Sponsor
+        fields = "__all__"
+
+
+class SponsorListSerializer(ModelSerializer):
+    class Meta:
+        model = Sponsor
+        fields = [
+            "name",
+            "level",
+            "desc",
+            "eng_desc",
+            "url",
+            "logo_image",
+            "id",
+        ]

sponsor/viewsets.py

@@ -0,0 +1,37 @@
+from django.shortcuts import get_object_or_404
+
+from rest_framework.viewsets import ReadOnlyModelViewSet
+from rest_framework.permissions import IsAuthenticatedOrReadOnly
+from rest_framework.response import Response
+
+from sponsor.serializers import (
+    SponsorSerializer,
+    SponsorListSerializer,
+)
+from sponsor.models import Sponsor
+
+
+class SponsorViewSet(ReadOnlyModelViewSet):
+    serializer_class = SponsorSerializer
+    permission_classes = [IsAuthenticatedOrReadOnly]  # 로그인된 사용자에게만 허용
+
+    def get_queryset(self):
+        return Sponsor.objects.all()
+
+    def list(self, request, *args, **kwargs):
+        queryset = Sponsor.objects.filter(accepted=True).order_by("name")
+        serializer = SponsorListSerializer(queryset, many=True)
+        return Response(serializer.data)
+
+    def create(self, request, *args, **kwargs):
+        serializer = self.get_serializer(data=request.data)
+        serializer.is_valid(raise_exception=True)
+        serializer.save()
+        return Response(serializer.data)
+
+    def retrieve(self, request, *args, **kwargs):
+        pk = kwargs["pk"]
+        sponsor_data = get_object_or_404(Sponsor, pk=pk)
+
+        serializer = SponsorSerializer(sponsor_data)
+        return Response(serializer.data)