表单验证(WTF)

官方文档
在Flask项目开发中针对提交表单的校验,可以使用Flask-WTF扩展库进行快速的字段校验,也可以进行页面快速渲染,并提供跨站请求伪造的保护功能。

安装Flask-WTF

pip install flask-wtf

实现注册功能

注册表单模型定义

在定义的表单类中定义需要验证的username、password和password2字段,并实现如下校验:
校验密码password2和password相等
校验用户名是否存在
校验用户名的长度是否符合规范

 # 导入扩展类
 from flask_wtf import FlaskForm
 # 导入验证字段
 from wtforms import StringField, SubmitField, ValidationError
 # 导入表单验证
 from wtforms.validators import DataRequired, EqualTo
 
 from user.models import User
 
 
 class UserRegisterForm(FlaskForm):
     """
     登录注册表单验证
     """
     username = StringField('用户名', validators=[DataRequired()])
     password = StringField('密码', validators=[DataRequired()])
     password2 = StringField('确认密码', validators=[DataRequired(),
                                                 EqualTo('password', '密码不一致')]
                             )
     submit = SubmitField('注册')
 
     def validate_username(self, field):
         # 验证用户名是否重复
         if User.query.filter(User.username == field.data).first():
             raise ValidationError('用户名已存在')
 
         # 对用户名的长度进行判断
         if len(field.data) < 3:
             raise ValidationError('用户名长度不能少于3个字符')
 
         if len(field.data) > 6:
             raise ValidationError('用户名长度不能大于6个字符')

注意: 验证字段的方法名为: validate_字段(self, field)

定义注册视图函数
当HTTP请求为GET时,将表单验证对象返回给页面。
当HTTP请求为POST时,通过方法validate_on_submit()方法进行字段校验和提交判断,如果校验失败,则可以从form.errors中获取错误信息。
如果验证通过,则从form.字段.data中获取到字段的值。

@user_blueprint.route('/register/', methods=['GET', 'POST'])
def register():
    form = UserRegisterForm()
    if request.method == 'GET':
        return render_template('register.html',form=form)

    if request.method == 'POST':
        # 验证提交的字段信息
        if form.validate():
            username = form.username.data
            password = form.password.data
            # 实现注册,保存用户信息到User模型中
            user = User()
            user.username = username
            user.password = generate_password_hash(password)
            user.save()
            return redirect(url_for('user.login'))        
        # 验证失败,from.errors中存在错误信息
        return render_template('register.html',form=form)

模板展示

注册模板采用继承父模板base.html的形式。在register.html模压中分析如下:

  1. 定义字段名: {{ form.字段.label }}
  2. 定义input输入框: {{ form.字段 }}
  3. 展示错误信息: {{ form.errors.字段 }}
  4. 跨站请求伪造: {{ form.csrf_token }}
    注册register.html页面如下:
{% extends 'base.html' %}

{% block content %}
    <form action="" method="post">
        {{ form.csrf_token }}
        {{ form.username.label }} : {{ form.username }}
        {% if form.errors %}
            {{ form.errors.password2[0] }}
        {% endif %}<br>
        {{ form.password.label }} : {{ form.password }}<br>
        {{ form.password2.label }} : {{ form.password2 }}
        {% if form.errors %}
            {{ form.errors.password2[0] }}
        {% endif %}<br>
        {{ form.submit() }}
    </form>

{% endblock %}

注意: 通过form.字段解析的input标签中可以自定义样式,如{{ form.字段(class='xxx', style='color:red') }}

常见字段类型

字段类型说明
StringField普通文本字段
PasswordField密码文本字段
SubmitField提交按钮
HiddenField隐藏文本字段
TextAreaField多行文本字段
DateField文本字段,datetime.date格式
DateTimeField文本字段,datetime.datetime格式
IntegerField文本字段,整数类型
FloatField文本字段,小数类型
BooleanField复选框,值为True或False
RadioField单选框
SelectField下拉列表
FileField文件上传字段

验证器

验证器说明
DataRequired确保字段有值(并且if判断为真)
Email邮箱地址
IPAddressIPv4的IP地址
Length规定字符长度
NumberRange输入数值的范围
EqualTo验证两个字段的一致性
URL有效的URL
Regexp正则验证

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值