Flask通过Waitress部署时before_request不生效，本地调试正常

问题回顾 #

原本使用@bp.before_app_request实现全局认证钩子，本地运行正常，但添加测试后，所有请求（包括测试用的未认证请求）都会触发认证逻辑，导致测试失败。改为@bp.before_request后，本地调试（flask --app app_name run --debug）正常，但用Waitress部署时，该钩子完全不触发，引发未认证错误。

核心原因分析 #

• before_app_request：属于全局应用级钩子，所有请求（无论归属哪个蓝图）都会触发，因此测试时无法绕过认证。
• before_request：属于蓝图级钩子，仅当请求匹配该蓝图下的路由时才会触发。Waitress部署时钩子不触发，大概率是蓝图注册或路由归属的问题。

解决方案 #

方案1：修复蓝图级钩子的触发问题（推荐） #

确保蓝图正确注册，且路由确实归属目标蓝图：

1. 检查create_app函数中的蓝图注册代码，确认没有遗漏：

# app_name/__init__.py
from flask import Flask
from .auth import bp as auth_bp
from .index import bp as index_bp

def create_app():
    app = Flask(__name__)
    # 注册蓝图，若有url_prefix需确保路由匹配
    app.register_blueprint(auth_bp)
    app.register_blueprint(index_bp)
    # 其他配置...
    return app

2. 确认index.py中的路由属于对应蓝图，且蓝图导入正确：

# index.py
from flask import Blueprint, make_response, session

bp = Blueprint("index", __name__)  # 确保蓝图实例正确创建

@bp.route("/user")
def get_user():
    return make_response(session["user"], 200)

3. 验证Waitress部署时的请求路径是否匹配蓝图路由：如果蓝图注册时加了url_prefix（比如app.register_blueprint(index_bp, url_prefix='/api')），则请求路径应为/api/user而非/user。

方案2：给全局钩子添加测试环境跳过逻辑 #

如果更倾向于使用全局钩子，可通过环境判断跳过测试场景的认证：

# auth.py
from flask import current_app
bp = Blueprint("auth", __name__)

@bp.before_app_request
def validate_token():
    # 测试环境下直接跳过认证，模拟测试用户
    if current_app.config.get('TESTING'):
        session["user"] = {"id": "test_user"}
        return
    
    # 正常生产环境的认证逻辑
    # code...
    session["user"] = user

测试时，只需在测试用例中开启TESTING配置：

# test_app.py
def test_get_user():
    app = create_app()
    app.config['TESTING'] = True
    with app.test_client() as client:
        response = client.get("/user")
        assert response.status_code == 200

额外注意事项 #

• 检查Waitress启动命令是否正确调用了create_app函数，确保应用实例是通过蓝图注册后的正确实例。
• 若使用了Flask的SESSION_COOKIE_SECURE等生产环境配置，需确保测试环境的session正常工作。

内容的提问来源于stack exchange，提问作者Masamune