安装wtforms
pipenv install wtforms
安装email_validator
pipenv install email_validator
重写DataRequired
from wtforms.validators import DataRequired as WTFDataRrequired
class DataRequired(WTFDataRrequired):
"""
重写默认的WTF DataRequired,实现自定义message
DataRequired是一个比较特殊的验证器,当这个异常触发后,
后续的验证(指的是同一个validators中的验证器将不会触发。
但是其他验证器,比如Length就不会中断验证链条。
"""
def __call__(self, form, field):
if self.message is None:
field_text = field.label.text
self.message = field_text + '不能为空,请填写' + field_text
super(DataRequired, self).__call__(form, field)
邮件校验
from wtforms import StringField, PasswordField, Form
from wtforms.validators import Length, Email, \
ValidationError
from .base import DataRequired
class EmailForm(Form):
email = StringField('电子邮件', validators=[DataRequired(), Length(1, 64),
Email(message='电子邮箱不符合规范')])
用户注册Form校验
class RegisterForm(EmailForm):
nickname = StringField('昵称', validators=[
DataRequired(), Length(2, 10, message='昵称至少需要两个字符,最多10个字符')])
password = PasswordField('密码', validators=[
DataRequired(), Length(6, 20, message='密码长度6-20之间')])
def validate_email(self, field):
"""
email的自定义校验器
:param field:
:return:
"""
if User.query.filter_by(email=field.data).first():
raise ValidationError('电子邮件已被注册')
def validate_nickname(self, field):
"""
nickname的自定义校验器
:param field:
:return:
"""
if User.query.filter_by(nickname=field.data).first():
raise ValidationError('昵称已存在')
这里使用自定义校验逻辑validate_email和validate_nickname,自定义校验方法validate_xxx
校验form使用
form = RegisterForm(request.form)
if form.validate():
pass
# 校验通过