下面是一个示例代码,演示如何编写一个用于WTForms表单的验证器,该验证器比较两个字段的输入。
首先,我们将创建一个自定义验证器类,该类继承自wtforms.validators.Validator
类。在该类中,我们需要实现一个__call__
方法,该方法将在表单验证时被调用。
from wtforms import ValidationError
from wtforms.validators import Validator
class CompareFields(Validator):
def __init__(self, fieldname, message=None):
self.fieldname = fieldname
self.message = message
def __call__(self, form, field):
try:
other_field = form[self.fieldname]
if field.data != other_field.data:
if self.message is None:
self.message = f"{field.label.text}和{other_field.label.text}的值不匹配"
raise ValidationError(self.message)
except KeyError:
raise ValidationError(f"表单中不存在字段名为'{self.fieldname}'的字段")
在这个验证器类中,我们通过传递一个fieldname
参数来指定要比较的字段名,以及一个可选的message
参数来自定义验证失败时的错误消息。
然后,我们可以在表单类中使用这个验证器。以下是一个示例表单类:
from wtforms import Form, StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, EqualTo
class RegistrationForm(Form):
username = StringField('用户名', validators=[DataRequired()])
password = PasswordField('密码', validators=[DataRequired()])
confirm_password = PasswordField('确认密码', validators=[
DataRequired(),
EqualTo('password', message='确认密码必须与密码相同'),
CompareFields('password')
])
submit = SubmitField('注册')
在上述示例中,我们在confirm_password
字段上使用了EqualTo
验证器来确保确认密码与密码字段相同,还使用了我们自定义的CompareFields
验证器来比较两个密码字段的输入。
这样,在表单验证时,如果两个密码字段的值不匹配,将会出现一个验证错误消息。