15、Flask实战第15天：Flask-WTF

WTForms表单验证基本使用

Flask-WTF是简化了WTForms操作的一个第三方库。WTForm表单的两个主要功能是验证用户提交数据的合法性以及渲染模板。当然还包括一些其他的功能：CSRF保护，文件上传等。安装Flask-WTF默认也会安装WTForms，因此使用以下命令来安装Flask-WTF:

pip install flask-wtf

因为flask-wtf是对wtforms的一个封装，其原理还是使用wtforms，所以，我们先来学习wtforms怎么使用。

表单验证：我们会遇到如下场景：

　　注册的时候判断用户输入的信息是否合法，下面我们做个简单的demon

    <form action="" method="post">
        <table>
            <tr>
                <td>用户名：</td>
                <td><input type="text" name="username" /></td>
            </tr>
            <tr>
                <td>密码：</td>
                <td><input type="password" name="password_1" /></td>
            </tr>
            <tr>
                <td>重复密码：</td>
                <td><input type="password" name="password_2" /></td>
            </tr>
            <tr>
                <td><input type="submit" value="立即注册" /></td>
                <td>
                    {% if error %}
                        <p style="color: red">{{ error }}</p>
                    {% endif %}
                </td>
            </tr>
        </table>
    </form>

register

...
from flask import request, render_template

@app.route('/register/', methods=['GET', 'POST'])  #表明该url接受GET和POST两种方法
def register():
    if request.method == 'GET':
        return render_template('register.html')
    else:
        username = request.form.get('username')
        password_1 = request.form.get('password_1')
        password_2 = request.form.get('password_2')

        if len(username) < 5 or len(username) > 10:
            return render_template('register.html', error="用户名长度不正确")
        if len(password_1) < 5 or len(password_1) > 10:
            return render_template('register.html', error="密码长度不正确")
        if password_1 != password_2:
            return render_template('register.html', error="密码不一致")
        return '注册成功'

主程序my_flask.py

我们使用上面的代码去取验证表单，虽然可以实现需求，但是会显得非常冗长，并且也不能复用。当项目大的时候就尤其明显，不好管理。下面我们就使用WTForms来把它重构

现在我们可以启动Debug断点调试，注册页面故意输入不符合要求的信息提交

断点看到，form验证没有通过，并且在error中报出了错误的原因，password_1、password_2不符合长度要求，并且密码不一致

错误信息，我们可以自己定制

我们再次断点调试，可以看到错误信息已经变成我们自己定制的了

WTForm常用验证器

数据发送过来，经过表单验证，因此需要验证器来进行验证，如下是一些常用的内置验证器

• Email：验证上传的数据是否是邮箱
• EqualTo：验证上传的数据是否和另外一个字段相等，常用的就是密码和确认密码是否相等
• InputRequired：验证上传的数据是否为空，比如验证必填项
• Length：长度限制，有min和max两个值进行限制
• NumberRange： 数字的区间，有min和max两个值限制，如果处在和两个数据之间则满足
• Regexp：自定义正则表达式
• URL：必须是URL的形式
• UUID：验证UUID

我们可以创建一个forms.py文件专门用来处理表单验证，这样比较好管理，编辑forms.py

from wtforms import Form, StringField, IntegerField
from wtforms.validators import Email, Length, EqualTo, NumberRange, Regexp, URL, UUID

class RegisterForm(Form):
    email = StringField(validators=[Email()])
    password_1 = StringField(validators=[Length(5,10)])
    password_2 = StringField(validators=[EqualTo('password_1')])
    age = IntegerField(validators=[NumberRange(min=18, max=110)])
    phone = StringField(validators=[Regexp(r'1[3458]\d{9}')])
    blog = StringField(validators=[URL()])
    uuid = StringField(validators=[UUID()])

自定义验证器

DEBUG调试