美文网首页
11.Rest Framework视图集ViewSet

11.Rest Framework视图集ViewSet

作者: 小啊小狼 | 来源:发表于2020-09-22 10:07 被阅读0次

    一、与视图集配合使用的5个扩展类

    ListModelMixin:

    提供list(request, *args, **kwargs)方法快速实现列表视图,返回200状态码。该Mixin的list方法会对数据进行过滤和分页。

    CreateModelMixin:

    提供create(request, *args, **kwargs)方法快速实现创建资源的视图,成功返回201状态码。如果序列化器对前端发送的数据验证失败,返回400错误。

    RetrieveModelMixin:

    提供retrieve(request, *args, **kwargs)方法,可以快速实现返回一个存在的数据对象。如果存在,返回200, 否则返回404。

    UpdateModelMixin:

    提供update(request, *args, **kwargs)方法,可以快速实现更新一个存在的数据对象。同时也提供供 partial_update(request, *args, **kwargs)方法,更新部分字段,可以实现局部更新。成功返回200,序列化器校验数据失败时,返回400错误。

    DestroyModelMixin:

    提供destroy(request, *args, **kwargs)方法,可以快速实现删除一个存在的数据对象。成功返回204,不存在返回404。

    二、几个可用子类视图

    1) CreateAPIView

    提供 post 方法
    继承自: GenericAPIView、CreateModelMixin

    2)ListAPIView

    提供 get 方法
    继承自:GenericAPIView、ListModelMixin

    3)RetireveAPIView

    提供 get 方法
    继承自: GenericAPIView、RetrieveModelMixin

    4)DestoryAPIView

    提供 delete 方法
    继承自:GenericAPIView、DestoryModelMixin

    5)UpdateAPIView

    提供 put 和 patch 方法
    继承自:GenericAPIView、UpdateModelMixin

    三、视图集ViewSet

    使用视图集ViewSet,可以将一系列逻辑相关的动作放到一个类中:

    • list() 提供一组数据
    • retrieve() 提供单个数据
    • create() 创建数据
    • update() 保存数据
    • destory() 删除数据

    ViewSet继承自ViewSetMixin和views.APIView。因为ViewSetMixin里重写了as_view方法,所以ViewSet视图集类不再实现get()、post()等方法,而是实现动作 action 如 list() 、create() 等。视图集在使用as_view()方法的时候定义好action与请求方式的对应关系,才能将action动作与具体请求方式对应上。
    路由设置如下:

    urlpatterns = [
        # a.继承ViewSet之后,支持在定义路由时指定请求方法与action的映射
        # b.as_view需要接收一个字典
        # c.key为请求方法名,value为指定需要调用的action
        path('projects/', views.ProjectsViewSet.as_view({
            'get': 'list',
            'post': 'create'
        })),
        path('projects/names/', views.ProjectsViewSet.as_view({
            'get': 'names'
        })),
        path('projects/<int:pk>/', views.ProjectsViewSet.as_view({
            'get': 'retrieve',
            'put': 'update',
            'delete': 'destroy'
        })),
        path('projects/<int:pk>/interfaces/', views.ProjectsViewSet.as_view({
            'get': 'interfaces'
        })),
    ]
    

    action属性
    在视图集中,我们可以通过action对象属性来获取当前请求视图集时的action动作是哪个。例如:

        def get_serializer_class(self):
            if self.action == 'names':
                return ProjectsNamesModelSerializer
            elif self.action == 'interfaces':
                return InterfacesByProjectIdModelSerializer
    

    常用视图集父类

    1) ViewSet

    继承自ViewSetMixin和APIView,作用也与APIView基本类似,提供了身份认证、权限校验、流量管理等。
    在ViewSet中,没有提供任何动作action方法,需要我们自己实现action方法。

    2)GenericViewSet(ViewSetMixin, generics.GenericAPIView)

    继承自ViewSetMixin和GenericAPIView,作用也与GenericAPIVIew类似,提供了get_object、get_queryset等方法便于列表视图与详情信息视图的开发。

    ViewSetMixin重写了as_view方法,支持指定请求方法和action的映射

    GenericAPIView继承自APIView,拓展了对查询集的过滤排序和分页的实现,所以GenericViewSet实现了查询集的过滤、排序和分页的功能

    3)ModelViewSet

    继承自GenericAPIVIew,同时包括了ListModelMixin、RetrieveModelMixin、CreateModelMixin、UpdateModelMixin、DestoryModelMixin,支持create、list、retrieve、update、partial_update以及destroy这6个action

    4)ReadOnlyModelViewSet

    继承自GenericAPIVIew,同时包括了ListModelMixin、RetrieveModelMixin。

    几种ViewSet类的关系图:

    image.png

    视图集中定义附加action动作

    在视图集中,除了上述默认的方法动作外,还可以添加自定义动作。
    添加自定义动作需要使用rest_framework.decorators.action装饰器。

    以action装饰器装饰的方法名会作为action动作名,与list、retrieve等同。
    action装饰器可以接收两个参数:

    • methods: 该action支持的请求方式,列表传递,默认为get
    • detail: 表示是action中要处理的是否是视图资源的对象(即是否通过url路径获取主键)
      • True 表示使用通过URL获取的主键对应的数据对象
      • False 表示不使用URL获取主键
    from rest_framework import viewsets
    from rest_framework.decorators import action
    from .serializers import ProjectsModelSerializer
    
    class ProjectsViewSet(viewsets.ModelViewSet):
        queryset = Projects.objects.all()
        serializer_class = ProjectsModelSerializer
    
        @action(methods=['get'], detail=False)
        def names(self, request, *args, **kwargs):
            queryset = self.get_queryset()
            serializer = self.get_serializer(queryset, many=True)
            return Response(serializer.data)
    
        @action(detail=True)
        def interfaces(self, request, *args, **kwargs):
            response = self.retrieve(request, *args, **kwargs)
            response.data = response.data['interfaces']
            return Response(response.data)
    
        @action(methods=['post'], detail=True)
        def run(self, request, *args, **kwargs):
            # 取出并构造参数
            instance = self.get_object()
            response = super().create(request, *args, **kwargs)
            env_id = response.data.serializer.validated_data.get('env_id')
            ...
    

    对应的url设置(后续可用SimpleRouter模块自动生成路由表)

    from projects import views
    urlpatterns = [
    
        path('projects/names/', views.ProjectsViewSet.as_view({
            'get': 'names'
        })),
        path('projects/<int:pk>/interfaces/', views.ProjectsViewSet.as_view({
            'get': 'interfaces'
        })),
        path('projects/<int:pk>/run/', views.ProjectsViewSet.as_view({
            'post': 'run'
        })),
    ]
    

    相关文章

      网友评论

          本文标题:11.Rest Framework视图集ViewSet

          本文链接:https://www.haomeiwen.com/subject/mmuvyktx.html