Flask用户认证：Flask-Login实现登录功能

在Web应用中，用户认证是确保数据安全和个性化体验的基础。比如，只有登录用户才能访问个人中心、提交订单等功能。Flask-Login是一个轻量级的Flask扩展，专门用于管理用户会话，简化登录、登出和权限控制的实现。本文将用最简单的方式，带你一步步实现基于Flask-Login的用户登录功能。

在没有Flask-Login时，我们需要手动处理用户会话（比如用session存储用户ID）、验证密码、管理登录状态等。而Flask-Login帮我们封装了这些逻辑，只需几行代码就能实现：
- 自动维护用户会话（保存用户ID）
- 提供@login_required装饰器保护路由
- 处理用户登录、登出的核心逻辑

二、准备工作¶

首先安装必要的库：

pip install flask flask-login

Flask-Login的核心是LoginManager，它会管理用户会话和登录状态。此外，我们需要一个“用户模型”来存储用户信息（这里用简单的类模拟）。

三、实现步骤（附代码）¶

from flask import Flask, render_template, redirect, url_for, request, flash
from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user, current_user

# 初始化Flask应用
app = Flask(__name__)
app.secret_key = 'your-secret-key-here'  # 必须设置，用于加密会话数据

# 初始化Flask-Login的LoginManager
login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'login'  # 未登录时跳转的路由（比如/login）

2. 创建用户模型¶

Flask-Login要求用户类继承UserMixin，它会自动实现is_authenticated、is_active等属性（无需手动写这些方法）。我们用一个简单的User类模拟用户数据：

class User(UserMixin):
    def __init__(self, user_id, username, password):
        self.id = user_id  # 用户ID（必须）
        self.username = username  # 用户名
        self.password = password  # 密码（实际项目中必须加密存储！）

# 模拟用户数据库（实际项目用SQLite/MySQL等）
users_db = {
    1: User(id=1, username='admin', password='123456'),
    2: User(id=2, username='guest', password='654321')
}

3. 实现用户加载函数（关键！）¶

Flask-Login需要知道如何从会话中加载用户。通过user_loader回调函数，根据用户ID从数据库中找到对应的用户：

@login_manager.user_loader
def load_user(user_id):
    """根据用户ID加载用户对象"""
    return users_db.get(int(user_id))  # 从模拟数据库中查询用户

4. 实现登录功能¶

创建登录视图函数，处理用户输入的用户名和密码，验证成功后调用login_user(user)记录会话：

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        # 获取用户输入
        username = request.form.get('username')
        password = request.form.get('password')

        # 模拟用户验证（实际项目需查询数据库+加密比对）
        user = None
        for u in users_db.values():
            if u.username == username and u.password == password:
                user = u
                break

        if user:
            # 登录成功：调用login_user记录会话
            login_user(user)  # 会将用户ID存入session
            return redirect(url_for('home'))  # 重定向到首页
        else:
            flash('用户名或密码错误！')  # 显示错误提示

    # GET请求时渲染登录表单
    return render_template('login.html')

5. 实现登出功能¶

登出时调用logout_user清除会话：

@app.route('/logout')
@login_required  # 只有登录用户才能访问登出路由
def logout():
    logout_user()  # 清除会话，用户登出
    return redirect(url_for('home'))  # 重定向到首页

6. 保护路由（验证登录状态）¶

用@login_required装饰器标记需要登录才能访问的路由：

@app.route('/profile')
@login_required  # 未登录用户会被重定向到login_view
def profile():
    """个人中心：只有登录用户可见"""
    return f"欢迎回来，{current_user.username}！"

7. 首页和模板¶

首页显示欢迎信息和登录状态：

@app.route('/')
def home():
    return render_template('home.html')

模板文件（templates/home.html）：

<!DOCTYPE html>
<html>
<head>
    <title>Flask-Login示例</title>
</head>
<body>
    <h1>首页</h1>
    {% if current_user.is_authenticated %}
        <p>当前用户：{{ current_user.username }}</p>
        <a href="{{ url_for('profile') }}">进入个人中心</a>
        <br>
        <a href="{{ url_for('logout') }}">登出</a>
    {% else %}
        <p>您尚未登录，请<a href="{{ url_for('login') }}">登录</a></p>
    {% endif %}
</body>
</html>

登录模板（templates/login.html）：

<!DOCTYPE html>
<html>
<head>
    <title>登录</title>
</head>
<body>
    <h1>登录</h1>
    {% with messages = get_flashed_messages() %}
        {% if messages %}
            <p style="color: red;">{{ messages[0] }}</p>
        {% endif %}
    {% endwith %}
    <form method="POST">
        <input type="text" name="username" placeholder="用户名" required><br><br>
        <input type="password" name="password" placeholder="密码" required><br><br>
        <button type="submit">登录</button>
    </form>
</body>
</html>

四、运行测试¶

保存上述代码到app.py，并在同目录下创建templates文件夹，放入home.html和login.html。
运行应用：python app.py
访问 http://127.0.0.1:5000，点击“登录”，输入用户名admin/guest和密码123456/654321，即可进入个人中心。

五、注意事项¶

密码安全：示例中密码是明文存储，实际项目必须用bcrypt或werkzeug.security加密（例如generate_password_hash和check_password_hash）。
会话安全：生产环境中app.secret_key需设置为复杂随机字符串，避免硬编码。
扩展功能：可添加“记住我”功能（login_user(user, remember=True)）、密码重置、权限管理等，Flask-Login支持这些高级特性。

总结¶

Flask-Login通过简洁的API，快速实现了用户认证的核心功能：登录、登出、会话管理和路由保护。本文用最简单的示例带你入门，后续可结合数据库和前端框架（如Bootstrap）进一步完善。

（完整代码见上述步骤，可复制到本地运行测试）