美文网首页
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