Django-restframework13 ViewSets

1.简介

• 路由系统是用于接收合适的请求，然后返回相应的响应
• REST框架允许将一系列相关的业务逻辑函数写在一起，称为ViewSet.在其他框架中，相似的实现也被称为’Resources’或者’Controllers’.
• 一个ViewSet类也是一个基础的View类，不提供任何请求处理函数，如get或post,只提供类似list和created这样的方法
• 一个ViewSet类，通过.as_view() method绑定到相应的url上

- 一般不需要在路由管理器中另外注册视图函数，只需要注册这个viewset类就够了，然后就会自动帮你处理，例如：

# 定义一个简单的list和retrieve视图from django.contrib.auth.models import Userfrom django.shortcuts import get_object_or_404from myapps.serializers import UserSerializerfrom rest_framework import viewsetsfrom rest_framework.response import Responseclass UserViewSet(viewsets.ViewSet):    """    A simple ViewSet for listing or retrieving users.    """    def list(self, request):        queryset = User.objects.all()        serializer = UserSerializer(queryset, many=True)        return Response(serializer.data)    def retrieve(self, request, pk=None):        queryset = User.objects.all()        user = get_object_or_404(queryset, pk=pk)        serializer = UserSerializer(user)        return Response(serializer.data)

如果我们需要可以提取出来成为两个单独的部分，需要传入参数，方法名：函数名

user_list = UserViewSet.as_view({'get': 'list'})user_detail = UserViewSet.as_view({'get': 'retrieve'})

一般我们使用Router类自动帮我们处理路由

from myapp.views import UserViewSetfrom rest_framework.routers import DefaultRouterrouter = DefaultRouter()# 默认使用model小写复数， base_name为小写单数router.register(r'users', UserViewSet, base_name='user')urlpatterns = router.urls

• ViewSet和View视图类的区别
  
  • 重复的逻辑函数应该被写进一个单独的类中，我们仅需要指定queryset一次，然后就可以被多个函数使用
  • 利用路由管理器，我们就不再需要亲自写url了
  • 所以使用URL confs更明确，有更多的控制， ViewSets编写速度更快，需要使url一致时，应该被使用。

2. 为路由添加额外的功能

REST框架默认提供了create/retrieve/update/destroy等操作

class UserViewSet(viewsets.ViewSet):    """    一个空的演示实例，将会被路由类处理    如果使用过了后缀，确保每一个函数后面有一个`format=None`参数    """    def list(self, request):        pass    def create(self, request):        pass    def retrieve(self, request, pk=None):        pass    def update(self, request, pk=None):        pass    def partial_update(self, request, pk=None):        pass    def destroy(self, request, pk=None):        pass

如果临时有一个方法需要被路由处理，可以使用@detail_route或 @list_route装饰器
@detail_route包含一个pk，此方法为操作单个实例的方法
@list_route，是操作一系列对象的方法，如：

from django.contrib.auth.models import Userfrom rest_framework import statusfrom rest_framework import viewsetsfrom rest_framework.decorators import detail_route, list_routefrom rest_framework.response import Responsefrom myapp.serializers import UserSerializer, PasswordSerializerclass UserViewSet(viewsets.ModelViewSet):    """    A viewset that provides the standard actions    """    queryset = User.objects.all()    serializer_class = UserSerializer    @detail_route(methods=['post'])    def set_password(self, request, pk=None):        user = self.get_object()        serializer = PasswordSerializer(data=request.data)        if serializer.is_valid():            user.set_password(serializer.data['password'])            user.save()            return Response({'status': 'password set'})        else:            return Response(serializer.errors,                            status=status.HTTP_400_BAD_REQUEST)    # 最近登录的用户    @list_route()    def recent_users(self, request):        recent_users = User.objects.all().order('-last_login')        page = self.paginate_queryset(recent_users)        # page为分页好的一些列对象        if page is not None:            serializer = self.get_serializer(page, many=True)            return self.get_paginated_response(serializer.data)        serializer = self.get_serializer(recent_users, many=True)        return Response(serializer.data)

此装饰器也能为路由视图添加额外的视图。

@detail_route(methods=['post'], permission_classes=[IsAdminOrIsSelf])def set_password(self, request, pk=None):   ...

装饰器默认处理get请求，但也通过methods参数接受其他请求方法

# 删除密码@detail_route(methods=['post', 'delete'])def unset_password(self, request, pk=None):   ...

这两个新的函数的路由地址为：
^users/{pk}/set_password/和users/pk/unsetpassword/

3. API参考

• ViewSet
  继承自APIView，利用permission_classes, authentication_classes等来控制API策略
  一般都需要重写此类中的方法
• GenericViewSet
  GenericViewSet继承自GenericAPIView，提供了一系列如get_object, get_queryset方法，如果使用此类，需要重写此类或者与mixin类进行组合
• ModelViewSet
  也是继承自GenericAPIView，通过mixin类，实现了一些具体的操作，.list(), .retrieve(), .create(), .update(), .partial_update(), 和 .destroy()。
  例如：

class AccountViewSet(viewsets.ModelViewSet):    """    A simple ViewSet for viewing and editing accounts.    """    queryset = Account.objects.all()    serializer_class = AccountSerializer    permission_classes = [IsAccountAdminOrReadOnly]

你可以对GenericAPIView提供的各种方法和属性进行重写。

class AccountViewSet(viewsets.ModelViewSet):    """    A simple ViewSet for viewing and editing the accounts    associated with the user.    """    serializer_class = AccountSerializer    permission_classes = [IsAccountAdminOrReadOnly]    # 动态的获取查询集，视图类中必须提供get_queryset方法或者queryset属性    def get_queryset(self):        return self.request.user.accounts.all()

注意：当我们删除ViewSet的queryset属性时， 任何与此相关的路由都自动无法获得数据库模型的base_name，所以在路由注册时，我们必须设置base_name参数。
虽然类视图默认提供了各种create/list/retrieve/update/destroy操作，你也可以通过permission_classes管理用户操作权限
- ReadOnlyModelViewSet
也是继承自GenericAPIView，仅提供只读操作，.list()和.retrieve().如：

class AccountViewSet(viewsets.ReadOnlyModelViewSet):    queryset = Account.objects.all()    serializer_class = AccountSerializer

4. 自定义个性化视图实例

from rest_framework import mixinsclass CreateListRetrieveViewSet(mixins.CreateModelMixin,                                mixins.ListModelMixin,                                mixins.RetrieveModelMixin,                                viewsets.GenericViewSet):    """    A viewset that provides `retrieve`, `create`, and `list` actions.    To use it, override the class and set the `.queryset` and    `.serializer_class` attributes.    """    pass