问题描述:

使用 Flask 制作了一个demo,代码如下

from flask import Flask

app = Flask(__name__)


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


if __name__ == '__main__':
    # 启动 Flask
    print('#### Flask Start... ####')
    app.run(host='127.0.0.1', port=80, debug=True)

启动后发现出现了 print 语句 打印了两次,也就是程序执行了两次


原因分析:

经过一番查阅资料后,原来是 Flask Debug 模式导致的这个问题!

在 Debug 模式下,Flask 中的 Werkzeug WSGI 套件默认会启动一个额外的进程来监控文件变化,当我们修改文件时,自动帮我们重启项目


解决方案:

要解决这个启动两次的问题,有以下几种办法:

  • 取消自动重启

    在 Debug 模式下,为了方便调试,Flask 提供了当文件变化的时候自动重启实例的功能。关闭这个功能就可以避免初始化两次的情况,但同时也丧失了自动重启的功能

    app.run(debug=True, use_reloader=False)
    
  • 判断主进程是否执行

    restart_with_reloader 中,我们可以看到在新进程启动前,环境变量 WERKZEUG_RUN_MAIN 被置为 true

    new_environ['WERKZEUG_RUN_MAIN'] = 'true'
    

    通过判断这个变量的值,我们就能保证在启动时仅执行一次

    if __name__ == '__main__':
    # 启动 Flask
    import os
    if os.environ.get('WERKZEUG_RUN_MAIN') == 'true':
        print('#### Flask Start... ####')
    app.run(host='127.0.0.1', port=80, debug=True)
    
  • 在第一次请求的时候执行

    使用 before_first_request 这个钩子函数,把执行放在 Flask 第一次收到请求的时候,这就避免了两次初始化的干扰

    @app.before_first_request
    def init():
    	print('#### Flask Start... ####')
    

参考