Flask wtforms如何检测用户名已存在并闪现对应错误提示

方案一：表单类添加自定义验证器（推荐） #

这是最贴合WTForms设计逻辑的方案，无需修改现有路由和HTML错误展示代码，校验逻辑自动生效。

1. 首先在表单类所在文件导入验证异常类：
   from wtforms.validators import ValidationError
2. 在RegisterForm类中添加字段校验方法：

class RegisterForm(FlaskForm):
    # 你原有字段的定义保持不变，比如userName、email、password等
    userName = StringField('用户名', validators=[DataRequired()])
    email = StringField('邮箱', validators=[DataRequired(), Email()])
    password = PasswordField('密码', validators=[DataRequired()])

    # 新增用户名重复校验方法，命名格式必须为 validate_对应字段名
    def validate_userName(self, field):
        if User.query.filter_by(username=field.data).first():
            raise ValidationError('该用户名已被注册，请更换')
    
    # 可同步添加邮箱重复校验
    def validate_email(self, field):
        if User.query.filter_by(email=field.data).first():
            raise ValidationError('该邮箱已被绑定，请更换')

添加完成后，表单执行validate_on_submit()时会自动运行上述校验方法，错误信息会直接存入form.errors，你现有HTML的错误展示代码可以直接渲染提示。

方案二：路由内添加校验逻辑 #

如果不想修改表单类，可直接在注册路由的提交逻辑中提前做数据查询校验：
修改后的路由代码如下：

@app.route("/signup", methods=['GET', 'POST'])
def signup():
    form = RegisterForm()

    if form.validate_on_submit():
        # 校验用户名是否已存在
        if User.query.filter_by(username=form.userName.data).first():
            flash('用户名已被占用，请更换后重试')
            return render_template("signup.html", form=form)
        # 校验邮箱是否已存在
        if User.query.filter_by(email=form.email.data).first():
            flash('邮箱已被注册，请更换后重试')
            return render_template("signup.html", form=form)
        
        hashed_password = generate_password_hash(form.password.data, method='sha256')
        new_user = User(username=form.userName.data, email=form.email.data, password=hashed_password)
        db.session.add(new_user)
        # 兜底捕获数据库唯一约束异常，避免极端并发下程序报错
        try:
            db.session.commit()
            flash("注册成功，请登录")
            return redirect(url_for('login'))
        except:
            db.session.rollback()
            flash("注册失败，用户名或邮箱已被占用")
            return render_template("signup.html", form=form)

    return render_template("signup.html", form=form)

使用该方案需要在HTML模板中添加flash消息的展示模块，可放在表单上方：

{% with messages = get_flashed_messages() %}
  {% if messages %}
    {% for msg in messages %}
      <div class="alert alert-warning">{{ msg }}</div>
    {% endfor %}
  {% endif %}
{% endwith %}

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