经常看到有人把 FastAPI 和 Flask 放到一起比较,但是却没有意识到这完全是两种东西——前者是基于 Web 框架 Starlette 添加了 Web API 功能支持的(框架之上的)框架,而后者是和 Starlette 同类的通用 Web 框架。你怎么能让小明和骑电动车的小军赛跑然后还夸小军好快好强?为了让框架 PK 爱好者们有一个更公平的比较对象,从一份 0.6.2 版本的 fork 开始,我实现了一个基于 Flask 的 Web API 框架——:
- 主页:
- GitHub: https://github.com/greyli/apiflask
- Twitter: https://twitter.com/apiflask
APIFlask 在 Flask 的基础上添加了更多 Web API 相关的功能支持,核心特性包括:
- 更多方便的装饰器,比如
@app.input()
、@app.output()
、@app.get()
、@app.post()
等等 - 自动反序列化和验证请求格式,当请求数据不符合模式类要求时,会自动生成包含错误详细信息的错误响应(基于 )
- 自动格式化和序列化响应数据,在定义好响应模式后,你可以直接在视图函数返回一个模型类对象,或是返回字典(基于 )
- 自动生成 文件,你可以把这个文件导入到 API 调试工具或是用来生成客户端代码(基于 )
- 自动生成交互式 API 文档,并自动为蓝本和视图设置对应的标签分类(基于 and )
- 自动为 HTTP 错误生成 JSON 格式的错误响应
下面是一个最基础的示例程序:
from apiflask import APIFlask, Schema, abort from apiflask.fields import Integer, String from apiflask.validators import Length, OneOf app = APIFlask(__name__) # 可以使用 title 和 version 参数来自定义 API 的名称和版本 pets = [ { 'id': 0, 'name': 'Kitty', 'category': 'cat' }, { 'id': 1, 'name': 'Coco', 'category': 'dog' } ] # 定义一个请求数据模式类 class PetInSchema(Schema): name = String(required=True, validate=Length(0, 10)) # 可以使用 description 参数添加字段描述 category = String(required=True, validate=OneOf(['dog', 'cat'])) # 定义一个响应数据模式类 class PetOutSchema(Schema): id = Integer() name = String() category = String() @app.get('/pets/<int:pet_id>') @app.output(PetOutSchema) # 使用 @output 装饰器标记响应数据模式 def get_pet(pet_id): if pet_id > len(pets) - 1: abort(404) # 在真实程序里,你可以直接返回 ORM 模型类的实例,比如 # return Pet.query.get(1) return pets[pet_id] @app.patch('/pets/<int:pet_id>') @app.input(PetInSchema(partial=True)) # 使用 @input 装饰器标记请求数据模式 @app.output(PetOutSchema) def update_pet(pet_id, json_data): # 通过验证后的请求数据字典会注入到视图函数,默认参数名为 json_data if pet_id > len(pets) - 1: abort(404) for attr, value in json_data.items(): pets[pet_id][attr] = value return pets[pet_id]
P.S. 你也可以使用类视图(class-based views),具体示例见这里。
如果你想在你的电脑上运行这个示例,可以先用下面的命令安装 APIFlask(需要 Python 3.7 及以上版本,Flask 1.1 及以上版本):
Linux 和 macOS:
$ pip3 install apiflask
Windows:
> pip install apiflask
安装完成后把上面的代码保存到文件 app.py
,然后执行下面的命令运行程序:
$ flask run
现在你可以在浏览器访问 http://localhost:5000/docs 查看基于 Swagger UI 自动生成的交互式 API 文档:
或者访问 http://localhost:5000/redoc 查看基于 Redoc 生成的 API 文档:
访问 http://localhost:5000/openapi.json 可以获取自动生成的 OpenAPI spec 文件。
这个示例程序的完整版本可以在 https://github.com/greyli/apiflask/tree/master/examples 看到。如果你想了解更多用法,可以阅读文档中的一章(目前只有英文)。
- 创建程序实例的时候使用
APIFlask
类(from apiflask import APIFlask
)。 - 创建蓝本实例的时候使用
APIBlueprint
类(from apiflask import APIBlueprint
)。 - 使用
apiflask.abort()
函数返回 JSON 格式的错误响应。
以下面的 Flask 程序为例:
from flask import Flask, request, escape app = Flask(__name__) @app.route('/') def hello(): name = request.args.get('name', 'Human') return f'Hello, {escape(name)}'
迁移到 APIFlask 只需要改动两行代码:
from apiflask import APIFlask # 第一行 from flask import request, escape app = APIFlask(__name__) # 第二行 @app.route('/') def hello(): name = request.args.get('name', 'Human') return f'Hello, {escape(name)}'
欢迎提出改进建议,报告 bug 或是分享其他任何相关的想法。你也可以在