flask学习文档03-异常捕获和请求钩子

版权声明:内容版权为CSDN用户:kayseen 所有,若您需要引用、转载,需要注明来源及原文链接 https://blog.csdn.net/m0_43394876/article/details/88540158

1. 异常捕获

1. HTTP 异常主动抛出

  • abort 方法
    • 抛出一个给定状态代码的 HTTPException 或者 指定响应,例如想要用一个页面未找到异常来终止请求,你可以调用 abort(404)。
  • 参数:
    • code – HTTP的错误状态码
# abort(404)
abort(500)

抛出状态码的话,只能抛出 HTTP 协议的错误状态码

2. 捕获错误

  • errorhandler 装饰器
    • 注册一个错误处理程序,当程序抛出指定错误状态码的时候,就会调用该装饰器所装饰的方法
  • 参数:
    • code_or_exception – HTTP的错误状态码或指定异常
  • 例如统一处理状态码为500的错误给用户友好的提示:
@app.errorhandler(500)
def internal_server_error(e):
    return '服务器搬家了'
  • 捕获指定异常
@app.errorhandler(ZeroDivisionError)
def zero_division_error(e):
    return '除数不能为0'

3.异常捕获代码

from flask import Flask
from flask import abort

app = Flask(__name__)

@app.route('/')
def index():
    return 'index'

@app.route('/demo1')
def demo1():
    # 主动抛出HTTP指定错误状态码
    # abort(404)
    a = 0
    b = 1 / a
    return 'demo1'


# 使用装饰器的形式去捕获指定的错误码和异常
@app.errorhandler(404)
def page_not_found(error):  #error 是报错的信息
    return "咦~页面不见了~"


@app.errorhandler(ZeroDivisionError)
def zero_division_error(error):
    return '除数不能为0'


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

2.请求钩子

1.什么是请求钩子

在客户端和服务器交互的过程中,有些准备工作或扫尾工作需要处理,比如:

  • 在请求开始时,建立数据库连接;
  • 在请求开始时,根据需求进行权限校验;
  • 在请求结束时,指定数据的交互格式;

为了让每个视图函数避免编写重复功能的代码,Flask提供了通用设施的功能,即请求钩子。

请求钩子是通过装饰器的形式实现,Flask支持如下四种请求钩子:

  • before_first_request
    • 在处理第一个请求前执行
  • before_request
    • 在每次请求前执行
    • 如果在某修饰的函数中返回了一个响应,视图函数将不再被调用
  • after_request
    • 如果没有抛出错误,在每次请求后执行
    • 接受一个参数:视图函数作出的响应
    • 在此函数中可以对响应值在返回之前做最后一步修改处理
    • 需要将参数中的响应在此参数中进行返回
  • teardown_request:
    • 在每次请求后执行(有异常的话)
    • 接受一个参数:错误信息,如果有相关错误抛出

2.代码测试1

from flask import Flask
from flask import abort

app = Flask(__name__)

# 在第一次请求之前调用,可以在此方法内部做一些初始化操作
@app.before_first_request
def before_first_request():
    print("before_first_request")

# 在每一次请求之前调用,这时候已经有请求了,可能在这个方法里面做请求的校验
# 如果请求的校验不成功,可以直接在此方法中进行响应,直接return之后那么就不会执行视图函数
@app.before_request
def before_request():
    print("before_request")
    # if 请求不符合条件:
    #     return "laowang"

# 在执行完视图函数之后会调用,并且会把视图函数所生成的响应传入,可以在此方法中对响应做最后一步统一的处理
@app.after_request
def after_request(response):
    print("after_request")
    response.headers["Content-Type"] = "application/json"
    return response

# 请每一次请求之后都会调用,会接受一个参数,参数是服务器出现的错误信息
@app.teardown_request
def teardown_request(e):
    print("teardown_request")

@app.route('/')
def index():
    return 'index'

if __name__ == '__main__':
    app.run(debug=True)
  • 在第1次请求时的打印:
before_first_request
before_request
after_request
teardown_request
  • 在第2次请求时的打印:
before_request
after_request
teardown_request

3.代码测试2

from flask import Flask

app = Flask(__name__)

@app.before_first_request
def before_first_request():
    """在第一次请求之前会访问该函数"""
    print('before_first_request')


@app.before_request
def before_request():
    """在每次请求之前都会调用"""
    print('before_request')
    # 可以对一些非法的请求进行阻止
    # if 如果ip在黑名单:
    #     return "你在~~~"


@app.after_request
def after_request(response):  #要有参数
    """在请求之后会调用,并且函数里面接受一个参数:响应,还需要将响应进行返回"""
    print('after_request')
    # 可以在此函数中对响应数据做统一的处理
    return response   #要将参数返回


@app.teardown_request
def teardown_request(error):   #要有参数
    """在请求之后会执行,如果请求的函数报有异常,会把具体异常传入到此函数"""
    print('teardown_request')


@app.route('/')
def index():
    return 'index'


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

猜你喜欢

转载自blog.csdn.net/m0_43394876/article/details/88540158