引入flask_restful后，flask errorhandler在gunicorn中不起作用

Question

我正在烧瓶中开发 REST API 并计划在 Gunicorn 中运行它。在我的应用程序中，一个用户定义的异常由flask errorhandler 装饰器处理。它在 Flask 内置网络服务器和 Gunicorn 中都能正常工作。响应可以从装饰函数生成。引入flask_restful 后，内置服务器工作正常，但在Gunicorn 中，响应始终为{"message": "Internal Server Error"}。

这是源代码：myapp.py

from flask import Flask, jsonify, make_response
from flask_restful import Api, Resource


app = Flask(__name__)
api = Api(app)


class OrderNotExistError(Exception):
    def __init__(self, order_id):
        self.message = 'Order [{order_id}] does not exist.'.format(order_id=order_id)


@app.errorhandler(OrderNotExistError)
def order_not_exist(error):
    return make_response(jsonify({'message': error.message}), 404)


class OrderAPI(Resource):
    def get(self, order_id):
        raise OrderNotExistError(order_id)


api.add_resource(OrderAPI, '/orders/<int:order_id>', endpoint='order')


@app.route("/o/<int:order_id>")
def get_order(order_id):
    raise OrderNotExistError(order_id)


if __name__ == '__main__':
    app.debug = True
    app.run()

在 Gunicorn 中运行：gunicorn -w4 -b0.0.0.0:8000 myapp:app

访问“ http://127.0.0.1:8000/o/123 ”
它响应：{“消息”：“订单[123]不存在。”}。错误处理程序工作正常。

访问“ http://127.0.0.1:8000/orders/123 ”
它响应：{“消息”：“内部服务器错误”}。似乎错误处理程序不起作用。

在烧瓶内置服务器中运行时，不会出现此问题。

有人遇到同样的问题吗？这是flask_restful 或Gunicorn 中的错误吗？如何处理这个问题？

Answer 1

这是因为有两个级别的错误处理程序，一个在应用程序级别，一个在 API 级别。您正在直接调用 api，因此应用程序看不到这一点。（这解释了为什么通过 app.route 添加的路由会捕获异常，而不是通过 api.add_resource 添加的路由捕获异常）。

要捕获此错误，您需要覆盖 Werkzeug 的异常，这正是 Flask-restful 使用的。以下代码应该修复它：

errors={
    'InternalServerError': {
    'status': 500,
    'message': 'Internal Server Error'
},
}
api = Api(app, errors=errors)