Chapter 8: User registration (8d)

app/auth/forms.py

@@ -1,6 +1,8 @@
 from flask_wtf import FlaskForm
 from wtforms import StringField, PasswordField, BooleanField, SubmitField
-from wtforms.validators import DataRequired, Length, Email
+from wtforms.validators import DataRequired, Length, Email, Regexp, EqualTo
+from wtforms import ValidationError
+from ..models import User
 
 
 class LoginForm(FlaskForm):
@@ -9,3 +11,25 @@ class LoginForm(FlaskForm):
  password = PasswordField('Password', validators=[DataRequired()])
  remember_me = BooleanField('Keep me logged in')
  submit = SubmitField('Log In')
+
+
+class RegistrationForm(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,
+ 'Usernames must have only letters, numbers, dots or '
+ 'underscores')])
+ password = PasswordField('Password', validators=[
+ DataRequired(), EqualTo('password2', message='Passwords must match.')])
+ password2 = PasswordField('Confirm password', validators=[DataRequired()])
+ submit = SubmitField('Register')
+
+ def validate_email(self, field):
+ if User.query.filter_by(email=field.data.lower()).first():
+ raise ValidationError('Email already registered.')
+
+ def validate_username(self, field):
+ if User.query.filter_by(username=field.data).first():
+ raise ValidationError('Username already in use.')

app/auth/views.py

@@ -1,8 +1,11 @@
 from flask import render_template, redirect, request, url_for, flash
-from flask_login import login_user, logout_user, login_required
+from flask_login import login_user, logout_user, login_required, \
+ current_user
 from . import auth
+from .. import db
 from ..models import User
-from .forms import LoginForm
+from ..email import send_email
+from .forms import LoginForm, RegistrationForm
 
 
 @auth.route('/login', methods=['GET', 'POST'])
@@ -26,3 +29,17 @@ def logout():
  logout_user()
  flash('You have been logged out.')
  return redirect(url_for('main.index'))
+
+
+@auth.route('/register', methods=['GET', 'POST'])
+def register():
+ form = RegistrationForm()
+ if form.validate_on_submit():
+ user = User(email=form.email.data.lower(),
+ username=form.username.data,
+ password=form.password.data)
+ db.session.add(user)
+ db.session.commit()
+ flash('You can now login.')
+ return redirect(url_for('auth.login'))
+ return render_template('auth/register.html', form=form)

app/templates/auth/login.html

@@ -9,5 +9,7 @@ <h1>Login</h1>
 </div>
 <div class="col-md-4">
  {{ wtf.quick_form(form) }}
+ <br>
+ <p>New user? <a href="{{ url_for('auth.register') }}">Click here to register</a>.</p>
 </div>
 {% endblock %}

app/templates/auth/register.html

@@ -0,0 +1,13 @@
+{% 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>
+<div class="col-md-4">
+ {{ wtf.quick_form(form) }}
+</div>
+{% endblock %}