Python flask 如何让 response 按照一定的格式返回？

[tianxin8431]

想让每个 resp 都按照这样的三段式返回：

{
    "data": xxx,
    "msg": xxx,
    "code": xxx
}

现在想的是用装饰器，这样可以实现视图函数的返回都按照这个来；但是有一些框架返回的东西没有办法按照这个 schema 返回，想问还有什么更好的办法吗？

[linw1995]

试一下 WSGI middleware ？

[NLL]

after_request ？

[Yuxiaoy]

import json
from flask import Flask, Response


class MyResponse(Response):
def __init__(self, response, **kwargs):
response = json.dumps({'data': 'xxx', 'msg': 'xxx', 'code': 'xxx'})
return super(MyResponse, self).__init__(response, **kwargs)


app = Flask(__name__)
app.response_class = MyResponse


@app.route('/')
def index():
return 'I will be overrided'