DRF视图的使用及源码流程分析

django rest framework中对于APIView、GenericAPIView、ModelViewSet、mixins扩展类的分析。

APIView

示例

根据实际程序来分析：

urls.py

urlpatterns = [
    re_path('users', UserAPIView.as_view())
]

views.py

class UserAPIView(APIView):
  
    def get(self, request):
        users = User.objects.filter().all()
        ser = UserSerializer(instance=users, many=True)
        return CommonResponse(status=status.HTTP_200_OK, data=ser.data)

源码分析

首先，UserAPIView继承了APIView，而APIView继承了View。

原生View里面定义了容许的http访问方式。

urls.py中，UserAPIView.as_view()这条语句其实执行的是APIView中的as_view()方法，而APIView中的as_view()方法执行父类View的as_view()方法。

APIView中的as_view()

View中的as_view()

其中关键的方法为dispatch方法，根据UserAPIView-->APIView-->View的继承顺序，该方法执行的是APIView中的dispatch方法。看下图。

dispatch方法会获取请求方式，判断是不是http容许的请求方式，若是是的话，则分发执行UserAPIView中对应的同名方法。

其实咱们也能够在UserAPIView中自定义dispatch方法，以下：

def dispatch(self, request, *args, **kwargs):
    func = getattr(self,request.method.lower())
    return func(request,*args,**kwargs)

总结：CBV基于反射实现根据请求方式不一样，执行不一样的方法。

View与APIView的执行流程

View：

1. as_view()是入口，获得view函数
2. 请求来了调用view函数，内部调用dispatch函数完成请求分发
3. dispatch函数将请求方式映射为视图类的同名方法，获得相应结果

APIView：

1. as_view()是入口，经过执行父类中的as_view方法获得view函数，而后在返回view函数的时候免除csrf验证。
2. 请求来了调用view函数，内部调用（APIView类中的）dispatch方法完成请求分发。
3. 在dispatch方法中，将会二次封装request，完成三大验证（认证、受权、节流），再将请求方式映射为视图类的同名方法，完成请求的处理。

GenericAPIView

执行流程

首先看看GenericAPIView的源码。

能够看出，GenericAPIView继承了APIView。而后还有几个比较重要的类属性，稍后用到再讲。

再往下看它对外暴露的方法：

实际上是比较少的，并且大可能是get_xxx之类的方法，也就是说，它的as_view()、dispatch等方法，其实都是按照APIView的流程处理的。

重要方法

首先咱们看get_queryset()，它的意思是批量查找数据库的数据:

• 能够看到，queryset属性必须进行赋值，按照继承顺序，咱们直接在本身定义的视图类中声明queryset属性并进行赋值。
• 在if判断中，若是queryset是一个QuerySet对象，就获取所有，获得一个QuerySetDict对象，不然原样返回。

QuerySetDict类型以下：

接下来看get_object()，它的意思是从数据库中获取单个对象:

• filter_kwargs封装查找条件，get_object_or_404负责在queryset中查找符合条件的对象(object)。
• 看到这里就发现了，默认查询条件是用pk，也就是说你的url中必需要用pk这个形参名进行分组捕获。不然就须要声明lookup_url_kwarg，若是url中的参数为name，那么lokup_url_kwarg="name"，而后进行替换组建filter_kwargs。固然若是你的查询条件不是用的pk，就须要修改lookup_field为字段名，如我不是按照pk进行查询，而是按照name，就修改lookup_field为name。
• 同时也会检查用户是否有权限查看该对象。check_object_permissions

主要的就是以上两个，get_serializer()会调用get_serializer_class获取序列化类，返回序列化类的执行结果。

示例

class UserAPIView(GenericAPIView):
    queryset = User.objects
    serializer_class = UserSerializer

    def get(self, request):
        users = self.get_queryset()
        ser = self.get_serializer(instance=users, many=True)
        return CommonResponse(status=status.HTTP_200_OK, data=ser.data)

mixins扩展类

• 在rest_framework.mixins中，有五个扩展类，分别是：
  • ListModelMixin：该类主要负责查询全部记录
  • RetrieveModelMixin：该类主要负责查询单条记录
  • CreateModelMixin：该类主要负责建立记录
  • UpdateModelMixin：该类主要负责对记录作更新操做
  • DestroyModelMixin：该类主要负责删除记录
• 这五个类都继承于object，是独立的子类。
• 针对GenericAPIView更高级的封装，配合GenericAPIView使用有奇效。

类	方法	描述
ListModelMixin	list()	查询全部，并返回Response对象
RetrieveModelMixin	retrieve()	查询单条，并返回Response对象
CreateModelMixin	create()	建立记录，并返回Response对象
UpdateModelMixin	update()	更新记录，并返回Response对象
DestroyModelMixin	destroy()	删除记录，并返回Response对象

示例

class UserAPIView(GenericAPIView, ListModelMixin):
    queryset = User.objects
    serializer_class = UserSerializer

    def get(self, request):
        return CommonResponse(status=status.HTTP_200_OK, data=self.list(request).data)

ModelViewSet

若是继承全部的mixins类，则视图类就会写的很是冗长，可读性较差。

同时，每次都要在视图中return，那咱们会考虑能不能简化这步操做。

首先导入ModelViewSet

from rest_framework.viewsets import ModelViewSet

查看它的源码：

再看GenericViewSet：

再看ViewSetMixin：

注释的意思是说，它重写了as_view()方法，绑定了http方法与视图函数中的方法，这让它必须接收一个actions参数，actions参数设置为一个字典。

示例

urls.py

urlpatterns = [
    re_path('users', UserAPIView.as_view(actions={"get": 'list'}))
]

views.py

class UserAPIView(ModelViewSet):
    queryset = User.objects
    serializer_class = UserSerializer

generics扩展类

至于generics扩展类，提供了如下几个功能：

类	功能
CreateAPIView	建立记录
DestroyAPIView	删除记录
UpdateAPIView	更新单条记录
ListAPIView	查询全部记录
RetrieveAPIView	查询单条记录
ListCreateAPIView	建立以及查询全部记录
RetrieveUpdateAPIView	更新以及查询单条记录
RetrieveDestroyAPIView	删除以及查询单条记录
RetrieveUpdateDestroyAPIView	删除、更新、查询单条记录

他们的实现也很简单，以`RetrieveDestroyAPIView`类举例：

因而可知，generics中的扩展类都是结合了mixins扩展类和GenericAPIView，而且提供了标准的http方法同名方法，这些方法又都分别执行对应的mixins扩展类中的方法。这和咱们mixins扩展类部分中的示例大致一致。

示例

urls.py

urlpatterns = [
    re_path('users', UserAPIView.as_view(actions={"get": 'list'}))
]

Views.py

class UserAPIView(ViewSetMixin, ListAPIView):
    queryset = User.objects
    serializer_class = UserSerializer

注：UserAPIView的as_view方法要接收actions参数，而咱们要使用的就是ViewSetMixin中重写的as_view方法，所以为了避免和ListAPIView中的as_view产生冲突，继承时要把ViewSetMixin放在前面。

看完上面的内容，下面再看这张图片，但愿能够帮助梳理本篇文章的内容：