Python Web开发03

Echo_Wish

发布于 2023-11-30 19:18:15

1360

发布于 2023-11-30 19:18:15

flask 自动生成swagger 的api接口文档

安装flask-restplus 第三方包，使用pip install flask-restplus 安装即可。
在一个普通的正常的flask 应用项目结构下，应该是在extensions.py 下进行代码书写，因为这是进行程序扩展的代码编写处。导包，导入flask_restplus 下的Api，Resource，fields。获取一个app 实例。并进行namespace 的书写。代码如下：

    api = Api(doc='/swagger')
    api.init_app(app, version='1.0', title='Data Visualization And Analysis API',
                 description='A Charting and Data analysis API')
    bar_line = api.namespace('drawing bar and line', path='/', description="draw bar and line chart")
    pie = api.namespace('drawing pie', path='/', description="draw pie chart")
    radar = api.namespace('drawing radar', path='/', description="draw radar chart")
    scatter = api.namespace('drawing scatter', path='/', description="draw scatter chart")
    data_analysis = api.namespace('data analysis', path='/', description="data analysis")

获取一个实例化Api对象，app是一个实例化的flask对象，通过在实例化Api对象时通过doc 参数可以指定最终的接口文档通过什么路由可以访问到。api.namespace ：是命名空间，很多接口都有get，post，命名空间把他们分隔开，可理解为蓝图。 path：代表他们的路由地址，这里让他们都使用route的地址，不写的话会把命名空间的name加到路由地址的最前面 description：是对该组下所有接口的总的一个注释。

通过api.model 来描述请求的request 和响应的response，通过api.namespace.parser 来描述请求的headers 参数。

代码示例如下：

    # 使用parser 来描述接口的headers 和 query
    bar_line_parameter = bar_line.parser()
    bar_line_parameter.add_argument('Authorization', location='headers', default="a")
    bar_line_parameter.add_argument('User-Agent', location='headers', default="ua")
    # 使用model 来描述接口的请求体
    bar_line_model = api.model('Bar_Line_Request', {
        "type": fields.String(default="bar"),
        "title": DictItem(required=True, default={}, description="chart title option"),
        "item": DictItem(required=True, default={}, description="chart series item option"),
        "xaxis": DictItem(required=True, default={}, description="chart xaxis option"),
        "yaxis": DictItem(required=True, default={}, description="chart yaxis option"),
        "grid": DictItem(required=True, default={}, description="chart grid option"),
        "legend": DictItem(required=True, default={}, description="chart legend option"),
        "tooltip": DictItem(required=True, default={}, description="chart tooltip option"),
        "background": DictItem(required=True, default={}, description="chart tooltip option"),
    }, description="request api needed body parameter")
    # 使用model 来描述接口的响应
    bar_line_response = api.model('bar_line_Response', {
        'data': DictItem(required=True, default=bar_line_response_data_default, description="chart option"),
        'status': fields.Integer(required=True, default=200, description="response status"),
        'msg': fields.String(required=True, default="successful", description="api response message")
    })

如上，其中bar_line 是api.namespace() 的返回对象，使用parser 的add_argument() 方法来添加headers ，或query 中请求所需参数，同时可以定义默认值。使用model 来描述请求的请求体，响应也是。model 需要指定一个唯一的key 值，和一个 {} 字典键值对，在该字典键值对中key值是所需传输的name，value 是通过flask-restplus 下的fields 来指定数据类型以及默认值描述的值。如果fields中提供的数据类型满足不了使用，可以通过自定义类继承fields.Row ，并且实现format 方法，来使用自定义的数据类型。代码中的DictItem 就是自定义数据类型。

将以上定义的model，parser 应用到接口上。通过装饰器的方式，代码如下。

    # 使用api.namespace.route 来指定接口的访问路由，使用description来描述接口
    @bar_line.route('/api/chart/draw/bar_and_line',
                    doc={"description": "返回图表的echarts 配置项信息，当请求参数配置为空时返回默认配置的图表即示例样例，否则根据请求的配置参数返回对应的完整的图表配置信息"})
    # 这里的api.namespace.expect 需要与上面的api.namespace.expect 联用
    @bar_line.expect(bar_line_parameter)
    # 需要继承于Resource类
    class BarLineOption(Resource):
        # doc 用于描述接口，body=X 指定请求的body描述
        @bar_line.doc('Return to bar and line chart configuration item')
        @bar_line.doc(body=bar_line_model)
        # marshal_with 指定响应的描述
        @bar_line.marshal_with(bar_line_response)
        # 接口支持什么方法，就定义一个什么方法。
        def post(self):
            return {'data': {},
                    "status": 200, "msg": "successful"}

本文参与腾讯云自媒体同步曝光计划，分享自作者个人站点/博客。

原始发表：2023-11-30，如有侵权请联系 cloudcommunity@tencent.com 删除

python