diff --git a/app/auth/views.py b/app/auth/views.py index 812f1e1e..417aae50 100644 --- a/app/auth/views.py +++ b/app/auth/views.py @@ -1,7 +1,8 @@ from flask import flash, redirect, render_template, request, url_for from flask_login import login_required, login_user, logout_user from . import auth -from .forms import LoginForm, PasswordResetRequestForm +from .. import db +from .forms import LoginForm, PasswordResetRequestForm, RegistrationForm from ..email import send_email from ..models import User @@ -31,6 +32,14 @@ def logout(): @auth.route('/register', methods=['GET', 'POST']) def register(): + form = RegistrationForm() + if form.validate_on_submit(): + user = User(email=form.email.data, username=form.username.data, + password=form.password.data) + db.session.add(user) + db.session.commit() + flash('Successfully registered! You can now login.') + return redirect(url_for('auth.login')) return render_template('auth/register.html.j2')