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

在 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_blueprintapp = 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'] = usernameflash('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 信息。