DRF檢視和檢視集
檢視說明
1. 兩個基類
1)APIView
rest_framework.views.APIView
APIView
是REST framework提供的所有檢視的基類,繼承自Django的View
父類。
APIView
與View
的不同之處在於:
- 傳入到檢視方法中的是REST framework的
Request
物件,而不是Django的HttpRequeset
物件; - 檢視方法可以返回REST framework的
Response
物件,檢視會為響應資料設定(render)符合前端要求的格式; - 任何
APIException
異常都會被捕獲到,並且處理成合適的響應資訊; - 在進行dispatch()分發前,會對請求進行身份認證、許可權檢查、流量控制。
支援定義的屬性:
- authentication_classes 列表或元祖,身份認證類
- permissoin_classes 列表或元祖,許可權檢查類
- throttle_classes 列表或元祖,流量控制類
在APIView
中仍以常規的類檢視定義方法來實現get() 、post() 或者其他請求方式的方法。
舉例:
from rest_framework.views import APIView from rest_framework.response import Response # url(r'^books/$', views.BookListView.as_view()), class BookListView(APIView): def get(self, request): books = BookInfo.objects.all() serializer = BookInfoSerializer(books, many=True) return Response(serializer.data)
2)GenericAPIView
rest_framework.generics.GenericAPIView
繼承自APIVIew
,主要增加了操作序列化器和資料庫查詢的方法,作用是為下面Mixin擴充套件類的執行提供方法支援。通常在使用時,可搭配一個或多個Mixin擴充套件類。
提供的關於序列化器使用的屬性與方法
-
屬性:
- serializer_class 指明檢視使用的序列化器
-
方法:
-
get_serializer_class(self)
返回序列化器類,預設返回
serializer_class
,可以重寫,例如:def get_serializer_class(self): if self.request.user.is_staff: return FullAccountSerializer return BasicAccountSerializer
-
get_serializer(self, args, *kwargs)
返回序列化器物件,主要用來提供給Mixin擴充套件類使用,如果我們在檢視中想要獲取序列化器物件,也可以直接呼叫此方法。
注意,該方法在提供序列化器物件的時候,會向序列化器物件的context屬性補充三個資料:request、format、view,這三個資料物件可以在定義序列化器時使用。
-
request 當前檢視的請求物件
-
view 當前請求的類檢視物件
-
format 當前請求期望返回的資料格式
-
-
提供的關於資料庫查詢的屬性與方法
-
屬性:
- queryset 指明使用的資料查詢集
-
方法:
-
get_queryset(self)
返回檢視使用的查詢集,主要用來提供給Mixin擴充套件類使用,是列表檢視與詳情檢視獲取資料的基礎,預設返回
queryset
屬性,可以重寫,例如:def get_queryset(self): user = self.request.user return user.accounts.all()
-
get_object(self)
返回詳情檢視所需的模型類資料物件,主要用來提供給Mixin擴充套件類使用。
在試圖中可以呼叫該方法獲取詳情資訊的模型類物件。
若詳情訪問的模型類物件不存在,會返回404。
該方法會預設使用APIView提供的check_object_permissions方法檢查當前物件是否有許可權被訪問。
舉例:
# url(r'^books/(?P<pk>\d+)/$', views.BookDetailView.as_view()), class BookDetailView(GenericAPIView): queryset = BookInfo.objects.all() serializer_class = BookInfoSerializer def get(self, request, pk): book = self.get_object() # get_object()方法根據pk引數查詢queryset中的資料物件 serializer = self.get_serializer(book) return Response(serializer.data)
-
其他可以設定的屬性
- pagination_class 指明分頁控制類
- filter_backends 指明過濾控制後端
2. 五個擴充套件類
作用:
提供了幾種後端檢視(對資料資源進行曾刪改查)處理流程的實現,如果需要編寫的檢視屬於這五種,則檢視可以通過繼承相應的擴充套件類來複用程式碼,減少自己編寫的程式碼量。
這五個擴充套件類需要搭配GenericAPIView父類,因為五個擴充套件類的實現需要呼叫GenericAPIView提供的序列化器與資料庫查詢的方法。
1)ListModelMixin
列表檢視擴充套件類,提供list(request, *args, **kwargs)
方法快速實現列表檢視,返回200狀態碼。
該Mixin的list方法會對資料進行過濾和分頁。
原始碼:
class ListModelMixin(object):
"""
List a queryset.
"""
def list(self, request, *args, **kwargs):
# 過濾
queryset = self.filter_queryset(self.get_queryset())
# 分頁
page = self.paginate_queryset(queryset)
if page is not None:
serializer = self.get_serializer(page, many=True)
return self.get_paginated_response(serializer.data)
# 序列化
serializer = self.get_serializer(queryset, many=True)
return Response(serializer.data)
舉例:
from rest_framework.mixins import ListModelMixin
class BookListView(ListModelMixin, GenericAPIView):
queryset = BookInfo.objects.all()
serializer_class = BookInfoSerializer
def get(self, request):
return self.list(request)
2)CreateModelMixin
建立檢視擴充套件類,提供create(request, *args, **kwargs)
方法快速實現建立資源的檢視,成功返回201狀態碼。
如果序列化器對前端傳送的資料驗證失敗,返回400錯誤。
原始碼:
class CreateModelMixin(object):
"""
Create a model instance.
"""
def create(self, request, *args, **kwargs):
# 獲取序列化器
serializer = self.get_serializer(data=request.data)
# 驗證
serializer.is_valid(raise_exception=True)
# 儲存
self.perform_create(serializer)
headers = self.get_success_headers(serializer.data)
return Response(serializer.data, status=status.HTTP_201_CREATED, headers=headers)
def perform_create(self, serializer):
serializer.save()
def get_success_headers(self, data):
try:
return {'Location': str(data[api_settings.URL_FIELD_NAME])}
except (TypeError, KeyError):
return {}
3) RetrieveModelMixin
詳情檢視擴充套件類,提供retrieve(request, *args, **kwargs)
方法,可以快速實現返回一個存在的資料物件。
如果存在,返回200, 否則返回404。
原始碼:
class RetrieveModelMixin(object):
"""
Retrieve a model instance.
"""
def retrieve(self, request, *args, **kwargs):
# 獲取物件,會檢查物件的許可權
instance = self.get_object()
# 序列化
serializer = self.get_serializer(instance)
return Response(serializer.data)
舉例:
class BookDetailView(RetrieveModelMixin, GenericAPIView):
queryset = BookInfo.objects.all()
serializer_class = BookInfoSerializer
def get(self, request, pk):
return self.retrieve(request)
4)UpdateModelMixin
更新檢視擴充套件類,提供update(request, *args, **kwargs)
方法,可以快速實現更新一個存在的資料物件。
同時也提供partial_update(request, *args, **kwargs)
方法,可以實現區域性更新。
成功返回200,序列化器校驗資料失敗時,返回400錯誤。
原始碼:
class UpdateModelMixin(object):
"""
Update a model instance.
"""
def update(self, request, *args, **kwargs):
partial = kwargs.pop('partial', False)
instance = self.get_object()
serializer = self.get_serializer(instance, data=request.data, partial=partial)
serializer.is_valid(raise_exception=True)
self.perform_update(serializer)
if getattr(instance, '_prefetched_objects_cache', None):
# If 'prefetch_related' has been applied to a queryset, we need to
# forcibly invalidate the prefetch cache on the instance.
instance._prefetched_objects_cache = {}
return Response(serializer.data)
def perform_update(self, serializer):
serializer.save()
def partial_update(self, request, *args, **kwargs):
kwargs['partial'] = True
return self.update(request, *args, **kwargs)
5)DestroyModelMixin
刪除檢視擴充套件類,提供destroy(request, *args, **kwargs)
方法,可以快速實現刪除一個存在的資料物件。
成功返回204,不存在返回404。
原始碼:
class DestroyModelMixin(object):
"""
Destroy a model instance.
"""
def destroy(self, request, *args, **kwargs):
instance = self.get_object()
self.perform_destroy(instance)
return Response(status=status.HTTP_204_NO_CONTENT)
def perform_destroy(self, instance):
instance.delete()
3. 幾個可用子類檢視
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
檢視集ViewSet
使用檢視集ViewSet,可以將一系列邏輯相關的動作放到一個類中:
- list() 提供一組資料
- retrieve() 提供單個數據
- create() 建立資料
- update() 儲存資料
- destory() 刪除資料
ViewSet檢視集類不再實現get()、post()等方法,而是實現動作 action 如 list() 、create() 等。
檢視集只在使用as_view()方法的時候,才會將action動作與具體請求方式對應上。如:
class BookInfoViewSet(viewsets.ViewSet):
def list(self, request):
books = BookInfo.objects.all()
serializer = BookInfoSerializer(books, many=True)
return Response(serializer.data)
def retrieve(self, request, pk=None):
try:
books = BookInfo.objects.get(id=pk)
except BookInfo.DoesNotExist:
return Response(status=status.HTTP_404_NOT_FOUND)
serializer = BookInfoSerializer(books)
return Response(serializer.data)
在設定路由時,我們可以如下操作
urlpatterns = [
url(r'^books/$', BookInfoViewSet.as_view({'get':'list'}),
url(r'^books/(?P<pk>\d+)/$', BookInfoViewSet.as_view({'get': 'retrieve'})
]
1. 常用檢視集父類
1) ViewSet
繼承自APIView
與ViewSetMixin
,作用也與APIView基本類似,提供了身份認證、許可權校驗、流量管理等。
ViewSet主要通過繼承ViewSetMixin來實現在呼叫as_view()時傳入字典(如{'get':'list'}
)的對映處理工作。
在ViewSet中,沒有提供任何動作action方法,需要我們自己實現action方法。
2)GenericViewSet
使用ViewSet通常並不方便,因為list、retrieve、create、update、destory等方法都需要自己編寫,而這些方法與前面講過的Mixin擴充套件類提供的方法同名,所以我們可以通過繼承Mixin擴充套件類來複用這些方法而無需自己編寫。但是Mixin擴充套件類依賴與GenericAPIView
,所以還需要繼承GenericAPIView
。
GenericViewSet就幫助我們完成了這樣的繼承工作,繼承自GenericAPIView
與ViewSetMixin
,在實現了呼叫as_view()時傳入字典(如{'get':'list'}
)的對映處理工作的同時,還提供了GenericAPIView
提供的基礎方法,可以直接搭配Mixin擴充套件類使用。
舉例:
from rest_framework import mixins
from rest_framework.viewsets import GenericViewSet
from rest_framework.decorators import action
class BookInfoViewSet(mixins.ListModelMixin, mixins.RetrieveModelMixin, GenericViewSet):
queryset = BookInfo.objects.all()
serializer_class = BookInfoSerializer
url的定義
urlpatterns = [
url(r'^books/$', views.BookInfoViewSet.as_view({'get': 'list'})),
url(r'^books/(?P<pk>\d+)/$', views.BookInfoViewSet.as_view({'get': 'retrieve'})),
]
3)ModelViewSet
繼承自GenericViewSet
,同時包括了ListModelMixin、RetrieveModelMixin、CreateModelMixin、UpdateModelMixin、DestoryModelMixin。
4)ReadOnlyModelViewSet
繼承自GenericViewSet
,同時包括了ListModelMixin、RetrieveModelMixin。
2. 檢視集中定義附加action動作
在檢視集中,除了上述預設的方法動作外,還可以新增自定義動作。
舉例:
from rest_framework import mixins
from rest_framework.viewsets import GenericViewSet
from rest_framework.decorators import action
class BookInfoViewSet(mixins.ListModelMixin, mixins.RetrieveModelMixin, GenericViewSet):
queryset = BookInfo.objects.all()
serializer_class = BookInfoSerializer
def latest(self, request):
"""
返回最新的圖書資訊
"""
book = BookInfo.objects.latest('id')
serializer = self.get_serializer(book)
return Response(serializer.data)
def read(self, request, pk):
"""
修改圖書的閱讀量資料
"""
book = self.get_object()
book.bread = request.data.get('read')
book.save()
serializer = self.get_serializer(book)
return Response(serializer.data)
url的定義
urlpatterns = [
url(r'^books/$', views.BookInfoViewSet.as_view({'get': 'list'})),
url(r'^books/latest/$', views.BookInfoViewSet.as_view({'get': 'latest'})),
url(r'^books/(?P<pk>\d+)/$', views.BookInfoViewSet.as_view({'get': 'retrieve'})),
url(r'^books/(?P<pk>\d+)/read/$', views.BookInfoViewSet.as_view({'put': 'read'})),
]
3. action屬性
在檢視集中,我們可以通過action物件屬性來獲取當前請求檢視集時的action動作是哪個。
例如:
def get_serializer_class(self):
if self.action == 'create':
return OrderCommitSerializer
else:
return OrderDataSerializer
4. 檢視集的繼承關係