Flask 什么是 Werkzeug

在本文中，我们将介绍 Flask 框架中的一个重要组件 Werkzeug。Werkzeug 是一个 Python 的 WSGI 工具集，它为 Web 应用提供了一些底层的功能和抽象。作为 Flask 的核心库之一，Werkzeug 提供了一些重要的功能，如路由匹配、HTTP 请求和响应处理、文件上传等。

阅读更多：Flask 教程

Werkzeug 的功能和重要模块

Werkzeug 提供了丰富的功能和多个重要模块，下面我们将对一些重要的模块和功能进行介绍。

路由匹配

Flask 使用 Werkzeug 来处理路由匹配。Werkzeug 提供了一个非常强大的路由匹配器，能够根据不同的 URL 规则来匹配对应的路由。例如，我们可以定义一个 URL 规则为 “/users/“，通过 Werkzeug 的路由匹配器，可以很方便地匹配到对应的 URL，同时提取出用户的 id。

from werkzeug.routing import Map, Rule
from werkzeug.exceptions import NotFound

# 定义路由规则
rules = [Rule('/users/<int:user_id>', endpoint='user')]
# 创建路由映射
url_map = Map(rules)

# 匹配 URL
try:
    endpoint, args = url_map.bind('').match('/users/123')
    print(endpoint)  # 输出 'user'
    print(args)  # 输出 {'user_id': 123}
except NotFound:
    print('URL not found')

HTTP 请求和响应处理

Werkzeug 提供了方便的接口来处理 HTTP 请求和响应。通过 Werkzeug 的 Request 和 Response 对象，我们可以轻松地获取请求的参数、设置响应的内容和状态码。

from werkzeug.wrappers import Request, Response

@Request.application
def application(request):
    # 获取请求参数
    username = request.args.get('username')

    # 构造响应
    response = Response()
    response.headers['Content-Type'] = 'text/plain'
    response.data = f'Hello, {username}!'.encode('utf-8')

    return response

if __name__ == '__main__':
    from werkzeug.serving import run_simple
    run_simple('', 5000, application)

通过上述代码，我们创建了一个简单的应用程序，它接受一个名为 username 的参数，并返回一个包含欢迎信息的响应。

文件上传

Werkzeug 还提供了处理文件上传的功能。我们可以使用 Werkzeug 的 FileStorage 对象来处理客户端上传的文件。

from werkzeug.utils import secure_filename
from werkzeug.datastructures import FileStorage

def save_uploaded_file(file: FileStorage, upload_dir: str) -> str:
    if file.filename == '':
        raise ValueError('No selected file')

    filename = secure_filename(file.filename)
    file_path = os.path.join(upload_dir, filename)
    file.save(file_path)

    return file_path

@app.route('/upload', methods=['POST'])
def upload_file():
    uploaded_file = request.files['file']
    file_path = save_uploaded_file(uploaded_file, '/path/to/upload/dir')

    return f'File saved at: {file_path}'