Merge pull request #104 from golony6449/devdev

카테고리 별로 조회할 수 있는 API 작성

Author: golony6449

pyconweb2022/program/admin.py

@@ -9,7 +9,10 @@
 
 class ProgramCategoryAdmin(ImportExportModelAdmin):
     resource_class = ProgramCategoryResource
-    list_display = ("name",)
+    list_display = (
+        "id",
+        "name",
+    )
 
 
 admin.site.register(ProgramCategory, ProgramCategoryAdmin)

pyconweb2022/program/routers.py

@@ -5,6 +5,7 @@
     ProposalDay1ViewSet,
     ProposalDay2ViewSet,
     ProposalDetailViewSet,
+    CategorySessionViewSet,
 )
 
 
@@ -13,6 +14,7 @@ def get_router():
     router.register("list/day1", ProposalDay1ViewSet, basename="program-day1")
     router.register("list/day2", ProposalDay2ViewSet, basename="program-day2")
     router.register("list", ProposalViewSet, basename="program")
-    router.register("edit", ProposalDetailViewSet, basename="program-edit")
+    # router.register("edit", ProposalDetailViewSet, basename="program-edit")
+    router.register("category", CategorySessionViewSet, basename="session-in-category")
 
     return router

pyconweb2022/program/viewsets.py

@@ -1,8 +1,8 @@
-from rest_framework.viewsets import ReadOnlyModelViewSet, ModelViewSet
+from rest_framework.viewsets import ReadOnlyModelViewSet, ViewSet
 from rest_framework.permissions import IsAuthenticated, AllowAny
 from rest_framework.response import Response
 
-from program.models import Proposal
+from program.models import Proposal, ProgramCategory
 from program.serializers import ProposalSerializer, ProposalDetailSerializer
 from pyconweb2022 import config
 
@@ -35,3 +35,22 @@ def retrieve(self, request, *args, **kwargs):
 
         serializer = self.get_serializer(queryset, many=False)
         return Response(serializer.data)
+
+
+class CategorySessionViewSet(ProposalViewSet):
+    queryset = Proposal.objects.none()
+
+    def retrieve(self, request, *args, **kwargs):
+        queryset = self.get_queryset()
+        serializer = self.get_serializer(queryset, many=True)
+
+        return Response(serializer.data)
+
+    def get_queryset(self):
+        pk = self.kwargs.get("pk")
+        target_category = ProgramCategory.objects.get(id=pk)
+        sessions = Proposal.objects.filter(category=target_category).order_by(
+            "video_open_at"
+        )
+
+        return sessions