Flask中使用flask_login.current_user实现用户登录状态的管理

Flask-Login是一个用于管理用户登录状态的Flask扩展，它提供了在用户登录、登出和会话中保存当前用户的功能。其中，current_user是一个全局变量，可以在视图函数和模板中使用，用于获取当前用户的信息。

要使用current_user，首先需要设置一个用户加载函数，该函数用于根据用户ID加载用户对象。例如，假设我们的用户模型为User，并且有一个load_user函数用于加载用户对象：

from flask_login import UserMixin

class User(UserMixin):
    def __init__(self, id):
        self.id = id

    @staticmethod
    def load_user(user_id):
        # 根据用户ID加载用户对象
        return User(user_id)

接下来，在Flask应用中初始化Flask-Login，并设置用户加载函数：

from flask import Flask
from flask_login import LoginManager

app = Flask(__name__)
app.secret_key = 'secret_key'

login_manager = LoginManager()
login_manager.init_app(app)

@login_manager.user_loader
def load_user(user_id):
    return User.load_user(user_id)

在登录视图函数中，可以使用login_user函数将用户对象登录：

from flask import request
from flask_login import login_user

@app.route('/login', methods=['GET', 'POST'])
def login():
    # 根据用户请求获取用户名和密码
    username = request.form.get('username')
    password = request.form.get('password')

    # 验证用户名和密码
    if username == 'admin' and password == 'password':
        # 获取用户对象
        user = User(username)
        
        # 登录用户
        login_user(user)
        return 'Login Success'
    else:
        return 'Login Failed'

在登出视图函数中，可以使用logout_user函数将用户登出：

from flask_login import logout_user

@app.route('/logout')
def logout():
    # 登出当前用户
    logout_user()
    return 'Logout Success'

在需要登录才能访问的视图函数中，可以使用login_required装饰器来限制只有登录用户才能访问：

from flask_login import login_required

@app.route('/profile')
@login_required
def profile():
    # 获取当前登录用户
    user = current_user
    return 'Welcome, {}'.format(user.id)

在模板中，可以直接使用current_user获取当前用户的信息：

{% if current_user.is_authenticated %}
    <p>Welcome, {{ current_user.id }}</p>
{% else %}
    <p>Please login to view your profile.</p>
{% endif %}

通过以上步骤，在Flask应用中就可以使用current_user来管理和使用用户的登录状态。current_user会返回当前请求的用户对象，如果用户未登录，则返回一个匿名用户对象。你可以在视图函数和模板中使用它来获取用户的信息、判断用户是否登录等。

需要注意的是，为了将当前用户保存在会话中，Flask-Login需要一个secret_key来保护会话。所以在初始化Flask应用时，需要设置一个secret_key。

以上是使用current_user实现用户登录状态管理的一个例子。当用户登录成功时，调用login_user函数将用户对象保存在会话中；当用户登出时，调用logout_user函数将会话中的用户对象移除。在需要登录访问的视图函数中，可以使用login_required装饰器来限制只有登录用户才能访问。同时，可以在模板中使用current_user获取当前用户的信息，判断用户是否登录。