flask第十一篇——自定义url转换器

Posted 2021-04-12 自动化测试实战

首先和大家说一下flask的组成，flask其实是werkzeug（路由和http处理）+SQLAlchemy（数据库）+Jinja2（模板）组成的，今天我们要看到的东西就是`werkzeug`里面的内容。

我们在代码里导入from werkzeug.routing import BaseConverter，然后进入BaseConverter中看一下这个方法的源码：

其中to_python方法的作用就是将你传过去的参数转换成对应类型的数据，比如你设置传参是uuid类型数据，那么当你传参以后，就会调用to_python方法，将参数转换为对应的uuid类型。

此外，下面还有PathConverter、UUIDConverter等，再往下，我们会看到，原来这里写好了一个映射，每当我们在url中传入指定类型的参数时，系统会自动的来寻找这里的映射：

# coding: utf-8

from flask import Flask
from werkzeug.routing import BaseConverter
import flask

app = Flask(__name__)  # type: Flask
app.debug = True

# 根目录
@app.route('/')
def hello_world():

    return 'Hello World!'

class ListConverter(BaseConverter):

    def to_python(self, values):
        """        将url中的参数转换为我们需要的数据类型        """
        # split方法就是去掉加号并返回list类型数据
        tmp = values.split('+')
        print 'tmp: %s' % tmp
        return tmp
    
    def to_url(self, values):
        """        将[1,2,3]转换成1+2+3        """
        # 遍历列表values中的数据，以+连接，最后tmp1的值即1+2+3
        # BaseConverter.to_url是对url进行编码
        tmp1 = '+'.join([BaseConverter.to_url(self, value) for value in values])
        print 'tmp1:%s' % tmp1
        return tmp1


# 将写好的类注册到DEFAULT_CONVERTERS
app.url_map.converters['list'] = ListConverter

@app.route('/detail/<list:params>/')
def detail(params):
    print 'parmas:%s' % params
    return 'success for url'

with app.test_request_context():
    print 'detail函数的url是:%s' % flask.url_for('detail', params=[1, 2, 3])


if __name__ == '__main__':
    app.run()

执行代码以后刷新页面可以看到：

• 控制台

tmp1:1+2+3
detail函数的url是:/detail/1+2+3/
 * Running on http://127.0.0.1:5000/ (Press CTRL+C to quit)
 * Restarting with stat
tmp1:1+2+3
detail函数的url是:/detail/1+2+3/
 * Debugger is active!
 * Debugger PIN: 176-123-878
127.0.0.1 - - [15/Mar/2018 00:21:23] "GET /detail/1+2+3/ HTTP/1.1" 200 -
tmp: [u'1', u'2', u'3']
params:[u'1', u'2', u'3']
tmp: [u'1', u'2', u'3']
127.0.0.1 - - [15/Mar/2018 00:23:28] "GET /detail/1+2+3/ HTTP/1.1" 200 -
params:[u'1', u'2', u'3']

• 页面

  地址：http://127.0.0.1:5000/detail/1+2+3/
  页面显示：success for url

这里有点不好理解，很难理解，大家看懂更好，看不懂就知道有这么回事就好，后面还会有这里的说明的。