flask中的abort函数和自定义异常

from flask import Flask, abort

app = Flask(__name__)

@app.route('/page-not-found')
def page_not_found():
    abort(404)

@app.errorhandler(404) # 这个函数的目的是对于一切404结果的获取
def page_not_found_error(error):
    return "404 Error: Page Not Found", 404

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

from flask import Flask, jsonify

# 自定义异常类
class CustomError(Exception):
    # 默认状态码为 400
    status_code = 400

    # 初始化方法，接受消息、状态码和载荷
    def __init__(self, message, status_code=None, payload=None):
        super().__init__(self)
        self.message = message
        # 如果传入状态码则使用传入的状态码，否则使用默认状态码
        if status_code is not None:
            self.status_code = status_code
        self.payload = payload

    # 将异常信息转换为字典格式
    def to_dict(self):
        rv = dict(self.payload or ())
        rv['error'] = {
            'message': self.message,
            'status_code': self.status_code
        }
        return rv

# 创建 Flask 应用
app = Flask(__name__)

# 路由，抛出自定义异常
@app.route('/custom-error')
def custom_error():
    # 抛出自定义异常
    raise CustomError('An error occurred while processing your request.', status_code=401)

# 错误处理器，处理自定义异常
@app.errorhandler(CustomError)
def handle_custom_error(error):
    # 使用 jsonify 方法将异常信息转换为 JSON 格式
    response = jsonify(error.to_dict())
    # 设置响应状态码为异常中定义的状态码
    response.status_code = error.status_code
    return response

# 主函数，启动 Flask 应用
if __name__ == '__main__':
    app.run(debug=True)