DRF的路由生成类的使用

DRF路由生成类的使用

对于视图集ViewSet，我们除了可以自己手动指明请求方式与动作action之间的对应关系外，还可以使用Routers来帮助我们快速实现路由信息。

REST framework提供了两个router

SimpleRouter
DefaultRouter

使用方法：

1）创建router对象，并注册视图集，例如

from rest_framework import routers

router = routers.SimpleRouter()
router.register(r'books', BookInfoViewSet, base_name='book')

prefix 该视图集的路由前缀
viewset 视图集
base_name 路由名称的前缀

如上述代码会形成的路由如下：

^books/$    name: book-list
^books/{pk}/$   name: book-detail

2）添加路由数据

扫描二维码关注公众号，回复： 7013940 查看本文章

可以有两种方式：
urlpatterns = [
    ...
]
urlpatterns += router.urls
或
urlpatterns = [
    ...
    url(r'^', include(router.urls))
]

在views.py中操作完视图函数后，路由我们可以用drf中自带的路由类，route，

它或会帮助我们将视图函数的增删改查都自动生成

from rest_framework.routers import DefaultRouter

案例说明：

serializers.py

# 创建一个Bookinfo 序列化器
from rest_framework import serializers
from .models import BookInfo

class BookInfoSerializer(serializers.ModelSerializer):
    class Meta:
        model = BookInfo
        fields = '__all__'

views.py

from django.shortcuts import render

# Create your views here.
from rest_framework.viewsets import ModelViewSet
from .serializers import BookInfoSerializer
from .models import BookInfo

class BookInfoViewSet(ModelViewSet):
    queryset = BookInfo.objects.all()
    serializer_class = BookInfoSerializer

urls.py

from . import views

from rest_framework.routers import DefaultRouter

urlpatterns = [

]

route = DefaultRouter()
# route.register("路由地址的前缀", 视图集类) # 
route.register(r"books", views.BookInfoViewSet) # 

urlpatterns += route.urls

DefaultRouter与SimpleRouter的区别是，DefaultRouter会多附带一个默认的API根视图，返回一个包含所有列表视图的超链接响应数据。

api根视图

另外附加说明：视图集中附加action的声明

上面的route路由自动会产生增删改查的路由，但是如果我要访问自定义的那个视图函数，它是不会自动产生的

比如在views.py 文件中创建了一个查询这个图书的前五名阅读量的视图函数，但是route类并没有给我产生，我该怎么办呢？

这个时候就用到了附加action的的申明

在视图集中，如果想要让Router自动帮助我们为自定义的动作生成路由信息，需要使用rest_framework.decorators.action装饰器。

以action装饰器装饰的方法名会作为action动作名，与list、retrieve等同。

action装饰器可以接收两个参数：

methods: 声明该action对应的请求方式，列表传递
detail

: 声明该action的路径是否与单一资源对应，及是否是
```
xxx/<pk>/action方法名/
```

True 表示路径格式是xxx/<pk>/action方法名/
False 表示路径格式是xxx/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

    # detail为False 表示路径名格式应该为 books/latest/
    @action(methods=['get'], detail=False)
    def latest(self, request):
        """
        返回最新的图书信息
        """
        ...

    # detail为True，表示路径名格式应该为 books/{pk}/read/
    @action(methods=['put'], detail=True)
    def read(self, request, pk):
        """
        修改图书的阅读量数据
        """
        ...

由路由器自动为此视图集自定义action方法形成的路由会是如下内容：

^books/latest/$    name: book-latest
^books/{pk}/read/$  name: book-read