Flask使用Flask在WTForms中自定义验证器

在本文中，我们将介绍如何在Flask中使用Flask在WTForms中自定义验证器。WTForms是一个流行的Python表单验证库，而Flask是一个轻量级的Web应用程序框架。结合使用这两个工具，可以方便地实现表单的输入验证和数据处理。

阅读更多：Flask 教程

使用WTForms进行表单验证

在开始介绍Flask中自定义验证器之前，首先我们需要了解如何使用WTForms进行表单验证。

WTForms通过定义表单类的方式来实现表单验证。表单类继承自wtforms.Form，并使用wtforms的各种字段类来定义表单字段。验证规则通过字段类的参数来设置。

以下是一个简单的使用WTForms进行表单验证的示例：

from flask import Flask, render_template, request
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, Length

app = Flask(__name__)
app.config["SECRET_KEY"] = "secretkey"

class LoginForm(FlaskForm):
    username = StringField("Username", validators=[DataRequired(), Length(min=4, max=20)])
    password = PasswordField("Password", validators=[DataRequired()])
    submit = SubmitField("Log In")

@app.route("/login", methods=["GET", "POST"])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        # 处理登录逻辑
        return "Login successful"
    return render_template("login.html", form=form)

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

在上述代码中，我们定义了一个名为LoginForm的表单类，其中有一个username字段和一个password字段。这两个字段的验证规则分别为DataRequired()和Length()。

DataRequired()用于验证字段是否为空，而Length()用于验证字段的长度是否符合要求。

在login视图函数中，我们实例化了表单类，并在调用validate_on_submit()方法时进行表单数据的验证。如果验证通过，则继续处理登录逻辑。

在Flask中自定义验证器

WTForms为我们提供了一些内置的验证器，但有时我们需要自定义一些验证规则。在Flask中，可以使用Flask提供的@app.route装饰器的before_request方法来实现自定义验证器。

以下是一个使用Flask在WTForms中自定义验证器的示例，我们将实现一个自定义的验证器is_unique_username，用于验证用户名是否唯一。

from flask import Flask, render_template, request, flash
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, Length
from flask_sqlalchemy import SQLAlchemy

app = Flask(__name__)
app.config["SECRET_KEY"] = "secretkey"
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(20), unique=True, nullable=False)
    password = db.Column(db.String(128), nullable=False)

class UniqueValidator:
    def __init__(self, model, column):
        self.model = model
        self.column = column

    def __call__(self, form, field):
        query = self.model.query.filter(self.column == field.data)
        if query.first():
            raise ValidationError("The entered value already exists")

class LoginForm(FlaskForm):
    username = StringField("Username", validators=[DataRequired(), Length(min=4, max=20), UniqueValidator(User, User.username)])
    password = PasswordField("Password", validators=[DataRequired()])
    submit = SubmitField("Log In")

@app.route("/login", methods=["GET", "POST"])
def login():
    form = LoginForm()
    if form.validate_on_submit():
        # 处理登录逻辑
        return "Login successful"
    return render_template("login.html", form=form)

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

在上述代码中，我们定义了一个名为UniqueValidator的自定义验证器类。该类接受两个参数，model表示数据模型，column表示数据库表中的列名。在验证过程中，我们通过查询数据库来判断字段的值是否已存在。

在LoginForm表单类中的username字段中使用了我们自定义的验证器UniqueValidator。

使用自定义验证器可以轻松地实现更复杂的验证规则，扩展了WTForms的功能。