当前位置:

首页 > temp > python入门教程 >

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

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：

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

APIView：

as_view()是入口，通过执行父类中的as_view方法得到view函数，然后在返回view函数的时候免除csrf验证。
请求来了调用view函数，内部调用（APIView类中的）dispatch方法完成请求分发。
在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	删除、更新、查询单条记录