前言
在drf中，我们写接口可以通过继承modelViewSet从而达到非常快速的功能实现，这十分的方便，但是modelViewSet由于需要根据不同的参数来对应不同的处理，所以我们写的url最少都需要两条，如下所示：
path('api/users/', views.UserAPI.as_view(actions={"get":"list","post":"create"})),
re_path('^api/users/(?P<uid>d+)?',views.UserAPI.as_view(actions={"get":"retrieve","patch":"update","delete":"destroy"}))
有没有什么办法能够快速的生成两条url呢？其实是有的，但是在这之前我们也可以对他手动封装一个组件，达到自动生成路由的功能。
手动封装
下面是手动封装url的一个示例，首先url本身就是一个列表，所以我们的组件最终可以返回一个可迭代对象就行，直接将生成的url添加至已有的url列表中。
from django.contrib import admin
from django.urls import path, re_path
from app01 import views
from rest_framework.routers import SimpleRouter
class GenerateRouter:
def __init__(self):
self.urls = None
def register(self, prefix, viewset, basename=None):
# prefix 匹配规则，不用加 /
# viewset 继承自ModelViewSet视图类
# basename 别名
from django.urls import path, re_path
name = viewset.serializer_class.Meta.model.__name__.lower() # 获取查询的数据表名字
self.urls = (
re_path("^%s/$" % prefix, viewset.as_view(actions={"get": "list", "post": "create"}),
name=basename or name + "-list"), # 不需要参数，查全部和新增
re_path('^%s/(?P<%s>[^/.]+)/$' %(prefix,viewset.lookup_url_kwarg or "pk"), # 如果设置捕获字段，就用捕获字段，否则就用pk
viewset.as_view(
actions={"get": "retrieve", "patch": "update", "delete": "destroy", "put": "update"}),
name=basename or name + "-detail"),
)
router = GenerateRouter() # 实例化
router.register("api/users", views.UserAPI) # 注册
urlpatterns = [
path('admin/', admin.site.urls),
]
urlpatterns.extend(router.urls) # 添加
它生成的路由是这样子的：
^api/users/$ [name='user-list']
^api/users/(?P<uid>[^/.]+)/$ [name='user-detail']
router组件
简单路由
在drf中也提供了这样的组件，它可以帮助我们快速的生成路由。比我们上面自己封装的要强大一些，但是本质上都是一样的。
下面是简单使用，首先我们需要进行导入。
from rest_framework.routers import SimpleRouter
其次是对其进行实例化后注册一个路由信息。
def register(self, prefix, viewset, basename=None)
# prefix 匹配规则，不用加 /
# viewset 继承自ModelViewSet视图类
# basename 别名
from django.contrib import admin
from django.urls import path
from app01 import views
from rest_framework.routers import SimpleRouter # 第一步导入
router = SimpleRouter() # 第二步实例化
router.register("api/users", views.UserAPI) # 第三步注册路由
urlpatterns = [
path('admin/', admin.site.urls),
]
urlpatterns.extend(router.urls) # 第四步添加
它会自动生成两条url，这与我们上面的手动封装是一样的原理：
^api/users/$ [name='user-list']
^api/users/(?P<uid>[^/.]+)/$ [name='user-detail']
至于这里为什么是uid，这是因为我们的API中lookup_url_kwarg设置的就是uid。
可以看以下上面手动封装中，也是这么做的。
class UserAPI(ModelViewSet):
queryset = models.User.objects # 传入对象即可
serializer_class = ser.UserModelSerializers # 序列化类
lookup_field = "pk"
lookup_url_kwarg = "uid" # 由于捕获的是uid，需要声明
默认路由
简单的SimpleRouter()其实就足够我们用了，但是drf也提供了更高级的默认路由，它会生成六条url，但是基本没啥用。
from django.contrib import admin
from django.urls import path, re_path
from app01 import views
from rest_framework.routers import SimpleRouter
from rest_framework.routers import DefaultRouter
router = DefaultRouter()
router.register("api/users", views.UserAPI)
urlpatterns = [
path('admin/', admin.site.urls),
]
urlpatterns.extend(router.urls)
以下是生成的路由信息。
^api/users/$ [name='user-list']
^api/users.(?P<format>[a-z0-9]+)/?$ [name='user-list']
^api/users/(?P<uid>[^/.]+)/$ [name='user-detail']
^api/users/(?P<uid>[^/.]+).(?P<format>[a-z0-9]+)/?$ [name='user-detail']
^$ [name='api-root']
^.(?P<format>[a-z0-9]+)/?$ [name='api-root']
action装饰器
action干什么用？是为了给继承自ViewSetMixin的视图类中定义的函数也添加路由，因为ViewSetMixin中有一个actions的参数，使用action就能够填入这个参数。
from . import models
from . import ser
from rest_framework.decorators import action # 导入装饰器
from rest_framework.viewsets import ModelViewSet
from rest_framework.response import Response
class UserAPI(ModelViewSet):
queryset = models.User.objects.all() # 传入对象即可
serializer_class = ser.UserModelSerializers # 序列化类
lookup_field = "pk"
lookup_url_kwarg = "uid" # 由于捕获的是uid，需要声明
@action(methods=['GET','POST'],detail=True) # 还有个参数，url_path。匹配规则，默认是函数名，还有url_name，反向代理时的别名
def customize(self,request,uid):
book=self.get_queryset()[:2] # 从0开始截取一条
serializer=self.get_serializer(book,many=True) # 只要是返回queryset对象，就需要many=True
return Response(serializer.data)
# action的执行会覆盖掉本身的get、post、patch等所执行的方法，如list，update，create等默认行为。
# methods代表请求方式，当有该种请求到来时，则执行该方法
# detail代表是否需要捕获参数，True则是捕获。捕获参数名字就是lookup_url_kwarg
生成的路由信息如下：
^api/users/(?P<uid>[^/.]+)/customize/$ [name='user-customize']
^api/users/(?P<uid>[^/.]+)/customize.(?P<format>[a-z0-9]+)/?$ [name='user-customize']