Flask框架中如何实现用户认证

发布时间：2025-02-17 19:20:51 作者：小樊
来源：亿速云阅读：84

在Flask框架中，实现用户认证通常涉及以下几个步骤：

安装必要的库：
- Flask-Login：用于管理用户会话。
- Flask-SQLAlchemy：用于数据库操作。
- Flask-WTF：用于表单处理。
你可以使用pip来安装这些库：
```
pip install flask flask-login flask-sqlalchemy flask-wtf
```

设置Flask应用：创建一个基本的Flask应用，并配置数据库。

from flask import Flask, render_template, redirect, url_for, request
from flask_sqlalchemy import SQLAlchemy
from flask_login import LoginManager, UserMixin, login_user, login_required, logout_user, current_user
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, Email, EqualTo

app = Flask(__name__)
app.config['SECRET_KEY'] = 'your_secret_key'
app.config['SQLALCHEMY_DATABASE_URI'] = 'sqlite:///db.sqlite'
db = SQLAlchemy(app)
login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = 'login'

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    email = db.Column(db.String(120), unique=True, nullable=False)
    password = db.Column(db.String(60), nullable=False)

class LoginForm(FlaskForm):
    email = StringField('Email', validators=[DataRequired(), Email()])
    password = PasswordField('Password', validators=[DataRequired()])
    submit = SubmitField('Login')

@login_manager.user_loader
def load_user(user_id):
    return User.query.get(int(user_id))

@app.route('/login', methods=['GET', 'POST'])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        user = User.query.filter_by(email=form.email.data).first()
        if user and user.password == form.password.data:
            login_user(user)
            return redirect(url_for('protected'))
        else:
            return 'Invalid email or password'
    return render_template('login.html', form=form)

@app.route('/protected')
@login_required
def protected():
    return f'Hello, {current_user.email}!'

@app.route('/logout')
@login_required
def logout():
    logout_user()
    return redirect(url_for('login'))

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

创建数据库模型：在上面的代码中，我们定义了一个User模型，它继承自UserMixin和db.Model。
创建登录表单：使用Flask-WTF创建一个登录表单。
用户加载回调函数：使用@login_manager.user_loader装饰器定义一个回调函数，用于从数据库中加载用户。
登录视图：定义一个登录视图，处理用户登录请求。
受保护的视图：定义一个受保护的视图，只有登录用户才能访问。
登出视图：定义一个登出视图，处理用户登出请求。

模板文件：创建一个login.html模板文件，用于显示登录表单。

<!DOCTYPE html>
<html lang="en">
<head>
    <meta charset="UTF-8">
    <title>Login</title>
</head>
<body>
    <form method="post">
        {{ form.hidden_tag() }}
        <p>
            {{ form.email.label }}<br>
            {{ form.email(size=32) }}<br>
            {% for error in form.email.errors %}
                <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>
            {{ form.password.label }}<br>
            {{ form.password(size=32) }}<br>
            {% for error in form.password.errors %}
                <span style="color: red;">[{{ error }}]</span>
            {% endfor %}
        </p>
        <p>{{ form.submit() }}</p>
    </form>
</body>
</html>

通过以上步骤，你可以在Flask应用中实现基本的用户认证功能。

Flask框架中如何实现用户认证

相关阅读