Flask-WTF
它是简化了WTForms操作的一个第三方库。WTForms表单的两个主要功能是验证用户提交数据的合法性以及渲染模板。还包括一些其他的功能:CSRF保护,文件上传等。安装Flask-WTF默认也会安装WTForms,因此使用一下命令来安装Flask-WTF。
pip install html" class="superseo">flask-wtf
表单验证
安装完flask-wtf后,第一个功能就是用表单来做数据验证,现在又一个forms.py文件,然后再里面创建一个RegisForm的注册验证表单。
建立一个文件夹,在下面再建立一个templates,里面创建一个regist.htnl。
再文件夹下创建一个表单文件forms.py和主文件demo.py。
主文件:
from flask import Flask, request, render_template
from forms import RegistForm
app = Flask(__name__)
@app.route('/')
def index():
return "首页"
# 接收GET请求
@app.route('/regist/', methods=['GET','POST'])
def regist():
if request.method == 'GET':
return render_template('regist.html')
else:
form = RegistForm(request.form)
if form.validate():
return "success"
else:
# 验证的错误信息
print(form.errors)
return "fail"
# username = request.form.get('username')
# password = request.form.get('password')
# password_repate = request.form.get('password_repate')
# if len(username) <3 or len(username) >10:
# return "用户名长度不正确"
# if password != password_repate:
# return "两次密码不一致"
# if len(password) <3 or len(password) >10:
# return "密码长度不正确"
if __name__ == '__main__':
app.run(debug=True)
forms文件:
from wtforms import Form,StringField, validators
from wtforms.validators import Length, Regexp, EqualTo
class RegistForm(Form):
# 再后面加上message="xxxx" 出现错误就可以显示中文 在小括号里
username = StringField(validators=[Length(min=3, max=10)])
password = StringField(validators=[Length(min=3, max=10)])
password_repate = StringField(validators=[Length(min=3, max=10), EqualTo("password")])
html文件:
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
<form action="" method="post">
<!-- 表单提交到flask的名字-->
用户名: <input type="text" name="username"><br>
密码: <input type="text" name="password"><br>
确认密码: <input type="text" name="password_repate"><br>
<input type="submit" value="注册">
</form>
</body>
</html>
其他验证
在主文件处demo.py文件定义一个登录
from forms import RegistForm, LoginForm
@app.route('/login/', methods=['GET', 'POST'])
def login():
if request.method == 'GET':
return render_template('login.html')
else:
form = LoginForm(request.form)
if form.validate(): # 与forms文件建立联系
return "success"
else:
# 验证的错误信息
print(form.errors)
return "fail"
在templates文件夹下新建一个login文件:
<!DOCTYPE html>
<html lang="en">
<head>
<meta charset="UTF-8">
<title>Title</title>
</head>
<body>
<form action="" method="post">
<table>
<!-- <tr>-->
<!-- <td>邮箱:</td>-->
<!-- <td><input type="text" name="email"></td>-->
<!-- </tr>-->
<!-- <tr>-->
<!-- <td>年龄:</td>-->
<!-- <td><input type="text" name="age"></td>-->
<!-- </tr>-->
<!-- <tr>-->
<!-- <td>用户名:</td>-->
<!-- <td><input type="text" name="username"></td>-->
<!-- </tr>-->
<!-- <tr>-->
<!-- <td>手机号:</td>-->
<!-- <td><input type="text" name="phone"></td>-->
<!-- </tr>-->
<!-- <tr>-->
<!-- <td>个人中心:</td>-->
<!-- <td><input type="text" name="info"></td>-->
<!-- </tr>-->
<tr>
<td>验证码:</td>
<td><input type="text" name="captcha"></td>
</tr>
<tr>
<td><input type="submit" value="提交"></td>
</tr>
</table>
</form>
</body>
</html>
在forms文件里定义验证:
from wtforms import Form,StringField, validators,IntegerField
from wtforms.validators import Length, Regexp, EqualTo, Email, NumberRange, InputRequired
from wtforms.validators import URL, ValidationError
class LoginForm(Form):
# email = StringField(validators=[Email()])
# age = IntegerField(validators=[NumberRange(1, 120, message='年龄范围错误')])
# username = StringField(validators=[InputRequired(message='用户名必须要添加')])
# phone = StringField(validators=[Regexp(r'1[385]\d{9}')])
# info = StringField(validators=[URL()])
# 验证码
captcha = StringField(validators=[Length(min=4, max=4)])
# 内容?
def validate_captcha(self, field):
print(field)
if field.data != '5213':
raise ValidationError('验证码错误')
RegistForm传递的是request.form进去进行初始化,并且判断form.validate会返回用户提交的数据是否满足表单的验证。
渲染模板
form还可以渲染模板,可以少写一点点的代码,比如重写以上例子,RegistForm表单代码如下:
class RegistForm(Form):
name = StringField('用户名', validators=[Length(min=3, max=10)])
emial = StringField('邮箱', validators=[email()])
password = StringField('密码', validators=[DataRequired(), Length(min =6, max=10),EqualTo('confinEqualTo("password")])
confirm = StringField('确认密码')
以上增加了第一个位置参数,用来在html文件中,做标签提示作用。
在app中的视图函数中, 修改为如下:
@app.route('/regist/', methods=['GET', 'POST']def
def regist():
form = RegistForm(request.form)
if request.method == 'POST' and form.validate():
user = User(name=form.name.data,email=form.email.data,password=form.password.data)
db.session.add(user)
db.session.commit()
return '注册成功'
return render_tamplate('regist.html', form=form)
以上唯一不同的是在渲染模板的时候传入了form表单参数进去,这样在模板中就可以使用表单form变量了。
regist.html文件:诸如此类的形式添加
<tr>
<td>{{ form.name.label }}</td>
<td>{{ form.name() }}</td>
</tr>