후원사 기본 CRUD 기능 작성

Author: golony6449

pyconweb2022/pyconweb2022/urls.py

@@ -16,9 +16,15 @@
 from django.contrib import admin
 from django.urls import path, include
 
+import sponsor.routers
+
+
 urlpatterns = [
     path('admin/', admin.site.urls),
 
     # DRF
-    path('api-auth/', include('rest_framework.urls'))
+    path('api-auth/', include('rest_framework.urls')),
+
+    # App
+    path('sponsor/', include(sponsor.routers.get_router().urls))
 ]

pyconweb2022/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

pyconweb2022/sponsor/serializers.py

@@ -0,0 +1,15 @@
+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 = ['slug', 'name', 'level', 'desc', 'url', 'logo_image']

pyconweb2022/sponsor/viewsets.py

@@ -0,0 +1,32 @@
+from django.shortcuts import get_object_or_404
+from django.core.exceptions import PermissionDenied
+
+from rest_framework.viewsets import ModelViewSet
+from rest_framework.permissions import IsAuthenticatedOrReadOnly, AllowAny
+from rest_framework.response import Response
+
+from sponsor.serializers import SponsorSerializer, SponsorListSerializer
+from sponsor.models import Sponsor
+
+
+class SponsorViewSet(ModelViewSet):
+    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)    # 모든 절차가 완료된 후원사만 리스팅
+        serializer = SponsorListSerializer(queryset, many=True)
+        return Response(serializer.data)
+
+    def retrieve(self, request, *args, **kwargs):
+        pk = kwargs['pk']
+        sponsor_data = get_object_or_404(Sponsor, pk=pk)
+
+        if sponsor_data.creator != request.user:
+            raise PermissionDenied
+
+        serializer = SponsorSerializer(sponsor_data)
+        return Response(serializer.data)