如何在Python Flask中处理认证错误并自定义提示信息

核心思路是接管凭证验证逻辑，不依赖Flask-Login默认的错误提示，手动检查用户输入的用户名和密码，通过Flask的flash机制传递自定义错误消息，并重定向回登录页面。

1. 完善User模型的密码验证方法 #

确保你的User模型包含密码验证的逻辑（示例使用bcrypt加密，可替换为你实际使用的加密方式）：

from flask_login import UserMixin
from flask_sqlalchemy import SQLAlchemy
import bcrypt

db = SQLAlchemy()

class User(UserMixin, db.Model):
    id = db.Column(db.Integer, primary_key=True)
    username = db.Column(db.String(100), unique=True, nullable=False)
    password_hash = db.Column(db.String(200), nullable=False)

    def check_password(self, password):
        # 验证密码是否匹配
        return bcrypt.checkpw(password.encode('utf-8'), self.password_hash.encode('utf-8'))

2. 修改登录路由，自定义错误处理 #

在登录路由中手动完成用户查询、密码验证，失败时添加自定义错误消息：

from flask import Flask, render_template, redirect, url_for, request, flash
from flask_login import LoginManager, login_user, login_required

app = Flask(__name__)
app.secret_key = "your-secure-secret-key-here"  # 必须设置，flash消息依赖此密钥
app.config['SQLALCHEMY_DATABASE_URI'] = 'your-db-uri'

login_manager = LoginManager()
login_manager.init_app(app)
login_manager.login_view = '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():
    if request.method == 'POST':
        username = request.form.get('username')
        password = request.form.get('password')
        user = User.query.filter_by(username=username).first()

        # 自定义验证逻辑
        if not user:
            flash('该用户名未注册，请检查输入', 'error')
            return redirect(url_for('login'))
        if not user.check_password(password):
            flash('密码错误，请重新输入', 'error')
            return redirect(url_for('login'))
        
        # 验证通过，登录用户
        login_user(user)
        # 重定向到用户原本要访问的页面，若无则跳转到仪表盘
        next_page = request.args.get('next')
        return redirect(next_page or url_for('dashboard'))
    
    # GET请求渲染登录页面
    return render_template('login.html')

@app.route('/dashboard')
@login_required
def dashboard():
    return render_template('dashboard.html')

3. 在登录模板中显示错误消息 #

修改login.html模板，添加flash消息的渲染逻辑：

<!DOCTYPE html>
<html>
<head>
    <title>用户登录</title>
    <style>
        .error { color: red; margin: 10px 0; }
    </style>
</head>
<body>
    <h2>登录</h2>
    <!-- 渲染自定义错误消息 -->
    {% with messages = get_flashed_messages(with_categories=true) %}
        {% if messages %}
            {% for category, msg in messages %}
                <div class="{{ category }}">{{ msg }}</div>
            {% endfor %}
        {% endif %}
    {% endwith %}
    <form method="POST">
        <div>
            <label>用户名：</label>
            <input type="text" name="username" required>
        </div>
        <div>
            <label>密码：</label>
            <input type="password" name="password" required>
        </div>
        <button type="submit">登录</button>
    </form>
</body>
</html>

关键说明 #

• 必须设置app.secret_key：Flask的flash消息依赖Session机制，需要密钥加密Session数据。
• 细粒度错误提示：可以根据验证失败的原因（用户不存在/密码错误）返回不同的自定义消息，也可以合并为统一提示。
• 禁用默认提示：若之前设置过login_manager.login_message，可以将其设为空字符串login_manager.login_message = ""，避免默认消息干扰。

内容的提问来源于stack exchange，提问作者Bernardo Almeida