python - OAuth登录完成后重定向到之前的URL(flask-dance)

Question

我正在开发一个 Flask 应用程序，允许用户使用 OAuth(以 Github 作为提供者)和 Flask-dance 库登录。由于某种原因，我无法在成功登录后重定向到我将用户发送到登录页面的页面。

当用户尝试连接到例如 http://localhost:6675/examples/tutorial.first/ 时，用户被重定向到登录页面，在 URL 中显示我们应该重定向到的页面 ( http://localhost:6675/login?next=%2Fexamples%2Ftutorial.first%2F )

问题是，当我成功使用 Github 登录后，应用程序只是返回到主页。

我正在查看 Flask-dance documentation make_github_blueprint() 函数的文档提到了参数 redirect_to 和 redirect_url，但是当我尝试使用它们时，我什至无法完成登录步骤。此外，它似乎只适用于静态地址，而理想情况下我想跳回到登录之前的页面。我还检查了 this SO question ，但问题似乎有所不同。

是否有关于如何在使用 Flask dance 登录后正确进行重定向的示例？

这里有一些可能相关的代码片段。在 init.py 文件中:

bp_github = make_github_blueprint(
    client_id="...",
    client_secret="...",
)
login_manager = LoginManager()
login_manager.login_github_view = 'github.login'
login_manager.login_view = 'login'

在 app.py 文件中:

@app.route("/login", methods=['GET', 'POST'])
def login():
    if current_user.is_authenticated:
        return flask.redirect(flask.url_for('/'))
    return flask.render_template('login.html')

@app.route("/logout")
@login_required
def logout():
    logout_user()
    flask.flash("You have logged out")
    return flask.redirect(flask.url_for("login"))

@oauth_authorized.connect_via(bp_github)
def logged_in(blueprint, token):
    """
    create/login local user on successful OAuth login with github
    :param blueprint:
    :param token:
    :return:
    """
    if not token:
        flask.flash("Failed to log in.", category="error")
        return False

    session = blueprint.session

    resp = session.get("/user")

    if not resp.ok:
        msg = "Failed to fetch user info."
        flask.flash(msg, category="error")
        return False

    user_id = str(info["id"])

    # Find this OAuth token in the database, or create it
    query = OAuth.query.filter_by(
        provider=blueprint.name,
        provider_user_id=user_id,
    )
    try:
        oauth = query.one()
    except NoResultFound:
        oauth = OAuth(
            provider=blueprint.name,
            provider_user_id=user_id,
            token=token,
        )

    if oauth.user:
        login_user(oauth.user)
        flask.flash("Successfully signed in.")

    else:
        # Create a new local user account for this user
        name = info['login']

        user = User(
            email=info["email"],
            name=name,
            provider=provider
        )
        # Associate the new local user account with the OAuth token
        oauth.user = user
        # Save and commit our database models
        db.session.add_all([user, oauth])
        db.session.commit()
        # Log in the new local user account
        login_user(user)
        flask.flash("Successfully signed in.")

    # Disable Flask-Dance's default behavior for saving the OAuth token
    return False

Answer 1

我的回答可能有点晚了，因为您提到您的公司决定只允许用户在您的域上注册。不过，我仍然为可能有此问题的其他用户发布此内容。

要解决此问题，您需要有一个基本登录页面，该页面首先存储用户通过域登录或 OAuth 登录完成登录后应重定向到的 next_url。当您在 URL 上使用 @login_required 时，Flask 会在 URL 中添加 next 参数，该参数只允许登录用户使用。您需要在登录路由中添加 session['next_url'] = request.args.get('next') 来捕获 next 参数并将其存储在 session 中作为next_url。现在，在 OAuth 登录的情况下，一旦用户成功登录后重定向到默认的 redirect_url，您将从 session 中读取 next_url 值并将用户重定向到该 URL(如果存在)或呈现默认页面(通常是主页)。

我假设两件事，1. OAuth 登录成功后您的默认 URL 是 /home。这是使用 make_provider_blueprint 函数的 redirect_url 设置的，在您的情况下它是 make_github_blueprint。其次，我假设您的登录页面是用户未登录时的登录页面，并且显示 GitHub 登录按钮，该按钮启动 OAuth 登录过程。您可能需要根据您的情况稍微修改此流程才能相应地捕获 next_url。

我在下面分享了处理这种情况所需的最少代码。请注意，我在重定向之前从 session 中弹出 next_url 值。这样，如果用户在登录后尝试访问主页，他们不应该总是被重定向到 next_url(如果 session 中存在)。

bp_github = make_github_blueprint(
    client_id="...",
    client_secret="...",
    redirect_url="/home"
)

@app.route("/login", methods=['GET', 'POST'])
def login():
    if current_user.is_authenticated:
        return flask.redirect(flask.url_for('/'))
    session['next_url'] = request.args.get('next')
    return flask.render_template('login.html')

@auth.route("/")
@auth.route("/home")
@login_required
def home():
    if session.get('next_url'):
        next_url = session.get('next_url')
        session.pop('next_url', None)
        return redirect(next_url)
    return render_template('home.html')

请告诉我这是否有帮助或者您需要任何其他信息。我很乐意提供帮助。