Django Rest Framework 視圖和路由

DRF的視圖

APIView

我們django中寫CBV的時候繼承的是View，rest_framework繼承的是APIView，那么他們兩個有什么不同呢~~~

urlpatterns = [
    url(r'^book$', BookView.as_view()),
    url(r'^book/(?P<id>\d+)$', BookEditView.as_view()),
]

我們可以看到，不管是View還是APIView最開始調用的都是as_view()方法~~那我們走進源碼看看~~

我們能看到，APIView繼承了View, 並且執行了View中的as_view()方法，最后把view返回了，用csrf_exempt()方法包裹后去掉了csrf的認證。

那我們看看View中的as_view()方法做了什么~

我們看到了~在View中的as_view方法返回了view函數，而view函數執行了self.dispatch()方法~~但是這里的dispatch方法應該是我們APIView中的~~

我們去initialize_request中看下把什么賦值給了request，並且賦值給了self.request, 也就是我們在視圖中用的request.xxx到底是什么~~

我們看到，這個方法返回的是Request這個類的實例對象~~我們注意我們看下這個Request類中的第一個參數request，是我們走我們django的時候的原來的request~

我們看到了，這個Request類把原來的request賦值給了self._request, 也就是說以后_request是我們老的request，新的request是我們這個Request類~~

那我們繼承APIView之后請求來的數據都在哪呢~~

我們用了rest_framework框架以后，我們的request是重新封裝的Request類~

request.query_params 存放的是我們get請求的參數

request.data 存放的是我們所有的數據，包括post請求的以及put，patch請求~~~

相比原來的django的request，我們現在的request更加精簡，清晰了~~~

現在我們知道了APIView和View的一些區別~~當然還有~~后面我們還會說~~

我們寫的視圖可能對多個表進行增刪改查，就導致我們的視圖特別多重復的代碼~~

那么我們嘗試着來進行封裝一下~~

第一次封裝

class BookView(APIView):

    def get(self, request):
        query_set = Book.objects.all()
        book_ser = BookSerializer(query_set, many=True)
        return Response(book_ser.data)

    def post(self, request):
        query_set = request.data
        book_ser = BookSerializer(data=query_set)
        if book_ser.is_valid():
            book_ser.save()
            return Response(book_ser.validated_data)
        else:
            return Response(book_ser.errors)


class BookEditView(APIView):

    def get(self, request, id):
        query_set = Book.objects.filter(id=id).first()
        book_ser = BookSerializer(query_set)
        return Response(book_ser.data)

    def patch(self, request, id):
        query_set = Book.objects.filter(id=id).first()
        book_ser = BookSerializer(query_set, data=request.data, partial=True)
        if book_ser.is_valid():
            book_ser.save()
            return Response(book_ser.validated_data)
        else:
            return Response(book_ser.errors)

    def delete(self, request, id):
        query_set = Book.objects.filter(id=id).first()
        if query_set:
            query_set.delete()
            return Response("")
        else:
            return Response("刪除的書籍不存在")

class GenericAPIView(APIView):
    queryset = None
    serializer_class = None

    def get_queryset(self):
        return self.queryset.all()

    def get_serializer(self, *args, **kwargs):
        return self.serializer_class(*args, **kwargs)


class ListModelMixin(object):
    def list(self, request, *args, **kwargs):
        queryset = self.get_queryset()
        serializer = self.get_serializer(queryset, many=True)
        return Response(serializer.data)


class CreateModelMixin(object):
    def create(self, request, *args, **kwargs):
        serializer = self.get_serializer(data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response(serializer.validated_data)
        else:
            return Response(serializer.errors)


class RetrieveModelMixin(object):
    def retrieve(self, request, id, *args, **kwargs):
        book_obj = self.get_queryset().filter(pk=id).first()
        book_ser = self.get_serializer(book_obj)
        return Response(book_ser.data)


class UpdateModelMixin(object):
    def update(self, request, id, *args, **kwargs):
        book_obj = self.get_queryset().filter(pk=id).first()
        book_ser = self.get_serializer(book_obj, data=request.data, partial=True)
        if book_ser.is_valid():
            book_ser.save()
            return Response(book_ser.validated_data)
        else:
            return Response(book_ser.errors)


class DestroyModelMixin(object):
    def destroy(self, request, id, *args, **kwargs):
        queryset = self.get_queryset()
        try:
            queryset.get(pk=id).delete()
            return Response("")
        except Exception as e:
            return Response("信息有誤")
# 我們把公共的部分抽出來 這樣不管寫多少表的增刪改查都變的很簡單
# 這樣封裝后我們的視圖會變成這樣

class BookView(GenericAPIView, ListModelMixin, CreateModelMixin):
    queryset = Book.objects.all()
    serializer_class = BookSerializer

    def get(self, request, *args, **kwargs):
        return self.list(request, *args, **kwargs)

    def post(self, request, *args, **kwargs):
        return self.create(request, *args, **kwargs)


class BookEditView(GenericAPIView, RetrieveModelMixin, UpdateModelMixin, DestroyModelMixin):
    queryset = Book.objects.all()
    serializer_class = BookSerializer

    def get(self, request, id, *args, **kwargs):
        return self.retrieve(request, id, *args, **kwargs)

    def patch(self, request, id, *args, **kwargs):
        return self.update(request, id, *args, **kwargs)

    def destroy(self, request, id, *args, **kwargs):
        return self.delete(request, id, *args, **kwargs)

我們封裝的GenericAPIView，包括封裝每個方法的類，其實框架都幫我們封裝好了~~

我們可以直接繼承這些類~~來實現上面的視圖~~可是還有沒有更簡單的方法呢~我們再次封裝一下~~

第二次封裝

# 上面我們寫的繼承類太長了~~我們再改改

class ListCreateAPIView(GenericAPIView, ListModelMixin, CreateModelMixin):
    pass


class RetrieveUpdateDestroyAPIView(GenericAPIView, RetrieveModelMixin, UpdateModelMixin, DestroyModelMixin):
    pass


class BookView(ListCreateAPIView):
    queryset = Book.objects.all()
    serializer_class = BookSerializer

    def get(self, request, *args, **kwargs):
        return self.list(request, *args, **kwargs)

    def post(self, request, *args, **kwargs):
        return self.create(request, *args, **kwargs)


class BookEditView(RetrieveUpdateDestroyAPIView):
    queryset = Book.objects.all()
    serializer_class = BookSerializer

    def get(self, request, id, *args, **kwargs):
        return self.retrieve(request, id, *args, **kwargs)

    def patch(self, request, id, *args, **kwargs):
        return self.update(request, id, *args, **kwargs)

    def delete(self, request, id, *args, **kwargs):
        return self.delete(request, id, *args, **kwargs)

這次我們只是讓繼承變的簡單了一點而已，好像並沒有什么大的進步~~

我們可不可以把這兩個視圖合並成一個視圖呢~~~框架給我們提供了一個路由傳參的方法~~

我們看下ViewSetMixin

actions這個默認參數其實就是我們路由可以進行傳參了~~~

下面這個循環~可以看出~我們要傳的參數是一個字段~key應該是我們的請求方式，value應該對應我們處理的方法~

這樣我們每個視圖就不用在寫函數了~因為已經和內部實現的函數相對應了~

第三次封裝

urlpatterns = [
    # url(r'^book$', BookView.as_view()),
    # url(r'^book/(?P<id>\d+)$', BookEditView.as_view()),
    url(r'^book$', BookView.as_view({"get": "list", "post": "create"})),
    url(r'^book/(?P<pk>\d+)$', BookView.as_view({"get": "retrieve", "patch": "update", "delete": "destroy"})),
]

from rest_framework.viewsets import ViewSetMixin


# class BookView(ViewSetMixin, ListCreateAPIView, RetrieveUpdateDestroyAPIView):
#     queryset = Book.objects.all()
#     serializer_class = BookSerializer
    
    
# 如果我們再定義一個類
class ModelViewSet(ViewSetMixin, ListCreateAPIView, RetrieveUpdateDestroyAPIView):
    pass


class BookView(ModelViewSet):
    queryset = Book.objects.all()
    serializer_class = BookSerializer

 我們現在的視圖就只要寫兩行就可以了~~~

其實我們寫的所有的視圖~框架都幫我們封裝好了~

注意一點~~用框架封裝的視圖~我們url上的那個關鍵字參數要用pk~~系統默認的~~

奉獻一張圖來看下我們的繼承順序~~~

DRF的路由

我們上面的路由傳參寫的特別多~~框架也幫我們封裝好了~

from .views import BookView
from rest_framework.routers import DefaultRouter

router = DefaultRouter()
router.register(r"book", BookView)

urlpatterns = [
    # url(r'^book$', BookView.as_view()),
    # url(r'^book/(?P<id>\d+)$', BookEditView.as_view()),
    # url(r'^book$', BookView.as_view({"get": "list", "post": "create"})),
    # url(r'^book/(?P<pk>\d+)$', BookView.as_view({"get": "retrieve", "patch": "update", "delete": "destroy"})),

]
urlpatterns += router.urls

我們可以看到~~通過框架我們可以把路由視圖都變的非常簡單~~

但是需要自定制的時候還是需要我們自己用APIView寫~~當不需要那么多路由的時候~也不要用這種路由注冊~~

總之~~一切按照業務需要去用~~~