Step4: 使用视图集简化代码

Author: zmrenwu

README.md

@@ -192,6 +192,7 @@ $ git clone https://github.com/HelloGitHub-Team/HelloDjango-REST-framework-tutor
 3. [初始化 RESTful API 风格的博客系统](https://www.zmrenwu.com/courses/django-rest-framework-tutorial/materials/92/)
 4. [实现博客首页文章列表 API](https://www.zmrenwu.com/courses/django-rest-framework-tutorial/materials/93/)
 5. [用类视图实现首页 API](https://www.zmrenwu.com/courses/django-rest-framework-tutorial/materials/94/)
+6. [使用视图集简化代码](https://www.zmrenwu.com/courses/django-rest-framework-tutorial/materials/95/)
 
 ## 公众号
 <p align="center">

blog/urls.py

@@ -11,5 +11,6 @@
     ),
     path("categories/<int:pk>/", views.CategoryView.as_view(), name="category"),
     path("tags/<int:pk>/", views.TagView.as_view(), name="tag"),
-    path("api/index/", views.IndexPostListAPIView.as_view()),
+    # path("api/index/", views.IndexPostListAPIView.as_view()),
+    path("api/index/", views.index),
 ]

blog/views.py

@@ -2,7 +2,7 @@
 from django.db.models import Q
 from django.shortcuts import get_object_or_404, redirect, render
 from django.views.generic import DetailView, ListView
-from rest_framework import status
+from rest_framework import mixins, status, viewsets
 from rest_framework.decorators import api_view
 from rest_framework.generics import ListAPIView
 from rest_framework.pagination import PageNumberPagination
@@ -72,3 +72,12 @@ class IndexPostListAPIView(ListAPIView):
     queryset = Post.objects.all()
     pagination_class = PageNumberPagination
     permission_classes = [AllowAny]
+
+
+class PostViewSet(mixins.ListModelMixin, viewsets.GenericViewSet):
+    serializer_class = PostListSerializer
+    queryset = Post.objects.all()
+    permission_classes = [AllowAny]
+
+
+index = PostViewSet.as_view({"get": "list"})

blogproject/urls.py

@@ -18,8 +18,11 @@
 from rest_framework import routers
 
 from blog.feeds import AllPostsRssFeed
+import blog.views
 
 router = routers.DefaultRouter()
+router.register(r"posts", blog.views.PostViewSet, basename="post")
+
 urlpatterns = [
     path("admin/", admin.site.urls),
     path("search/", include("haystack.urls")),