使用 Flask 框架写用户登录功能的Demo时碰到的各种坑（五）——实现注册功能

使用 Flask 框架写用户登录功能的Demo时碰到的各种坑（一）——创建应用

使用 Flask 框架写用户登录功能的Demo时碰到的各种坑（二）——使用蓝图功能进行模块化

使用 Flask 框架写用户登录功能的Demo时碰到的各种坑（三）——使用Flask-Login库实现登录功能

使用 Flask 框架写用户登录功能的Demo时碰到的各种坑（四）——对 run.py 的调整

使用 Flask 框架写用户登录功能的Demo时碰到的各种坑（五）——实现注册功能

一、创建 Flask-WTF 验证类

创建 register.py 文件

/demo/modules/users/forms/register.py

# config=utf-8
from flask_wtf import Form
from wtforms import StringField, PasswordField
from wtforms.validators import DataRequired

class RegisterForm(Form):
    accountNumber = StringField('accountNumber', validators=[DataRequired()])
    password = PasswordField('password', validators=[DataRequired()])
    name = StringField('name', validators=[DataRequired()])

二、实现注册功能

/demo/modules/users/views.py

1.引用上一步创建的 register.py 文件中的 RegisterForm 类

from demo.modules.users.forms.users import RegisterForm

2.引入数据库访问类

from demo.common import db

3.增加 register 函数

@loginRoute.route('/register', methods=['GET', 'POST'])
def register():
    form = RegisterForm()

    if request.method == 'POST':
        if not form.validate_on_submit():
            return render_template('register.html', form=form)

        user = User.query.filter(User.accountNumber == form.accountNumber.data).first()

        if user:
            flash("账号已存在")
            return render_template('register.html', form=form)

        user = User()
        user.accountNumber = form.accountNumber.data
        user.password = md5(form.password.data)
        user.name = form.name.data

        db.session.add(user)
        db.session.commit()

        return render_template('login.html', form=form)

    return render_template('register.html', form=form)

4.增加 register.html 页面

在 /demo/modules/users/templates 目录创建 register.html 页面:

{% extends "base.html" %}
{% block title %}注册新用户{% endblock %}
{% block head %}
    <style type="text/css"></style>
{% endblock %}
{% block content %}
    <form action="{{ url_for('user.register') }}" method="post">
        {% if form.errors or get_flashed_messages() %}
            <ul>
                {% for name, errors in form.errors.items() %}
                    {% for error in errors %}
                        <li>{{ error }}</li>
                    {% endfor %}
                {% endfor %}
                {% for message in get_flashed_messages() %}
                    <li>{{ message }}</li>
                {% endfor %}
            </ul>
        {% endif %}
        账号：{{ form.accountNumber(size=20) }}<label>{{ form.accountNumber.errors[0] }}</label><br/>
        密码：<input name="password" type="password"/><br/>
        昵称：<input name="name" type="text"/><br/>
        {{ form.hidden_tag() }}
        <button type="submit">注册</button>
    </form>
    <a href="/">返回</a>
{% endblock %}

完成！现在可以通过 /user/register 来访问并注册新用户了

注意：在这里需要进行账号是否存在的验证，需要返回自定义信息，返回的办法是：

# 视图函数需要调用
flash('your response message for user')

# 前端页面调用
for message in get_flashed_messages()
就可以输出反馈信息