Flask校验

WTForms 是一个 Python 库,用于处理和验证 Web 表单。它提供了很多功能来简化表单处理,包括字段类型、验证器、错误消息等。在 WTForms 中,validate 机制是用于确保表单数据满足特定条件的关键部分。

1.验证器(Validators)

WTForms 提供了多种内置的验证器,如 DataRequired, Length, Email, EqualTo 等,这些可以被添加到表单字段上以执行特定的验证规则。例如:

python 复制代码
from wtforms import StringField, validators

class LoginForm(Form):
    username = StringField('Username', [validators.DataRequired()])
    password = PasswordField('Password', [
        validators.DataRequired(),
        validators.Length(min=8)
    ])

在这个例子中,DataRequired 确保字段不为空,Length 则确保密码长度至少为 8。

2.自定义验证器

你也可以定义自己的验证函数,并将其作为验证器使用。这个函数应该接收表单实例和字段实例作为参数,并在验证失败时抛出 ValidationError 异常。例如:

python 复制代码
from wtforms import StringField, ValidationError
from wtforms.validators import StopValidation

def unique_username(form, field):
    if User.query.filter_by(username=field.data).first():
        raise ValidationError('Username already exists.')

class RegistrationForm(Form):
    username = StringField('Username', [unique_username])

在这个例子中,unique_username 函数检查数据库中是否已存在相同的用户名,如果存在,则抛出 ValidationError

3.全局验证

除了字段级别的验证之外,你还可以定义全局验证方法,在表单类中定义以 validate_ 开头的方法,后面跟着字段名,这会触发对整个表单的验证。这种方法通常用于需要跨多个字段的复杂验证逻辑。例如:

python 复制代码
class RegistrationForm(Form):
    username = StringField('Username')
    password = PasswordField('Password')
    confirm_password = PasswordField('Confirm Password')

    def validate_confirm_password(self, field):
        if field.data != self.password.data:
            raise ValidationError('Passwords do not match.')

这里,validate_confirm_password 方法确保用户输入的密码与其确认密码相同。

4. 使用多个验证器

你可以将多个验证器应用于同一个字段,例如,确保电子邮件格式正确并且非空:

python 复制代码
from wtforms import StringField, EmailField
from wtforms.validators import DataRequired, Email

class ContactForm(Form):
    email = EmailField('Email', validators=[DataRequired(), Email()])

5. 自定义验证器示例 - 检查年龄

假设你想要验证用户年龄必须大于18岁:

python 复制代码
from datetime import date

def is_adult(form, field):
    today = date.today()
    age = today.year - field.data.year - ((today.month, today.day) < (field.data.month, field.data.day))
    if age < 18:
        raise ValidationError('You must be at least 18 years old.')

class RegistrationForm(Form):
    birth_date = DateField('Birth Date', validators=[is_adult])

6. 验证唯一性 - 检查用户名或邮箱是否已被注册

在数据库中检查某个值是否唯一:

python 复制代码
from myapp.models import User

def unique_username(form, field):
    if User.query.filter_by(username=field.data).first() is not None:
        raise ValidationError('Username already exists.')

def unique_email(form, field):
    if User.query.filter_by(email=field.data).first() is not None:
        raise ValidationError('Email already exists.')

class RegistrationForm(Form):
    username = StringField('Username', validators=[unique_username])
    email = StringField('Email', validators=[unique_email])

7. 自定义错误消息

你可以覆盖默认的错误消息,使其更加友好或具体:

python 复制代码
class ContactForm(Form):
    subject = StringField('Subject', validators=[DataRequired(message='Please enter a subject.')])
    message = TextAreaField('Message', validators=[Length(max=200, message='Your message is too long.')])

下面是一个使用 验证器的用户注册完整例子:

python 复制代码
from wtforms import Form, StringField, PasswordField, validators

class RegistrationForm(Form):
    username = StringField('Username', [validators.Length(min=4, max=25)])
    email = StringField('Email Address', [validators.Length(min=6, max=35),
                                          validators.Email()])
    password = PasswordField('New Password', [
        validators.DataRequired(),
        validators.EqualTo('confirm', message='Passwords must match')
    ])
    confirm = PasswordField('Repeat Password')

# 使用表单
form = RegistrationForm(request.form)

if request.method == 'POST' and form.validate():
    # 如果表单验证成功,可以进一步处理数据,比如保存到数据库
    user = User(username=form.username.data, email=form.email.data,
                password=generate_password_hash(form.password.data))
    db.session.add(user)
    db.session.commit()
else:
    # 如果表单验证失败,显示错误消息
    flash_errors(form)

在这个例子中,RegistrationForm 包含了 username, email, password, 和 confirm 字段。password 字段使用了 EqualTo 验证器,它会检查 passwordconfirm 字段的值是否相同。如果这两个字段的值不同,EqualTo 验证器会抛出一个 ValidationError,并附带错误消息 "Passwords must match"。

EqualTo 验证器的第一个参数是要与当前字段比较的另一个字段的名称。在上面的例子中,password 字段通过 EqualTo('confirm')confirm 字段进行比较。

如果在实际应用中遇到问题,或者需要进一步定制验证逻辑,你也可以自定义验证函数来配合 EqualTo 或其他验证器一起使用。

相关推荐
Hsu_kk1 小时前
MySQL 批量删除海量数据的几种方法
数据库·mysql
编程学无止境1 小时前
第02章 MySQL环境搭建
数据库·mysql
knight-n1 小时前
MYSQL库的操作
数据库·mysql
丕羽2 小时前
【Pytorch】基本语法
人工智能·pytorch·python
包饭厅咸鱼2 小时前
QML----复制指定下标的ListModel数据
开发语言·数据库
bryant_meng2 小时前
【python】Distribution
开发语言·python·分布函数·常用分布
生命几十年3万天2 小时前
redis时间优化
数据库·redis·缓存
Elastic 中国社区官方博客2 小时前
释放专利力量:Patently 如何利用向量搜索和 NLP 简化协作
大数据·数据库·人工智能·elasticsearch·搜索引擎·自然语言处理
力姆泰克2 小时前
看电动缸是如何提高农机的自动化水平
大数据·运维·服务器·数据库·人工智能·自动化·1024程序员节
力姆泰克2 小时前
力姆泰克电动缸助力农业机械装备,提高农机的自动化水平
大数据·服务器·数据库·人工智能·1024程序员节