flask - 蓝图初始化，我可以在第一次请求蓝图之前运行一个函数吗

Question

是否可以在对特定的第一个请求之前运行一个函数blueprint？

@my_blueprint.before_first_request
def init_my_blueprint():
    print 'yes'

目前这将产生以下错误：

AttributeError: 'Blueprint' object has no attribute 'before_first_request'

Answer 1

蓝图等价物称为@Blueprint.before_app_first_request：

@my_blueprint.before_app_first_request
def init_my_blueprint():
    print('yes')

该名称反映了它在任何请求之前被调用，而不仅仅是此蓝图特定的请求。

对于蓝图要处理的第一个请求，没有运行代码的钩子。您可以使用测试它是否已运行的@Blueprint.before_request处理程序来模拟它：

from threading import Lock

my_blueprint._before_request_lock = Lock()
my_blueprint._got_first_request = False

@my_blueprint.before_request
def init_my_blueprint():
    if my_blueprint._got_first_request:
        return
    with my_blueprint._before_request_lock:
        if my_blueprint._got_first_request:
            return

        # first request, execute what you need.
        print('yes')

        # mark first request handled *last*
        my_blueprint._got_first_request = True

这模仿了 Flask 在这里所做的事情；需要锁定，因为单独的线程可能会争先恐后地成为第一个。