如何在 Flask 中实现用户登录

在 Flask 中实现用户登录功能通常涉及以下几个步骤：设置 Flask 应用、创建用户模型、处理用户注册、实现登录逻辑以及保护受限路由。下面就是我总结得一些经验，可以一起聊一聊。

1、问题背景

在使用 Flask 框架构建 Web 应用程序时，通常需要实现用户登录功能。常见的需求是将用户名和密码与数据库中的数据进行比较，并根据比较结果进行相应的操作。例如，如果用户名不存在，则提示"用户不存在"；如果密码不匹配，则提示"密码错误"；如果登录成功，则提示"您已登录"。

2、解决方案

为了实现上述需求，可以采用以下步骤：

使用 Flask-SQLAlchemy 与数据库进行连接。
使用 bcrypt 模块对密码进行哈希处理。
定义用户模型并将其映射到数据库表。
在登录视图函数中，从数据库中查询用户名和密码，并与用户输入的用户名和密码进行比较。
根据比较结果，使用 Flask 的 flash() 函数提示相应的错误信息或成功信息。
如果登录成功，则将用户 ID 存储在会话中。

以下是一个示例代码：

复制代码

from flask import Flask, request, redirect, url_for, flash
from flask_sqlalchemy import SQLAlchemy
import bcrypt

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String, unique=True)
    password = db.Column(db.String)

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        user = User.query.filter_by(username=request.form['username']).first()
        if not user:
            flash('Invalid username')
        elif bcrypt.hashpw(request.form['password'], user.password) != hashed:
            flash('Invalid password')
        else:
            session['user_id'] = user.id
            flash('You were logged in')
            return redirect(url_for('show_entries'))

    return render_template('login.html')

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

在上面的代码中，我们需要先创建一个数据库表来存储用户的用户名和密码，可以使用以下命令来创建表：

复制代码

flask db init
flask db migrate
flask db upgrade

然后，我们需要在应用程序中定义一个用户模型，并将其映射到数据库表。可以在 models.py 文件中定义用户模型：

复制代码

from flask_sqlalchemy import SQLAlchemy

db = SQLAlchemy()

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String, unique=True)
    password = db.Column(db.String)

最后，我们需要在视图函数中实现登录逻辑。可以在 views.py 文件中定义视图函数：

复制代码

from flask import Flask, request, redirect, url_for, flash
from flask_sqlalchemy import SQLAlchemy
import bcrypt

app = Flask(__name__)
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///database.db'
db = SQLAlchemy(app)

class User(db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String, unique=True)
    password = db.Column(db.String)

@app.route('/login', methods=['GET', 'POST'])
def login():
    if request.method == 'POST':
        user = User.query.filter_by(username=request.form['username']).first()
        if not user:
            flash('Invalid username')
        elif bcrypt.hashpw(request.form['password'], user.password) != hashed:
            flash('Invalid password')
        else:
            session['user_id'] = user.id
            flash('You were logged in')
            return redirect(url_for('show_entries'))

    return render_template('login.html')

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

这样，我们就实现了使用 Flask-SQLAlchemy 和 bcrypt 模块来实现用户登录的功能。

通过以上步骤，我们可以在 Flask 应用中实现一个简单的用户登录系统。这个示例展示了如何使用 Flask-Login 来管理用户会话，处理登录、注销，并保护受限路由。