基礎(chǔ)的ViewSet

views.py

from rest_framework import viewsets
from .models import Merchant
from .serializers import MerchantSerializer
from rest_framework.response import Response
from rest_framework import status
from django.shortcuts import get_object_or_404

class MerchantViewSet(viewsets.ViewSet):
    def list(self,request):
        # get請求
        queryset = Merchant.objects.all()
        serializer = MerchantSerializer(queryset,many=True)
        return Response(data=serializer.data)

    def create(self,request):
        # post
        serializer = MerchantSerializer(data=request.data)
        if serializer.is_valid():
            serializer.save()
            return Response("success")
        else:
            return Response("fail",status=status.HTTP_400_BAD_REQUEST)

    def retrieve(self,request,pk=None):
        # 查詢
        queryset = Merchant.objects.all()
        merchant = get_object_or_404(queryset,pk=pk)
        serializer = MerchantSerializer(merchant)
        return Response(serializer.data)

    def update(self,request,pk=None):
        # put
        queryset = Merchant.objects.all()
        merchant = get_object_or_404(queryset, pk=pk)
        serializer = MerchantSerializer()
        serializer.update(merchant,request.data)
        return Response('success')

    def destroy(self,request,pk=None):
        # delete
        queryset = Merchant.objects.all()
        merchant = get_object_or_404(queryset,pk=pk)
        merchant.delete()
        return Response('success')

進階版：ModelViewSet

上面的例子直接用下面的views.py內(nèi)容即可實現(xiàn)
views.py

class MerchantViewSet(viewsets.ModelViewSet):
    queryset = Merchant.objects.all()
    serializer_class = MerchantSerializer

高級版

在當前的視圖集中增加其他的url，這時候可以使用@active實現(xiàn)
views.py

from rest_framework.decorators import action

class MerchantViewSet(viewsets.ModelViewSet):
    queryset = Merchant.objects.all()
    serializer_class = MerchantSerializer

    @action(['GET'],detail=False)
    def cs(self,request,*args,**kwargs):
        queryset = self.get_queryset()
        queryset = queryset.filter(name__contains="長沙")
        serializer = MerchantSerializer(queryset,many=True)
        return Response(serializer.data)

urls.py路由部分不需要修改。以后直接可以通過/merchant/cs/可以訪問到name中包含了"長沙"兩個字的所有商家。

url的書寫

使用視圖集時，路由應按照下面的方式寫。

urls.py
from rest_framework.routers import DefaultRouter

router = DefaultRouter(trailing_slash=False)
router.register('merchant/',views.MerchantView,basename='merchant')

urlpatterns = [] + router.urls