Flask笔记四之异常处理

2023-12-29 06:44:04

本文首发于公众号:Hunter后端

原文链接:Flask笔记四之异常处理

在系统运行中,有时候需要处理报错异常。

异常的来源可能是系统在运行中的报错,比如错误的运算的有 1/0 这种等,还有的是访问了不存在接口,又或者,我们在查验接口参数的时候,发现没有传必传参数,需要手动触发一个报错等。

这里就介绍如何在 Flask 中进行异常处理。

本篇笔记的代码都已经提交到 github 上,可使用下面的操作获取代码:

git clone https://github.com/x1204604036/flask_backend.git

1、异常处理示例

这里以请求的接口不存在为示例介绍一下异常处理流程。

处理一个异常,需要先对指定的异常进行注册,对于请求接口不存在这种请求相关的异常,Flask 里定义了一个 HTTPException 异常对其进行处理

这里我们在 app/ 下创建一个文件夹 utils,再创建一个文件 exception_handler.py:

# app/utils/exception_handler.py

from werkzeug.exceptions import HTTPException
from flask import jsonify

def init_error_exception(app):

    @app.errorhandler(HTTPException)
    def handler_http_exception(exception):
        print(exception)
        return jsonify({"code": -1, "msg": exception.description}), exception.code

这里创建了一个异常处理的函数,通过 @app.errorhandler(HTTPException) 的方式注册,在 app/__init__.py 中引入:

from flask import Flask
from app.utils.exception_handler import init_error_exception

def create_app():
    app = Flask(__name__)

    init_error_exception(app)

    return app

flask.jsonify 用来返回 json 数据给客户端,后面跟着的是 HTTP 的 status code,就是 200,500 那些表示请求状态的的状态码

这里我们用 postman 请求一个不存在的接口,就可以看到返回的信息如下:

{
    "code": -1,
    "msg": "The requested URL was not found on the server. If you entered the URL manually please check your spelling and try again."
}

status code 是 404,表示接口不存在

这样,关于 HTTP 请求的异常处理就从注册,到触发,到返回结果完成了一个闭环。

2、其他异常处理

除了上面的请求异常,可能还有一些异常,比如我们在接口里进行了 1/0 等不合法的运算,为了捕捉这种报错,我们可以如下操作:

ERROR_HTTP_CODE = 417


def init_error_exception(app):
    @app.errorhandler(Exception)
    def server_exception(exception):
        print(exception)
        return jsonify({"code": -1, "msg": "内部错误"}), ERROR_HTTP_CODE

这样,在系统里发生异常异常之后,就会被捕捉,返回给客户端 内部错误 的信息

还有一些时候,举个例子,比如在接口里,客户端没有传来一些必传的参数,我们可以手动触发一个报错,自定义报错信息和HTTP状态码,可以如下这样操作:

class UserException(Exception):
    def __init__(self, code=-1, msg="error", http_code=417):
        self.code = code
        self.msg = msg
        self.http_code = http_code

def init_error_exception(app):
    @app.errorhandler(UserException)
    def user_exception(exception):
        print(exception)
        return jsonify({"code": exception.code, "msg": exception.msg}), exception.http_code

然后我们在接口里这样调用触发异常:

raise UserException(msg="手动测试error", http_code=417)

我们可以传入 code,msg 和 http_code 参数,用于返回给前端

以上这些介绍的都是异常的处理,还可以在处理异常的时候打印日志,用于记录报错的信息,这个我们后面介绍日志的时候再介绍。

3、代码总览

以下给出前面介绍的三种异常处理的代码:

# app/utils/exception_handler.py

from werkzeug.exceptions import HTTPException
from flask import jsonify


ERROR_HTTP_CODE = 417


class UserException(Exception):
    def __init__(self, code=-1, msg="error", http_code=417):
        self.code = code
        self.msg = msg
        self.http_code = http_code

def init_error_exception(app):
    @app.errorhandler(HTTPException)
    def handler_http_exception(exception):
        print(exception)
        return jsonify({"code": -1, "msg": exception.description}), exception.code

    @app.errorhandler(Exception)
    def server_exception(exception):
        print(exception)
        return jsonify({"code": -1, "msg": "内部错误"}), ERROR_HTTP_CODE


    @app.errorhandler(UserException)
    def user_exception(exception):
        print(exception)
        return jsonify({"code": exception.code, "msg": exception.msg}), exception.http_code

记得要在 app/__init__.py 中引入初始化处理哦

文章来源:https://blog.csdn.net/weixin_43354181/article/details/135281007
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。