flask项目统一捕获异常并自定义异常信息

背景：

     在日常编写了一些flask项目，主要用于测试方面的mock场景，但迭代到后期发现有时候会抛出各种乱七八糟的异常或者直接500且没有任何异常信息，这种情况一般都是某个代码块没有用try来进行异常捕获；又或者是api的地址编写有误，直接抛出404了，这类提示都很不友好，故结合网上的教材，进行了针对异常的统一捕获并自动以了异常信息，用于接口回显，这样相对来说，对于使用者友好一些

 

errorhandler

errorhandler捕捉当前app或蓝图的状态码，并进行自定制处理

下面是一份示例代码，admin是一个蓝本或者app，发生404错误或500错误，会返回一个Json对象给请求段

from flask import jsonify
from . import admin
@admin.errorhandler(404)
def error_404(error):
    """这个handler可以catch住所有abort(404)以及找不到对应router的处理请求"""
    response = dict(status=0, message="404 Not Found")
    return jsonify(response), 404
@admin.errorhandler(Exception)
def error_500(error):
    """这个handler可以catch住所有的abort(500)和raise exeception."""
    response = dict(status=0, message="500 Error")
    return jsonify(response), 400
class MyError(Exception):
    """自定义错误类"""
    pass
@admin.errorhandler(MyError)
def MyErrorHandle(error):
    response = dict(status=0, message="400 Error")
    return jsonify(response), 400

 

app_errorhandler

app_errorhandler捕捉全局状态码，并进行自定制异常处理

在蓝本中编写错误处理程序有点不同，如果使用errorhandler修饰器，那么只有蓝本中的错误才会触发。如果想注册全局的错误处理程序，要用app_errorhandler

from . import auth
@auth.app_errorhandler(404)
def error_404(error):
    response = dict(status=0, message="404 Not Found")
    return jsonify(response), 404

 

实践 

以下代码为我自己项目中的实现：

首先新建一个error.py文件

import json
from flask import Blueprint, Response

exception = Blueprint('exception',__name__)

@exception.app_errorhandler(404)
def error_404(error):
    """这个handler可以catch住所有abort(404)以及找不到对应router的处理请求"""
    res = {"status": 404, "message": "404错误,找不到对应router"}
    return Response(json.dumps(res), mimetype='application/json')

@exception.app_errorhandler(405)
def error_405(error):
    """这个handler可以catch住所有abort(405)以及请求方式有误的请求"""
    res = {"status": 405, "message": "请求方式有误"}
    return Response(json.dumps(res), mimetype='application/json')

# @exception.app_errorhandler(Exception)
# def error_500(error):
#     """这个handler可以catch住所有的abort(500)和raise exeception."""
#     res = {"status": 500, "message": "系统内部错误"}
#     return Response(json.dumps(res), mimetype='application/json')

class MyError(Exception):
    """自定义错误类"""
    pass

 

然后在flask启动文件中新增如下代码：

# 导入error.py文件中的exception蓝图
from error import exception
from flask import Flask, request, Response


app = Flask(__name__)
# 注册蓝图，并指定其对应的前缀（url_prefix）
app.register_blueprint(exception, url_prefix='/error')
# 以下可以写业务代码


if __name__ == '__main__':
    app.run(host='0.0.0.0', port=8089, debug=False, threaded=True)

 

补充说明

　　当我们不是使用的工厂模式创建app时，app.errorhandler(401)，即可捕捉全局401状态；若使用了create_app方式创建app，则无法进行捕捉，若想捕捉，可以在蓝图中写，如admin.errorhandler(401)，即捕捉admin蓝图下所有401状态码，admin.app_errorhandler(401)，则是捕捉的全局的401状态码，即其他蓝图中的401状态，也会被捕捉，进行处理

 

 

文章来源：

https://www.cnblogs.com/huchong/p/9205651.html