Airflow自定义SSO认证完成后如何重定向用户至首页？

你在继承AuthOAuthView实现自定义CustomAuthRemoteUserView时，完成用户信息拉取、权限映射后，按以下3步处理即可完成跳转，不需要额外找零散的路由逻辑：

• 完成用户登录态写入
  拿到SSO返回的用户信息字典后，先同步为Airflow FAB（Flask AppBuilder，Airflow Web端底层权限框架）可识别的用户对象，再写入登录态，参考代码：

from flask import redirect, request
from flask_login import login_user
from airflow.auth.managers.utils.fab import get_airflow_appbuilder_role_names_from_oauth_user_info

class CustomAuthRemoteUserView(AuthOAuthView):
    # 你已经实现的SSO回调逻辑，拿到user_info字典后追加以下逻辑
    def handle_sso_callback(self, user_info: dict):
        # 按你的业务规则映射Airflow角色，可直接用内置方法也可自定义
        target_roles = get_airflow_appbuilder_role_names_from_oauth_user_info(user_info)
        # 同步本地用户，不存在则自动创建
        airflow_user = self.appbuilder.sm.auth_user_oauth(
            provider=self.oauth_providers[0]["name"],
            response=user_info,
            role_names=target_roles
        )
        # 写入Flask-Login全局登录态
        login_user(airflow_user)

• 获取合法的重定向目标地址
  不要硬编码首页路由，优先兼容用户登录前的访问目标，同时做安全校验避免开放重定向漏洞：

# 默认取Airflow当前版本配置的首页地址，自动适配2.x全版本差异
        redirect_target = self.appbuilder.get_url_for_index
        # 如果登录前用户正在访问特定DAG/页面，优先跳回原地址
        req_next = request.args.get("next")
        if req_next and self.appbuilder.is_safe_url(req_next):
            redirect_target = req_next

• 执行重定向响应
  直接返回Flask重定向响应即可，不需要额外封装逻辑：

return redirect(redirect_target)

避坑提示：

1. 不要硬编码/、/home、/dags这类路由作为首页，不同Airflow小版本默认首页路由存在差异，用self.appbuilder.get_url_for_index会自动读取配置适配
2. 必须调用login_user写入登录态，否则跳转到首页后会因为无有效登录凭证被重新拦截回登录页
3. 必须对next参数做安全校验，不要直接跳转用户传入的任意地址，避免产生安全漏洞

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