当前位置：首页 > article >正文

如何在 Flask 中实现用户认证？

article 2025/1/23 20:07:35

在 Flask 中实现用户认证，可以通过以下方式完成：

基础步骤

设置用户数据库：存储用户信息（如用户名、密码）。
注册功能：允许用户创建账号。
登录功能：验证用户输入的凭据。
会话管理：使用 Flask 的 session 或第三方工具管理登录状态。
登出功能：清除用户的登录状态。

实现步骤

以下是一个完整示例，展示如何实现用户认证功能：

1. 项目文件结构

flask_auth/
├── app.py         # 主应用
├── auth/          # 用户认证模块
│   ├── __init__.py
│   ├── routes.py
├── templates/     # HTML 模板文件
│   ├── login.html
│   ├── register.html
│   ├── dashboard.html

2. 安装必要库

安装 Flask 和 Flask-WTF（表单处理）：

pip install flask flask-wtf werkzeug

3. 配置应用（app.py）

app.py:

from flask import Flask
from auth.routes import auth_blueprint

app = Flask(__name__)

# 配置密钥，用于会话管理
app.secret_key = 'your_secret_key'

# 注册蓝图
app.register_blueprint(auth_blueprint)

if __name__ == '__main__':
    app.run(debug=True)

4. 用户认证路由（auth/routes.py）

auth/routes.py:

from flask import Blueprint, render_template, request, redirect, url_for, session, flash
from werkzeug.security import generate_password_hash, check_password_hash

# 模拟数据库（在生产环境中使用真正的数据库）
users = {}

auth_blueprint = Blueprint('auth', __name__)

# 注册功能
@auth_blueprint.route('/register', methods=['GET', 'POST'])
def register():
    if request.method == 'POST':
        username = request.form['username']
        password = request.form['password']

        if username in users:
            flash('Username already exists!')
            return redirect(url_for('auth.register'))

        # 存储哈希后的密码
        users[username] = generate_password_hash(password)
        flash('Registration successful! You can now log in.')
        return redirect(url_for('auth.login'))

    return render_template('register.html')

# 登录功能
@auth_blueprint.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        username = request.form['username']
        password = request.form['password']

        # 检查用户是否存在
        if username not in users or not check_password_hash(users[username], password):
            flash('Invalid username or password!')
            return redirect(url_for('auth.login'))

        # 登录成功，设置会话
        session['user'] = username
        flash('Login successful!')
        return redirect(url_for('auth.dashboard'))

    return render_template('login.html')

# 用户主页
@auth_blueprint.route('/dashboard')
def dashboard():
    if 'user' not in session:
        flash('Please log in first!')
        return redirect(url_for('auth.login'))

    return render_template('dashboard.html', username=session['user'])

# 登出功能
@auth_blueprint.route('/logout')
def logout():
    session.pop('user', None)
    flash('You have been logged out.')
    return redirect(url_for('auth.login'))

5. HTML 模板文件

templates/register.html:

<!DOCTYPE html>
<html>
<head><title>Register</title></head>
<body>
    <h1>Register</h1>
    <form method="post">
        <label>Username:</label>
        <input type="text" name="username" required>
        <label>Password:</label>
        <input type="password" name="password" required>
        <button type="submit">Register</button>
    </form>
    <a href="{{ url_for('auth.login') }}">Already have an account? Login</a>
</body>
</html>

templates/login.html:

<!DOCTYPE html>
<html>
<head><title>Login</title></head>
<body>
    <h1>Login</h1>
    <form method="post">
        <label>Username:</label>
        <input type="text" name="username" required>
        <label>Password:</label>
        <input type="password" name="password" required>
        <button type="submit">Login</button>
    </form>
    <a href="{{ url_for('auth.register') }}">Don't have an account? Register</a>
</body>
</html>

templates/dashboard.html:

<!DOCTYPE html>
<html>
<head><title>Dashboard</title></head>
<body>
    <h1>Welcome, {{ username }}</h1>
    <a href="{{ url_for('auth.logout') }}">Logout</a>
</body>
</html>

功能说明

注册功能：
- 用户输入用户名和密码。
- 密码会使用 werkzeug.security.generate_password_hash 进行加密存储。
登录功能：
- 用户输入用户名和密码。
- 使用 check_password_hash 检查密码是否正确。
- 登录成功后，将用户信息存入 session。
会话管理：
- 使用 Flask 的 session 来存储登录状态。
- 用户的会话在浏览器关闭或登出后清除。
登出功能：
- 清除用户的 session 信息。