flask-钩子函数&g对象

时间：2023-01-01 14:14:57

常用钩子函数

在Flask中钩子函数是使用特定的装饰器装饰的函数。钩子函数可以在正常执行的代码中，插入一段自己想要执行的代码。那么这种函数就叫做钩子函数。（hook）

  • before_first_request：顾名思义，注册一个在处理第一个请求之前运行的函数
@app.before_first_request   # 第一次访问的时候调用

def first_request():

    print('first time request')
  • before_request：注册一个在处理请求之前运行的函数。
@app.before_request   # 每次接受到请求的时候都会执行

def before_request():

    if not hasattr(g,'user'):

        setattr(g,'user','xxxx')

  • teardown_appcontext：不管是否有异常，注册的函数都会在每次请求之后执行。

@app.teardown_appcontext

def teardown(exc=None):

    if exc is None:

        db.session.commit()

    else:

        db.session.rollback()

        db.session.remove()
  • template_filter：在使用Jinja2模板的时候自定义过滤器。比如可以增加一个upper的过滤器
@app.template_filter

def upper_filter(s):

    return s.upper()
  • `context_processor`：必须返回一个字典。这个字典中的值在所有模版中都可以使用。这个钩子函数的函数是，如果一些在很多模版中都要用到的变量，那么就可以使用这个钩子函数来返回，而不用在每个视图函数中的`render_template`中去写，这样可以让代码更加简洁和好维护。
@app.context_processor

def my_context_processor():

    return {'current_user':'xxx'}
  • `errorhandler`：在发生一些异常的时候，比如404错误，比如500错误。可以使用`errorhandler`来出来。需要注意几点：
        * 在errorhandler装饰的钩子函数下，记得要返回相应的状态码。
        * 在errorhandler装饰的钩子函数中，必须要写一个参数，来接收错误的信息，如果没有参数，就会直接报错。
        * 使用`flask.abort`可以手动的抛出相应的错误，比如开发者在发现参数不正确的时候可以自己手动的抛出一个400错误。
@app.errorhandler(404)

def page_not_found(error):

    return 'This page does not exist',404
from flask import abort

def my_list():

# 如果user_id在数据库中不存在，这时候我就让他跳转到400错误

     abort(400)   # 抛出400错误

视图

g对象

　　g绑定到了Local对象，是线程隔离的。该对象的作用是绑定数据，绑定的数据可以在全局使用！

@app.route('/')

def index():

    username = request.args.get('username')
    g.username = username  # 绑定
    log_a()
    log_b()
    log_c(
from flask import g

def log_a():

    print('log a %s' % g.username)

def log_b():

    print('log b %s' % g.username)

def log_c():

    print('log c %s' % g.username)

相关文章