介紹ViewSet

• list() 提供一組數據
• retrieve() 提供單個數據
• create() 創建數據
• update() 保存數據
• destory() 刪除數據

ViewSet視圖集不再實現get()，post()等方法，而是實現動作action如list(),create()等。視圖集只在使用as_view()方法的時候，才會將action動作與具體請求方式對應上。

視圖集存在的意義：

1. 將所有的接口可以寫在一個類試圖中（類試圖中的方法不再用請求方法名小寫 行為/動作 list/create）
2. 重寫了as_view({‘get’:‘list’, ‘post’:‘create’})

使用ViewSet開發api案例

1.views.py

'''使用試圖集的ViewSet開發api'''

class BookViewSet(ViewSet):
    def list(self, request):
        qs = BookInfo.objects.all()
        serializer = BookInfoModelSerializer(instance=qs, many=True)
        return Response(serializer.data)

    def retrieve(self, request, pk):
        try:
            book = BookInfo.objects.get(id=pk)
        except BookInfo.DoesNotExist:
            Response(status=status.HTTP_404_NOT_FOUND)
        serializer = BookInfoModelSerializer(instance=book)
        return Response(serializer.data)

2.urls.py

from django.conf.urls import url
from . import views

# 視圖集路由
    url(r'^books/$', views.BookViewSet.as_view({'get':'list'})),
    url(r'^books/(?P<pk>\d+)$', views.BookViewSet.as_view({'get':'retrieve'}))

使用使用GenericViewSet + Mixin開發api案例

1.views.py

from booktest.models import BookInfo,HeroInfo
from .serializers import BookInfoModelSerializer,HeroInfoModelSerializer
from rest_framework.mixins import ListModelMixin,CreateModelMixin,DestroyModelMixin,UpdateModelMixin,RetrieveModelMixin
from rest_framework.viewsets import ViewSet,GenericViewSet,ReadOnlyModelViewSet,ModelViewSet

'''使用GenericViewSet + Mixin開發api'''
class BookViewSet( DestroyModelMixin, CreateModelMixin, UpdateModelMixin, ReadOnlyModelViewSet):
    queryset = BookInfo.objects.all()
    serializer_class =  BookInfoModelSerializer

附加viewsset的部分源碼，其實ReadOnlyModelViewSet 包含了 GenericViewSet 與 ListModelMixin 與 RetrieveModelMixin

class ViewSet(ViewSetMixin, views.APIView):
    """
    The base ViewSet class does not provide any actions by default.
    """
    pass


class GenericViewSet(ViewSetMixin, generics.GenericAPIView):
    """
    The GenericViewSet class does not provide any actions by default,
    but does include the base set of generic view behavior, such as
    the `get_object` and `get_queryset` methods.
    """
    pass


class ReadOnlyModelViewSet(mixins.RetrieveModelMixin,
                           mixins.ListModelMixin,
                           GenericViewSet):
    """
    A viewset that provides default `list()` and `retrieve()` actions.
    """
    pass

2. urls.py

from django.conf.urls import url
from . import views
# 視圖集路由
url(r'^books/$', views.BookViewSet.as_view({'get':'list', 'post': 'retrieve'})),
url(r'^books/(?P<pk>\d+)$', views.BookViewSet.as_view({'get':'retrieve', 'delete':'destroy', 'put':'update'}))

最牛逼的一種開發api方式(使用ModelViewSet + 自動注冊路由)

1. views.py

'''最叼繼承 ModelViewSet'''
class BookViewSet( ModelViewSet):
    queryset = BookInfo.objects.all()
    serializer_class =  BookInfoModelSerializer

2. urls.py

from rest_framework.routers import DefaultRouter
urlpatterns = [
    
]
router = DefaultRouter()   # 創建路由器
router.register(r'book', views.BookViewSet)  #注冊路由
urlpatterns += router.urls

自定義視圖（自己定義的邏輯）

1.views.py

@action(methods=[指定下面的行為接受什么請求], detail=是不是詳情視圖，詳情視圖為True，生成路由books/pk/latest,列表視圖為False，生成路由books/latest)

from booktest.models import BookInfo,HeroInfo
from .serializers import BookInfoModelSerializer,HeroInfoModelSerializer
from rest_framework.viewsets import ViewSet,GenericViewSet,ReadOnlyModelViewSet,ModelViewSet

# 用于urls.py自動創建自定義視圖的路由
from rest_framework.decorators import action
class BookViewSet( ModelViewSet):
    queryset = BookInfo.objects.all()
    serializer_class =  BookInfoModelSerializer

    @action(methods=['get'], detail=False)
    def latest(self, request):
        book = BookInfo.objects.latest('id')
        serializer = self.get_serializer(book)
        return Response(serializer.data)

    @action(methods=['put'], detail=True)
    def read(self, request, pk):
        '''修改圖書的數量'''
        book = self.get_object()
        book.bread = request.data.get('bread')
        book.save()
        serializer = self.get_serializer(book)
        return  Response(serializer.data)

2.urls.py

創建路由器(！！！路由器只可以結合試圖集一塊用！！！)默認只生成標準的增刪改查，如果想生成自定義的路由需要在views.py的自定義視圖上添加裝飾器action（from rest_framework.decorators import action）
！！！路由器只可以結合試圖集一塊用！！！
！！！路由器只可以結合試圖集一塊用！！！
！！！路由器只可以結合試圖集一塊用！！！

from rest_framework.routers import DefaultRouter,SimpleRouter
urlpatterns = [
    
]
router = DefaultRouter()   # 創建路由器
router.register(r'book', views.BookViewSet, base_name='bookinfo')  #注冊路由
urlpatterns += router.urls

DefaultRouter與SimpleRouter的區別： DefaultRouter可以自動生成根視圖不報錯

謝謝觀看~