DRF - 序列化组件（GET/PUT/DELETE接口设计）、视图优化组件

一、序列化组件

基于上篇随笔的表结构 , 通过序列化组件的ModelSerializer设计如下三个接口 :

GET       127.0.0.1:8000/books/{id}    # 获取一条数据，返回值：{}
PUT       127.0.0.1:8000/books/{id}    # 修改数据，返回值：{}
DELETE    127.0.0.1:8000/books/{id}    # 删除数据，返回空

urls.py文件：

from django.urls import path, re_path
from serializer import views
urlpatterns = [
    re_path('books/(\d+)/$', views.BookFilterView.as_view())
]

Views.py文件：

class BookFilterView(APIView):
    def get(self, request, nid):
        book_obj = Book.objects.get(pk=nid)
        serialized_data = BookSerializer(book_obj, many=False)
        return Response(serialized_data.data)

    def put(self, request, nid):
        book_obj = Book.objects.get(pk=nid)
        verified_data = BookSerializer(data=request.data, instance=book_obj, many=False)
        if verified_data.is_valid():
            verified_data.save()
            return Response(verified_data.data)
        else:
            return Response(verified_data.errors)

    def delete(self, request, nid):
        Book.objects.get(pk=nid).delete()
        return Response()

二、视图组件引入

前面的介绍中，我们已经通过序列化组件设计除了符合REST规范的五个常用接口，现在假设，我们有多个数据接口，比如（Book，Author，Publish...）等数据表都需要定义类似的接口，可以预见，我们需要重复定义类似上面的五个接口，这种方式将会导致大量的重复代码，显然，我们的程序还有很多需要优化的地方，那么，如果是你，将会如何进行优化呢？

　　首先回顾以下混入类和多继承的知识，有如下一个Animal类：

class Animal(object):
    def eat(self):
        print("Eating")

    def sleepping(self):
        print("sleepping")

class Bird(Animal):
　　def flying(self):
    　　print("flying")

class Bee(Animal):
　　def flying(self):
    　　print("flying")

此时，我们可以改为如下代码（python支持多继承，不是所有语言都有这个特点）：

class Animal(object):
    def eat(self):
        print("Eating")

    def sleepping(self):
        print("sleepping")

class Flying(object):
    def flying(self):
    　　print("flying")

class Bird(Animal, Flying):
　　pass

class Bee(Animal, Flying):
　　pass

看到这里，结合上面的回顾的混合类和多继承，我们是否可以使用下面的方式优化之前的接口设计呢？

class GetAllData():
    def get(self, request):pass

class GetOneData():
    def get(self, request):pass

class DeleteOneData():
    def delete(self, request):pass

class UpdateOneData():
    def put(self, request):pass

class CreateData():
    def post(self, request):pass

class BookView(APIView, GetAllData, CreateData):pass

class BookFilterView(APIView, GetOneData, DeleteOneData, UpdateOneData):pass

像上面代码这样，将每个接口都写到独立的类中，然后使用多继承，或者成为mixin的这种方式，就可以对我们的程序进行优化，mixin的方式非常常见，在网络编程中学过的socketserver，其源码中就有对mixin的实现，即，假设我们需要进程的时候，我们继承进程类，如果我们需要线程的时候，我们就继承线程类即可。

三、视图组件使用

视图组件是用来优化接口逻辑的。

1、使用视图组件的mixin进行接口逻辑优化，上面五个接口可以改写如下：

　　urls.py代码：

from django.urls import re_path
from serializer import views

urlpatterns = [
    re_path('books/$', views.BookView.as_view()),
    re_path('books/(?P<pk>\d+)/$', views.BookFilterView.as_view()),
]

视图views.py代码：

from rest_framework.mixins import (
    ListModelMixin,
    CreateModelMixin,
    DestroyModelMixin,
    UpdateModelMixin,
    RetrieveModelMixin
)
from rest_framework.generics import GenericAPIView
from rest_framework.response import Response
# 导入序列化类
from .app_serializers import BookSerializer
from .models import Book, Publish, Author

class BookView(ListModelMixin, CreateModelMixin, GenericAPIView):
    # queryset和serializer_class是固定写法
    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 BookFilterView(RetrieveModelMixin, UpdateModelMixin, DestroyModelMixin, GenericAPIView):
    queryset = Book.objects.all()
    serializer_class = BookSerializer
    def get(self, request, *args, **kwargs):
        return self.retrieve(request, *args, **kwargs)

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

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

注意 :  单挑数据操作的url有变化 , 因为我们在视图中, 统一传的都是queryset , 所以, 需要通过传入一个名为pk 的命名参数 , 告诉视图组件 , 用户需要操作的具体数据,

2, 使用视图组件的genericview 进行接口逻辑优化

　　上面的代码看似已经优化的非常完美了，但是，在一个对性能要求极高的项目里面，我们的程序还可以继续优化，不断优化程序是每个程序员必备的技能，也是帮助我们成长的重要手段。同样的思路，同样的方法，我们可以将多个接口封装到一个功能类中，如下代码：

from rest_framework import generics

from .app_serializers import BookSerializer
from .models import Book, Publish, Author

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

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

3、使用视图组件的viewset进行接口逻辑优化

　　上面的代码已经看似非常完美了，但是，你有没有发现还有重复代码，该如何改进呢？使用viewset可以进一步优化，如下：

　　urls.py文件（注意跟之前有什么不同）：

from django.urls import re_path
from serializer import views

urlpatterns = [
    re_path('books/$', views.BookView.as_view({
        'get': 'list',
        'post': 'create'
    })),
    re_path('books/(?P<pk>\d+)/$', views.BookView.as_view({
        'get': 'retrieve',
        'put': 'update',
        'delete': 'destroy'
    }))

视图views.py部分：

from rest_framework.viewsets import ModelViewSet
from .app_serializers import BookSerializer
from .models import Book, Publish, Author

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

使用方式是不是很简单，接下来去看以下源码都为我们做了什么吧！其实整个viewset优化方案最重要的地方就是urls.py中传入了参数，然后对参数进行映射关系绑定。