apispec 使用教程-优快云博客

本文链接：https://blog.youkuaiyun.com/gitblog_00945/article/details/141150817

apispec 使用教程

apispecA pluggable API specification generator. Currently supports the OpenAPI Specification (f.k.a. the Swagger specification)..项目地址:https://gitcode.com/gh_mirrors/ap/apispec

1. 项目介绍

apispec 是一个可插拔的 API 规范生成器，主要支持 OpenAPI 规范（以前称为 Swagger 规范）。该项目设计用于帮助开发者轻松地从 Python 代码中提取接口规范，并自动生成相应的文档。apispec 提供了与多个流行库（如 Flask 和 Falcon）集成的能力，使得在现有 Web 应用程序中添加 API 文档变得简单。

核心特性

支持 OpenAPI v3 规范。
可以使用多种数据序列化库，比如 marshmallow。
集成了多种 Web 框架，如 Flask 和 Falcon。
插件式架构，易于扩展。

2. 项目快速启动

要开始使用 apispec，首先确保已经安装了所需的依赖库，包括 apispec 和 marshmallow：

pip install apispec marshmallow

然后，在你的 Flask 应用中集成 apispec：

from flask import Flask
from apispec.ext.marshmallow import MarshmallowPlugin
from apispec import APISpec

app = Flask(__name__)
app.config['APISPEC_SPEC'] = APISpec(
    title="My API",
    version="1.0",
    plugins=[MarshmallowPlugin()],
)

# 注册视图到 specs
from your_app.views import my_view  # 假设你有一个名为 my_view 的视图函数
docs = FlaskApiSpec(app)
docs.register(my_view)

接下来，你可以通过访问 /swagger/ URL 来查看生成的 Swagger UI 文档。

@app.route('/swagger/')
def swagger_ui():
    return render_template('swagger_ui.html')  # 自定义或使用现有的模板来展示 Swagger UI

3. 应用案例和最佳实践

3.1 使用 Schema 定义 API 参数

apispec 可以与 marshmallow 中的 Schema 配合，自动将 Schema 转换为 OpenAPI 规范：

from marshmallow import Schema, fields

class PetSchema(Schema):
    name = fields.Str()
    age = fields.Int()

@doc(description='获取宠物列表', responses={200: PetSchema(many=True)})
@app.route('/pets')
def get_pets():
    pets = [{'name': 'Fido', 'age': 5}, {'name': 'Whiskers', 'age': 3}]
    return jsonify(pets)

3.2 处理错误响应

apispec 还允许你定义错误响应：

@doc(responses={
    404: {'description': 'Pet not found'},
    500: {'description': 'Server error'}
})
@app.route('/pet/<int:id>')
def get_pet(id):
    pet = find_pet(id)
    if not pet:
        abort(404)
    return pet