使用 Flask 和 Flask-Login 构建用户认证的 Web 应用程序

在本篇技术博客中，我们将学习如何使用 Flask 框架和 Flask-Login 扩展构建一个具有用户认证功能的简单 Web 应用程序。我们将从创建 Flask 应用实例开始，然后逐步添加用户认证功能。

1. 安装依赖库

首先，确保您已经安装了 Flask、Flask-PyMongo 和 Flask-Login 库。您可以使用以下命令安装这些库：

pip install Flask Flask-PyMongo Flask-Login

2. 创建 Flask 应用实例

让我们开始创建一个简单的 Flask 应用：

from flask import Flask

app = Flask(__name__)

在这里，我们导入了 Flask 并创建了一个 Flask 应用实例。__name__ 是当前 Python 文件的名称。

3. 初始化 PyMongo 和 LoginManager

接下来，我们需要初始化 PyMongo 和 LoginManager：

from flask_pymongo import PyMongo
from flask_login import LoginManager

mongo = PyMongo(app)
login_manager = LoginManager()
login_manager.init_app(app)

我们创建了一个 PyMongo 实例，用于连接 MongoDB 数据库。然后，我们创建了一个 LoginManager 实例，用于处理用户认证。init_app 方法用于初始化 LoginManager。

4. 定义 User 类

现在，我们需要定义一个 User 类，继承自 Flask-Login 提供的 UserMixin 混入类：

from flask_login import UserMixin

class User(UserMixin):
    pass

UserMixin 包含了用户认证所需的基本方法，如 is_authenticated、is_active 等。

5. 注册用户加载回调函数

为了让 Flask-Login 知道如何从数据库中加载用户对象，我们需要注册一个回调函数：

@login_manager.user_loader
def load_user(user_id):
    return mongo.db.users.find_one({'_id': user_id})

这个回调函数接受一个参数 user_id，并从 MongoDB 数据库中查找相应的用户对象。

6. 定义路由和视图函数

现在我们可以定义一些路由和视图函数，例如一个受保护的路由，需要用户登录才能访问：

from flask import render_template, request, redirect, url_for
from flask_login import login_user, current_user, login_required, logout_user

@app.route('/login')
def login():
    # 这里可以添加表单验证和用户登录逻辑
    pass

@app.route('/protected')
@login_required
def protected_page():
    if current_user.is_active:
        return 'Logged in as: ' + current_user.id + 'Login is_active:True'

我们定义了一个名为 protected_page 的视图函数，它对应于 /protected路由。@login_required 装饰器表示该路由需要用户登录。当用户访问这个路由时，如果用户已登录（current_user.is_active 为 True），将返回一个字符串，显示用户的 ID 和登录状态。

7. 运行 Flask 应用

最后，我们需要运行 Flask 应用：

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

这行代码表示，如果当前脚本是主程序（而不是被导入），则运行 Flask 应用。debug=True 表示启用调试模式，port=8000 表示应用将运行在 8000 端口。

总结

在本篇技术博客中，我们学习了如何使用 Flask 框架和 Flask-Login 扩展构建一个具有用户认证功能的简单 Web 应用程序。我们从创建 Flask 应用实例开始，然后逐步添加用户认证功能。最后，我们定义了一个受保护的路由，需要用户登录才能访问。