Python Flask-Login模块使用案例详解

我会为你详细讲解“Python Flask-Login模块使用案例详解”的完整攻略，同时会为你提供两条示例。

标题

介绍

Flask-Login 是一个 Flask 扩展，它提供了用户登录和会话管理的一个方案。通过这个模块，我们可以快速简便地添加认证、保护和会话管理到我们的 Flask 应用程序中。

安装

要使用 Flask-Login 模块，需要先安装它。可以使用 pip 安装，在命令行里输入：

$ pip install flask-login

配置

在 Flask 应用程序中使用 Flask-Login，首先需要配置应用程序以使用登录管理器。配置示例如下：

from flask_login import LoginManager
from flask import Flask

app = Flask(__name__)

login_manager = LoginManager()
login_manager.init_app(app)

使用示例1：基本登录

以下是一个基本的登录示例：

from flask_login import login_user, logout_user
from flask import Flask, request, render_template, redirect, url_for
from werkzeug.security import check_password_hash
from models import User

app = Flask(__name__)

# 配置登录用户名
login_manager.login_view = 'login'
login_manager.login_message_category = 'warning'

@app.route('/login', methods=('GET', 'POST'))
def login():
    if request.method == 'POST':
        email = request.form.get('email')
        password = request.form.get('password')
        user = User.query.filter_by(email=email).first()

        if not user or not check_password_hash(user.password, password):
            flash('Invalid email or password', category='error')
            return redirect(url_for('login'))

        login_user(user)

        return redirect(url_for('home'))

    return render_template('login.html')

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

这个示例中，我们通过 flask_login 的 login_user 函数将已认证的用户标记为已登录。然后通过 redirect 函数将该用户重定向到主页（即 /home）。

当用户想注销登录时，我们调用 logout_user 函数来标记该用户已经退出登录。

使用示例2：自定义 User 模型

使用 Flask-Login 模块时，我们需要定义一个 User 模型。以下示例展示了一个自定义用户模型的实现方法：

from flask_login import UserMixin
from werkzeug.security import generate_password_hash
from . import db

class User(UserMixin, db.Model):
    __tablename__ = 'users'

    id = db.Column(db.Integer, primary_key=True)
    name = db.Column(db.String(255), unique=True, nullable=False)
    email = db.Column(db.String(255), unique=True, nullable=False)
    password_hash = db.Column(db.String(255), nullable=False)

    def __init__(self, name, email, password):
        self.name = name
        self.email = email
        self.set_password(password)

    def set_password(self, password):
        self.password_hash = generate_password_hash(password)

    def check_password(self, password):
        return check_password_hash(self.password_hash, password)

    def __repr__(self):
        return '<User %r>' % self.name

在这个示例中，我们使用 Flask 框架和 SQLAlchemy 定义了一个 User 模型，该模型继承自 flask_login 的 UserMixin 类，并且有一些附加功能，例如 password_hash、set_password 和 check_password 方法。

注意，我们还可以定义其他方法以满足我们的需求。模型定义完成后，我们可以通过数据库的 API 访问用户模型。

至此，“Python Flask-Login模块使用案例详解”的完整攻略包括两个示例都介绍完毕了，希望对你有帮助。