Flask Web 应用与 MongoDB 集成:用户登录和退出

发布于:2024-04-17 ⋅ 阅读:(70) ⋅ 点赞:(0)

在本文中,我们将探讨如何使用 Flask Web 框架和 MongoDB 数据库构建一个简单的 Web 应用。我们将使用 Flask-PyMongo 扩展来实现 MongoDB 集成,并使用 Flask-Login 扩展来处理用户认证。

1. 安装所需库

首先,我们需要安装 Flask、Flask-PyMongo 和 Flask-Login 扩展:

pip install Flask Flask-PyMongo Flask-Login

2. 创建 Flask 应用

接下来,我们将创建一个名为 app.py 的 Flask 应用:

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

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

3. 定义 User 类

我们将使用 Flask-Login 来处理用户认证。首先,我们需要定义一个 User 类,该类继承自 Flask-Login 的 UserMixin 类:

class User(UserMixin):
    pass

4. 用户加载回调

接下来,我们需要定义一个用户加载回调函数,该函数将使用 Flask-Login 的 user_loader 装饰器:

@login_manager.user_loader
def user_loader(username):
    user = User()
    user.id = username
    return user

5. 创建登录和注销路由

现在,我们将创建一个登录页面,并使用 Flask-Login 来处理用户登录和注销:

@app.route('/login', methods=['GET', 'POST'])
def login_page():
    if request.method == 'POST':
        username = request.form['username']
        password = request.form['password']
        # 在此处验证用户凭据
        user = User()
        user.id = username
        login_user(user)
        return redirect(url_for('protected_page'))
    return render_template('login.html')

@app.route('/logout')
def logout_page():
    if current_user.is_active:
        logout_user()
        return 'Logged out'
    else:
        return "you aren't login"

6. 创建受保护的页面

最后,我们将创建一个受保护的页面,该页面仅对登录用户可见:

@app.route('/protected_page')
@login_required
def protected_page():
    return 'This is a protected page.'

7. 运行 Flask 应用

现在,我们已经构建了整个 Flask 应用。在命令行中运行以下命令以启动应用:

python app.py

应用将在本地主机的 8000 端口上运行。您可以通过访问 /login、/logout 路径来测试登录、退出功能。