Flask中使用flask_login.current_user实现用户登录后跳转到上次访问页面

在Flask中使用flask_login.current_user可以方便地实现用户登录后跳转到上次访问页面的功能。flask_login.current_user是Flask-Login插件提供的一个全局变量，可以方便地获取当前登录用户的信息。

下面是一个简单的使用示例，演示了如何在用户登录后跳转到上次访问页面：

首先，需要在Flask应用中安装和配置Flask-Login插件。可以使用pip命令安装插件：

pip install flask-login

然后，在Flask应用中进行插件的初始化配置，如下所示：

from flask import Flask
from flask_login import LoginManager

app = Flask(__name__)

# 配置应用的secret key
app.config['SECRET_KEY'] = 'secret_key'

# 初始化Login Manager
login_manager = LoginManager(app)
login_manager.login_view = 'login'  # 指定登录视图函数

接下来，定义一个简单的登录视图函数login，用于处理用户的登录请求：

from flask import render_template, redirect, request
from flask_login import login_user

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        # 处理用户登录请求
        # 验证用户密码等操作...

        # 假设验证成功，登录用户
        # user = User.query.get(user_id)  # 根据表单中的用户名获取用户对象
        # login_user(user)

        return redirect(request.args.get('next') or '/')

    return render_template('login.html')

在这个登录视图函数中，我们首先判断请求的方法是否为POST，如果是POST方法，则表示用户提交了登录表单，我们可以在这里进行用户的身份验证。如果验证成功，可以使用login_user()函数登录用户。

同时，在登录成功后，我们通过request.args.get('next')获取URL参数中的next参数，这个参数会包含用户之前的访问页面URL。我们可以通过redirect()函数将用户重定向到上次访问页面，如果next参数不存在，则重定向到默认页面'/'。

最后，定义一个视图函数protected_page，用于演示用户登录后跳转到上次访问页面的效果：

from flask import render_template
from flask_login import login_required

@app.route('/protected')
@login_required
def protected_page():
    return render_template('protected_page.html')

这个protected_page视图函数使用了@login_required装饰器，表示这个页面需要用户登录才能访问。

在这个示例中，假设用户在未登录状态下访问了/protected页面，并被重定向到登录页面。用户在登录后，会被重定向到/protected页面，而不是默认页面。

通过以上的示例，我们可以看到，通过使用flask_login.current_user全局变量和redirect()函数，我们可以方便地实现用户登录后跳转到上次访问页面的功能。这对于提升用户体验和网站的友好度非常重要。