Flask表单验证-wtforms

安装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
    	# 校验通过