from flask import Flask, render_template
from flask_wtf import FlaskForm
from wtforms import StringField, PasswordField, SubmitField
from wtforms.validators import DataRequired, Length, EqualTo

app = Flask(__name__)
app.secret_key = '34sdfji9453#$@'


class RegisterForm(FlaskForm):
    username = StringField(label='用户名：',
                           validators=[DataRequired(message='用户名不能为空'),
                                       Length(3, 15, message='长度应该为3~15个字符')])
    password = PasswordField('密码：',
                             validators=[DataRequired(message='密码不能为空')])
    password2 = PasswordField('确认密码：',
                              validators=[DataRequired(message='密码不能为空'),
                                          EqualTo('password', message='两次密码不一致')])
    submit = SubmitField('注册')


@app.route('/register', methods=['GET', 'POST'])
def register():
    form = RegisterForm()
    return render_template('register_wtf.html', form=form)


if __name__ == '__main__':
    app.run(debug=True)