——————————————————————前言————————————————————————————
本节讲解如何为我们的web程序添加注册新用户的功能。
————————————————————————————————————————————————————
一. 修改|-app/-auth/forms.py
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField, ValidationError
from wtforms.validators import DataRequired, Length, Email, Regexp, EqualTo
from ..models import User
class RegisterForm(FlaskForm): #注册表单
email = StringField('Email', validators=[ DataRequired(), Length(1, 64), Email() ])
username = StringField('Username', validators=[ DataRequired(), Length(1, 64), Regexp('^[A-Za-z][A-Za-z0-9_.]*$', 0, 'Username must hava only letters, numbers, dots or underscores.') ])
password = PasswordField('Password', validators=[ DataRequired(), EqualTo('password2', 'Password must match.') ])
password2 = PasswordField('Confirm Password', validators=[ DataRequired() ])
submit = SubmitField('register')
def validate_email(self, field):
if User.query.filter_by(email=field.data).first() is not None:
raise ValidationError('Email already registered.')
def validate_username(self, field):
if User.query.filter_by(username=field.data).first() is not None:
raise ValidationError('Username already Register.')
注释:
验证函数:
Regexp:
该字段填写的字符串必须符合第一个参数给定的正则表达式——字母开头, 后面可以是字母数字下划线和点号。
第二个参数是正则表达式的旗标, 不太懂, 也没搜到,默认为0就好了
第三个参数是提醒字符串, 如果该字段的填写值不符合给定的正则表达式, 就会发送提醒。
EqualTo:
因为两次填写的密码必须相同, 所以我们在其中任意一个密码字段中设置该验证函数, 其第一个参数是另一个密码字段的名称, 第二个参数是提醒字符串, 如果两次填写密码不同, 发出提醒。
自定义验证函数:
我们在该脚本定义了两个自定义的验证函数, 自定义验证函数都以validate_开头, 后面紧跟字段名称, 效果跟validators参数中的验证函数一样。
validate_email:
我们为email字段添加了一个验证函数, 如果数据库已经存在该email, 则向用户发出提醒;
validate_username:
同上。
二. 修改|-app/-auth/views.py
我们只要添加一个视图函数用来返回注册表单即可:
@auth.route('/register', methods=['GET', 'POST'])
def register():
form = RegisterForm() #创建表单
if form.validate_on_submit(): #填写的表单数据通过验证函数
u = User(email=form.email.data, username=form.username.data, password=form.password.data) #创建用户
db.session.commit(u) #把用户添加到会话
flash('You can now login.') #提示用户已经可以登录
return redirect(url_for('auth.login')) #重定向
return render_template('auth/register.html', form=form)
三. 修改|-app/templates
1.auth/login.html——在登录页面添加一个注册页面的链接, 方便创建账号。
{% extends 'base.html' %}
{% import 'bootstrap/wtf.html' as wtf %}
{% block title %}Flasky - Login{% endblock %}
{% block page_content %}
<div class="page-header">
<h1>Login</h1>
</div>
<div class="col-md-4">
{{ wtf.quick_form(form) }}
<p><a href=" {{ url_for('auth.register') }} ">Click here to register</a></p>
</div>
{% endblock %}
#在登录表单下面添加注册页面的链接, 点击以后跳转到注册页面
2.auth/register.html —— 渲染注册表单
{% extends 'base.html' %}
{% import 'bootstrap/wtf.html' as wtf %}
{% block title %}Flasky - Register{% endblock %}
{% block page_content %}
<div class="page-header">
<h1>Register</h1>
</div>
{{ wtf.quick_form(form) }}
{% endblock %}
四. 效果演示
1.启动服务器:
2.用浏览器访问根地址:
3. 点击导航条右侧Sign In, 访问登录页面:
4. 点击登录表单下方链接, 访问注册页面:
5.输入完信息后点击Register提交按钮, 注册成功:
6. 输入完信息后, 点击log_in提交按钮:
登陆成功!