Flask如何在渲染模板的同时保存request.headers信息并规避上下文错误

Flask的request对象绑定了活跃请求上下文，只有当请求被后端接收、处于处理流程中时，上下文才会激活，request才能正常调用。你在请求处理流程外直接执行find_aad()，自然会触发Working outside of request context.报错。

你需要把邮箱获取逻辑放到请求处理的生命周期内，根据使用场景可以选择以下两种实现方式：

方式1：直接在路由函数内调用 #

在首页路由的处理逻辑中调用find_aad()，获取后根据你的使用需求存储：

from flask import g, session

# 先配置SECRET_KEY才能使用session存储跨请求数据
app.secret_key = "替换为你自己的安全随机密钥"

@app.route('/')
def index():
    user_email = find_aad()
    # 仅当前请求内使用：存到g对象，同请求下的其他函数可以直接读取
    g.user_email = user_email
    # 跨请求保留用户身份：存到session
    if user_email:
        session["user_email"] = user_email
    # 也可以直接传给模板，在index.html中使用
    return render_template('index.html', user_email=user_email)


def find_aad():
    aad_email = request.headers.get('X-MS-CLIENT-PRINCIPAL-NAME')  # aad email
    return aad_email

后续需要调用邮箱时，直接从对应存储位置读取即可：

def your_business_func():
    # 同请求内读取
    current_email = getattr(g, "user_email", None)
    # 跨请求读取
    # current_email = session.get("user_email")
    if not current_email:
        # 处理未获取到用户身份的逻辑
        pass

方式2：用before_request钩子全局处理 #

如果多个路由都需要用到用户邮箱，可以用请求前置钩子统一获取，不用每个路由重复写逻辑：

from flask import g, session

app.secret_key = "替换为你自己的安全随机密钥"

@app.before_request
def get_aad_email_before_request():
    g.user_email = request.headers.get('X-MS-CLIENT-PRINCIPAL-NAME')
    # 自动存到session，无需手动在每个路由处理
    if g.user_email and "user_email" not in session:
        session["user_email"] = g.user_email

@app.route('/')
def index():
    # 直接读取已经提前存好的邮箱
    return render_template('index.html', user_email=g.user_email)

如果需要离线测试find_aad()函数，可以手动推送测试请求上下文，不会触发上下文报错：

with app.test_request_context(headers={"X-MS-CLIENT-PRINCIPAL-NAME": "test@example.com"}):
    test_email = find_aad()
    print(test_email) # 正常输出test@example.com

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