DRF--路由组件和版本控制

路由组件

先来看下我们前面写的路由

from django.conf.urls import url, include
from .views import  BookModelView

urlpatterns = [
    url(r'^book/$', BookModelView.as_view({"get": "list", "post": "create"})),
    url(r'^book/(?P<pk>\d+)', BookModelView.as_view({"get": "retrieve", "put": "update", "delete": "destroy"})),
]

这样写路由也没有什么问题，DRF也提供给了我们另一种写法

from .views import  BookModelView
# 帮助我们生成带参数的路由
from rest_framework.routers import DefaultRouter

# 实例化DefaultRouter对象
router = DefaultRouter()

# 注册我们的路由以及视图
router.register(r'^book', BookModelView)

urlpatterns = [

]

# 把我们注册的路由添加到里面
urlpatterns += router.urls

我们在浏览器里输入错误的地址看DRF都给我们生成了哪些路由

^api/ ^^book/$ [name='book-list']
^api/ ^^book\.(?P<format>[a-z0-9]+)/?$ [name='book-list']
^api/ ^^book/(?P<pk>[^/.]+)/$ [name='book-detail']
^api/ ^^book/(?P<pk>[^/.]+)\.(?P<format>[a-z0-9]+)/?$ [name='book-detail']
^api/ ^$ [name='api-root']
^api/ ^\.(?P<format>[a-z0-9]+)/?$ [name='api-root']

这种注册路由得方法在实际项目中并不常用，因为把这些路由都开放，会造成一定的安全性

版本控制

我们的网站要在不同的阶段加入不同的功能，restful规定了我们的接口要有版本号。但是不是所有人看到有更新就会更新的，所以我们的要兼容不同的版本。而前端肯定能拿到用户访问的是哪个版本的，这样，让前端的妹子把版本号传给后端，后端拿到版本号，做不同的处理，在把结果返回给前端就可以了。

既然我们知道了前端妹子可以拿到版本号，那怎么传给后端，是放在url里还是放在header里，或者是请求体里。这些都可以，妹子怎么高兴怎么来。

我们先来分析一下DRF里的源码是怎么写的

进入APIView类里，找到dispatch方法，找到下面的代码

try:
    self.initial(request, *args, **kwargs)

进入initial方法里

def initial(self, request, *args, **kwargs):
    """
    Runs anything that needs to occur prior to calling the method handler.
    """
    self.format_kwarg = self.get_format_suffix(**kwargs)

    # Perform content negotiation and store the accepted info on the request
    neg = self.perform_content_negotiation(request)
    request.accepted_renderer, request.accepted_media_type = neg

    # Determine the API version, if versioning is in use.
    # 初始化我们的版本
    version, scheme = self.determine_version(request, *args, **kwargs)
    request.version, request.versioning_scheme = version, scheme

    # Ensure that the incoming request is permitted
    # 认证组件
    self.perform_authentication(request)
    # 权限组件
    self.check_permissions(request)
    # 频率
    self.check_throttles(request)

源码里会把 determine_version方法的结果解包赋值给version，然后把version赋值给了request.version去看看 determine_version做了什么

 def determine_version(self, request, *args, **kwargs):
     """
     If versioning is being used, then determine any API version for the
     incoming request. Returns a two-tuple of (version, versioning_scheme)
     """
     if self.versioning_class is None:
         return (None, None)
     scheme = self.versioning_class()
     return (scheme.determine_version(request, *args, **kwargs), scheme)

里面判断了versioning_class这个方法的返回值是不是None，默认api_settings里的DEFAULTS里的DEFAULT_VERSIONING_CLASS'为None，所以版本默认就是为None，我们可以写个方法来测试一下

class VersionDemo(APIView):
    def get(self, request):
        print(request.version)
        print(request.versioning_scheme)
        return Response("ok")

最后的结果就是None

所以我们可以重写DEFAULT_VERSIONING_CLASS这个方法。哪我们在哪里重写呢？Django给我们提供了settings.py文件，这里面的会覆盖默认的配置。好了，既然我们已经知道了在哪里重写，那我们就来实践一遍

全局版本类的配置

先在项目的根目录下创建个utils文件夹，在创建个version.py文件

在里面写如下代码

class MyVersion(object):
    def determine_version(self, request, *args, **kwargs):  # 必须是这个函数名
        # 方法的返回值是版本号
        # 获取前端传过来的版本号，并且把版本号返回
        version = request.query_params.get('version')  # 前端在url里携带版本号
        if not version:
            version = 'v1'
        return version

写好了去settings.py里配置

REST_FRAMEWORK = {
    "DEFAULT_VERSIONING_CLASS":"utils.version.MyVersion"
}

# utils.version.MyVersion 文件里的类名

然后访问类似这样的地址：http://127.0.0.1:8000/version/demo/?version=v3，控制台打印出的就是v3。既然我们能拿到版本号了，那我们就能根据版本号做不同的逻辑处理了。修改views下的VersionDemo类

class VersionDemo(APIView):
    def get(self, request):
        print(request.version)
        print(request.versioning_scheme)
        # 根据版本号处理不同的业务逻辑
        if request.version == 'v2':
            return Response('这是v2的业务逻辑')
        return Response("这是v1的业务逻辑")

上面的配置是全局的配置，假如我们现在有两个类

from rest_framework.views import APIView
from rest_framework.response import Response

class VersionDemo(APIView):

　　def get(self, request):
        print(request.version)
        print(request.versioning_scheme)
        # 根据版本号处理不同的业务逻辑
        if request.version == 'v2':
            return Response('这是v2的业务逻辑')
        return Response("这是v1的业务逻辑")

class VersionTest(APIView):
    def get(self, request):
        print(request.version)
        print(request.versioning_scheme)
        # 根据版本号处理不同的业务逻辑
        if request.version == 'v2':
            return Response('这是v3的业务逻辑')
        return Response("这是v4的业务逻辑")

现在不管访问哪个类都会走我们定义的MyVersion类，有时候，我们只希望一部分类有版本控制，一部分不需要版本控制。DRF可以允许我们局部配置

局部配置版本号

局部配置版本号要先注释掉settings.py里写的配置项

# REST_FRAMEWORK = {
#     "DEFAULT_VERSIONING_CLASS":"utils.version.MyVersion"
# }

然后在需要版本控制的方法前加上 versioning_class=版本类，类似于下面这样

from django.shortcuts import render
from rest_framework.views import APIView
from rest_framework.response import Response
from utils.version import MyVersion

class VersionDemo(APIView):
    versioning_class = MyVersion

    def get(self, request):
        print(request.version)
        print(request.versioning_scheme)
        # 根据版本号处理不同的业务逻辑
        if request.version == 'v2':
            return Response('这是v2的业务逻辑')
        return Response("这是v1的业务逻辑")

class VersionTest(APIView):
    def get(self, request):
        print(request.version)
        print(request.versioning_scheme)
        # 根据版本号处理不同的业务逻辑
        if request.version == 'v2':
            return Response('这是v3的业务逻辑')
        return Response("这是v4的业务逻辑")

这样配置好了之后，我们访问 VersionDemo，就会走我们自定义的版本类。如果访问 VersionTest，就不会走我们自定义的版本类，版本号为默认的None

使用原始的版本配置类

DRF已经替我们封装好了版本控制类，包括请求头的，url里的，主机里的。。。我们只需要使用就可以了，所有的类都在 versioning 方法里

from rest_framework import versioning

使用局部的时候只需要加上versioning_class就可以了，类似于下面这样

class VersionTest(APIView):
    versioning_class = versioning.QueryParameterVersioning  # 使用自带的配置
    def get(self, request):
        print(request.version)
        print(request.versioning_scheme)
        # 根据版本号处理不同的业务逻辑
        if request.version == 'v2':
            return Response('这是v3的业务逻辑')
        return Response("这是v4的业务逻辑")

如果要使用全局的配置，就需要在settings.py里进行配置了

REST_FRAMEWORK = {
    "DEFAULT_VERSIONING_CLASS":"rest_framework.versioning.QueryParameterVersioning"
}

这样就不需要我们自己写版本控制类了

其他的一些配置

REST_FRAMEWORK = {
    # 默认使用的版本控制类
    'DEFAULT_VERSIONING_CLASS': 'rest_framework.versioning.URLPathVersioning',
    # 允许的版本
    'ALLOWED_VERSIONS': ['v1', 'v2'],
    # 版本使用的参数名称
    'VERSION_PARAM': 'version',
    # 默认使用的版本
    'DEFAULT_VERSION': 'v1',
}