04 drf-视图组件 - 图1


一 视图

Django REST framwork 提供的视图的主要作用:

  • 控制序列化器的执行(检验、保存、转换数据)
  • 控制数据库查询的执行

REST framework 提供了众多的通用视图基类与扩展类,以简化视图的编写。

1 两个视图基类


1.1 APIView

  1. rest_framework.views.APIView

APIView是REST framework提供的所有视图的基类,继承自Django的View父类。
APIViewView的不同之处在于:

  • 传入到视图方法中的是REST framework的Request对象,而不是Django的HttpRequeset对象;
  • 视图方法可以返回REST framework的Response对象,视图会为响应数据设置(render)符合前端要求的格式;
  • 任何APIException异常都会被捕获到,并且处理成合适的响应信息;
  • 在进行dispatch()分发前,会对请求进行身份认证、权限检查、流量控制。

支持定义的类属性

  • authentication_classes 列表或元祖,身份认证类
  • permissoin_classes 列表或元祖,权限检查类
  • throttle_classes 列表或元祖,流量控制类

APIView中仍以常规的类视图定义方法来实现get() 、post() 或者其他请求方式的方法。
举例:

  1. from rest_framework.views import APIView
  2. from rest_framework.response import Response
  3. # url(r'^students/$', views.StudentsAPIView.as_view()),
  4. class StudentsAPIView(APIView):
  5. def get(self, request):
  6. data_list = Student.objects.all()
  7. serializer = StudentModelSerializer(instance=data_list, many=True)
  8. return Response(serializer.data)

1.2 GenericAPIView[通用视图类]

  1. rest_framework.generics.GenericAPIView

继承自APIVIew主要增加了操作序列化器和数据库查询的方法,作用是为下面Mixin扩展类的执行提供方法支持。通常在使用时,可搭配一个或多个Mixin扩展类。
提供的关于序列化器使用的属性与方法

  • 属性:
    • serializer_class 指明视图使用的序列化器
  • 方法:

    • get_serializer_class(self)
      当出现一个视图类中调用多个序列化器时,那么可以通过条件判断在get_serializer_class方法中通过返回不同的序列化器类名就可以让视图方法执行不同的序列化器对象了。
      返回序列化器类,默认返回serializer_class,可以重写,例如:

      1. def get_serializer_class(self):
      2. if self.request.user.is_staff:
      3. return FullAccountSerializer
      4. return BasicAccountSerializer
    • get_serializer(self,_args, _kwargs)返回序列化器对象,主要用来提供给Mixin扩展类使用,如果我们在视图中想要获取序列化器对象,也可以直接调用此方法。*注意,该方法在提供序列化器对象的时候,会向序列化器对象的context属性补充三个数据:request、format、view,这三个数据对象可以在定义序列化器时使用。

      • request 当前视图的请求对象
      • view 当前请求的类视图对象
      • format 当前请求期望返回的数据格式

提供的关于数据库查询的属性与方法

  • 属性:
    • queryset 指明使用的数据查询集
  • 方法:

    • get_queryset(self)
      返回视图使用的查询集,主要用来提供给Mixin扩展类使用,是列表视图与详情视图获取数据的基础,默认返回queryset属性,可以重写,例如:

      1. def get_queryset(self):
      2. user = self.request.user
      3. return user.accounts.all()
    • get_object(self)

返回详情视图所需的模型类数据对象,主要用来提供给Mixin扩展类使用。
在试图中可以调用该方法获取详情信息的模型类对象。
若详情访问的模型类对象不存在,会返回404。
该方法会默认使用APIView提供的check_object_permissions方法检查当前对象是否有权限被访问。
举例:

  1. # url(r'^books/(?P<pk>\d+)/$', views.BookDetailView.as_view()),
  2. class BookDetailView(GenericAPIView):
  3. queryset = BookInfo.objects.all()
  4. serializer_class = BookInfoSerializer
  5. def get(self, request, pk):
  6. book = self.get_object() # get_object()方法根据pk参数查找queryset中的数据对象
  7. serializer = self.get_serializer(book)
  8. return Response(serializer.data)

其他可以设置的属性

  • pagination_class 指明分页控制类
  • filter_backends 指明过滤控制后端

为了方便学习上面的GenericAPIView通用视图类,我们新建一个子应用。

  1. python manage.py startapp gen

代码:

  1. from rest_framework.generics import GenericAPIView
  2. from students.models import Student
  3. from .serializers import StudentModelSerializer, StudentModel2Serializer
  4. from rest_framework.response import Response
  5. class StudentsGenericAPIView(GenericAPIView):
  6. # 本次视图类中要操作的数据[必填]
  7. queryset = Student.objects.all()
  8. # 本次视图类中要调用的默认序列化器[玄天]
  9. serializer_class = StudentModelSerializer
  10. def get(self, request):
  11. """获取所有学生信息"""
  12. serializer = self.get_serializer(instance=self.get_queryset(), many=True)
  13. return Response(serializer.data)
  14. def post(self,request):
  15. data = request.data
  16. serializer = self.get_serializer(data=data)
  17. serializer.is_valid(raise_exception=True)
  18. instance = serializer.save()
  19. serializer = self.get_serializer(instance=instance)
  20. return Response(serializer.data)
  21. class StudentGenericAPIView(GenericAPIView):
  22. queryset = Student.objects.all()
  23. serializer_class = StudentModelSerializer
  24. def get_serializer_class(self):
  25. """重写获取序列化器类的方法"""
  26. if self.request.method == "GET":
  27. return StudentModel2Serializer
  28. else:
  29. return StudentModelSerializer
  30. # 在使用GenericAPIView视图获取或操作单个数据时,视图方法中的代表主键的参数最好是pk
  31. def get(self,request,pk):
  32. """获取一条数据"""
  33. serializer = self.get_serializer(instance=self.get_object())
  34. return Response(serializer.data)
  35. def put(self,request,pk):
  36. data = request.data
  37. serializer = self.get_serializer(instance=self.get_object(),data=data)
  38. serializer.is_valid(raise_exception=True)
  39. serializer.save()
  40. serializer = self.get_serializer(instance=self.get_object())
  41. return Response(serializer.data)

序列化器类:

  1. from rest_framework import serializers
  2. from students.models import Student
  3. class StudentModelSerializer(serializers.ModelSerializer):
  4. class Meta:
  5. model= Student
  6. fields = "__all__"
  7. class StudentModel2Serializer(serializers.ModelSerializer):
  8. class Meta:
  9. model= Student
  10. fields = ("name","class_null")

2 五个视图扩展类

作用:
提供了几种后端视图(对数据资源进行曾删改查)处理流程的实现,如果需要编写的视图属于这五种,则视图可以通过继承相应的扩展类来复用代码,减少自己编写的代码量。
这五个扩展类需要搭配GenericAPIView父类,因为五个扩展类的实现需要调用GenericAPIView提供的序列化器与数据库查询的方法。

1)ListModelMixin

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

  1. class ListModelMixin(object):
  2. """
  3. List a queryset.
  4. """
  5. def list(self, request, *args, **kwargs):
  6. # 过滤
  7. queryset = self.filter_queryset(self.get_queryset())
  8. # 分页
  9. page = self.paginate_queryset(queryset)
  10. if page is not None:
  11. serializer = self.get_serializer(page, many=True)
  12. return self.get_paginated_response(serializer.data)
  13. # 序列化
  14. serializer = self.get_serializer(queryset, many=True)
  15. return Response(serializer.data)

举例:

  1. from rest_framework.mixins import ListModelMixin
  2. class BookListView(ListModelMixin, GenericAPIView):
  3. queryset = BookInfo.objects.all()
  4. serializer_class = BookInfoSerializer
  5. def get(self, request):
  6. return self.list(request)

2)CreateModelMixin

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

  1. class CreateModelMixin(object):
  2. """
  3. Create a model instance.
  4. """
  5. def create(self, request, *args, **kwargs):
  6. # 获取序列化器
  7. serializer = self.get_serializer(data=request.data)
  8. # 验证
  9. serializer.is_valid(raise_exception=True)
  10. # 保存
  11. self.perform_create(serializer)
  12. headers = self.get_success_headers(serializer.data)
  13. return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
  14. def perform_create(self, serializer):
  15. serializer.save()
  16. def get_success_headers(self, data):
  17. try:
  18. return {'Location': str(data[api_settings.URL_FIELD_NAME])}
  19. except (TypeError, KeyError):
  20. return {}

3)RetrieveModelMixin

详情视图扩展类,提供retrieve(request, *args, **kwargs)方法,可以快速实现返回一个存在的数据对象。
如果存在,返回200, 否则返回404。
源代码:

  1. class RetrieveModelMixin(object):
  2. """
  3. Retrieve a model instance.
  4. """
  5. def retrieve(self, request, *args, **kwargs):
  6. # 获取对象,会检查对象的权限
  7. instance = self.get_object()
  8. # 序列化
  9. serializer = self.get_serializer(instance)
  10. return Response(serializer.data)

举例:

  1. class BookDetailView(RetrieveModelMixin, GenericAPIView):
  2. queryset = BookInfo.objects.all()
  3. serializer_class = BookInfoSerializer
  4. def get(self, request, pk):
  5. return self.retrieve(request)

4)UpdateModelMixin

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

  1. class UpdateModelMixin(object):
  2. """
  3. Update a model instance.
  4. """
  5. def update(self, request, *args, **kwargs):
  6. partial = kwargs.pop('partial', False)
  7. instance = self.get_object()
  8. serializer = self.get_serializer(instance, data=request.data, partial=partial)
  9. serializer.is_valid(raise_exception=True)
  10. self.perform_update(serializer)
  11. if getattr(instance, '_prefetched_objects_cache', None):
  12. # If 'prefetch_related' has been applied to a queryset, we need to
  13. # forcibly invalidate the prefetch cache on the instance.
  14. instance._prefetched_objects_cache = {}
  15. return Response(serializer.data)
  16. def perform_update(self, serializer):
  17. serializer.save()
  18. def partial_update(self, request, *args, **kwargs):
  19. kwargs['partial'] = True
  20. return self.update(request, *args, **kwargs)

5)DestroyModelMixin

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

  1. class DestroyModelMixin(object):
  2. """
  3. Destroy a model instance.
  4. """
  5. def destroy(self, request, *args, **kwargs):
  6. instance = self.get_object()
  7. self.perform_destroy(instance)
  8. return Response(status=status.HTTP_204_NO_CONTENT)
  9. def perform_destroy(self, instance):
  10. instance.delete()

使用GenericAPIView和视图扩展类,实现api接口,代码:

  1. """GenericAPIView结合视图扩展类实现api接口"""
  2. from rest_framework.mixins import ListModelMixin,CreateModelMixin
  3. class Students2GenericAPIView(GenericAPIView,ListModelMixin,CreateModelMixin):
  4. # 本次视图类中要操作的数据[必填]
  5. queryset = Student.objects.all()
  6. # 本次视图类中要调用的默认序列化器[玄天]
  7. serializer_class = StudentModelSerializer
  8. def get(self, request):
  9. """获取多个学生信息"""
  10. return self.list(request)
  11. def post(self,request):
  12. """添加学生信息"""
  13. return self.create(request)
  14. from rest_framework.mixins import RetrieveModelMixin,UpdateModelMixin,DestroyModelMixin
  15. class Student2GenericAPIView(GenericAPIView,RetrieveModelMixin,UpdateModelMixin,DestroyModelMixin):
  16. queryset = Student.objects.all()
  17. serializer_class = StudentModelSerializer
  18. # 在使用GenericAPIView视图获取或操作单个数据时,视图方法中的代表主键的参数最好是pk
  19. def get(self,request,pk):
  20. """获取一条数据"""
  21. return self.retrieve(request,pk)
  22. def put(self,request,pk):
  23. """更新一条数据"""
  24. return self.update(request,pk)
  25. def delete(self,request,pk):
  26. """删除一条数据"""
  27. return self.destroy(request,pk)

3 GenericAPIView的视图子类

1)CreateAPIView

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

2)ListAPIView

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

3)RetrieveAPIView

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

4)DestoryAPIView

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

5)UpdateAPIView

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

6)RetrieveUpdateAPIView

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

7)RetrieveUpdateDestoryAPIView

提供 get、put、patch、delete方法
继承自:GenericAPIView、RetrieveModelMixin、UpdateModelMixin、DestoryModelMixin

4 视图集ViewSet

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

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

ViewSet视图集类不再实现get()、post()等方法,而是实现动作 action 如 list() 、create() 等。
视图集只在使用as_view()方法的时候,才会将action动作与具体请求方式对应上。如:

  1. class BookInfoViewSet(viewsets.ViewSet):
  2. def list(self, request):
  3. books = BookInfo.objects.all()
  4. serializer = BookInfoSerializer(books, many=True)
  5. return Response(serializer.data)
  6. def retrieve(self, request, pk=None):
  7. try:
  8. books = BookInfo.objects.get(id=pk)
  9. except BookInfo.DoesNotExist:
  10. return Response(status=status.HTTP_404_NOT_FOUND)
  11. serializer = BookInfoSerializer(books)
  12. return Response(serializer.data)

在设置路由时,我们可以如下操作

  1. urlpatterns = [
  2. url(r'^books/$', BookInfoViewSet.as_view({'get':'list'}),
  3. url(r'^books/(?P<pk>\d+)/$', BookInfoViewSet.as_view({'get': 'retrieve'})
  4. ]

1 常用视图集父类

1) ViewSet

继承自APIViewViewSetMixin,作用也与APIView基本类似,提供了身份认证、权限校验、流量管理等。
ViewSet主要通过继承ViewSetMixin来实现在调用as_view()时传入字典(如{‘get’:’list’})的映射处理工作。
在ViewSet中,没有提供任何动作action方法,需要我们自己实现action方法。

2)GenericViewSet

使用ViewSet通常并不方便,因为list、retrieve、create、update、destory等方法都需要自己编写,而这些方法与前面讲过的Mixin扩展类提供的方法同名,所以我们可以通过继承Mixin扩展类来复用这些方法而无需自己编写。但是Mixin扩展类依赖与GenericAPIView,所以还需要继承GenericAPIView
GenericViewSet就帮助我们完成了这样的继承工作,继承自GenericAPIViewViewSetMixin,在实现了调用as_view()时传入字典(如{'get':'list'})的映射处理工作的同时,还提供了GenericAPIView提供的基础方法,可以直接搭配Mixin扩展类使用。
举例:

  1. from rest_framework.viewsets import GenericViewSet
  2. from rest_framework.mixins import ListModelMixin,CreateModelMixin,RetrieveModelMixin,UpdateModelMixin,DestroyModelMixin
  3. class Student4ViewSet(GenericViewSet,ListModelMixin,CreateModelMixin,RetrieveModelMixin,UpdateModelMixin,DestroyModelMixin):
  4. queryset = Student.objects.all()
  5. serializer_class = StudentModelSerializer

url的定义

  1. urlpatterns = [
  2. path("students7/", views.Student4ViewSet.as_view({"get": "list", "post": "create"})),
  3. re_path("students7/(?P<pk>\d+)/", views.Student4ViewSet.as_view({"get": "retrieve","put":"update","delete":"destroy"})),
  4. ]

3)ModelViewSet

继承自GenericViewSet,同时包括了ListModelMixin、RetrieveModelMixin、CreateModelMixin、UpdateModelMixin、DestoryModelMixin。

4)ReadOnlyModelViewSet

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

2 视图集中定义附加action动作

在视图集中,除了上述默认的方法动作外,还可以添加自定义动作。
举例:

  1. from rest_framework.viewsets import ModelViewSet,ReadOnlyModelViewSet
  2. class StudentModelViewSet(ModelViewSet):
  3. queryset = Student.objects.all()
  4. serializer_class = StudentModelSerializer
  5. def login(self,request):
  6. """学生登录功能"""
  7. return Response({"message":"登录成功"})

url的定义

  1. urlpatterns = [
  2. path("students8/", views.StudentModelViewSet.as_view({"get": "list", "post": "create"})),
  3. re_path("students8/(?P<pk>\d+)/",
  4. views.StudentModelViewSet.as_view({"get": "retrieve", "put": "update", "delete": "destroy"})),
  5. path("stu/login/",views.StudentModelViewSet.as_view({"get":"login"}))
  6. ]

3 action属性

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

  1. from rest_framework.viewsets import ModelViewSet
  2. from students.models import Student
  3. from .serializers import StudentModelSerializer
  4. from rest_framework.response import Response
  5. class StudentModelViewSet(ModelViewSet):
  6. queryset = Student.objects.all()
  7. serializer_class = StudentModelSerializer
  8. def get_new_5(self,request):
  9. """获取最近添加的5个学生信息"""
  10. # 操作数据库
  11. print(self.action) # 获取本次请求的视图方法名
  12. 通过路由访问到当前方法中.可以看到本次的action就是请求的方法名

5 总结

image.png