35 lines
1.1 KiB
Python

#!/usr/bin/env python3
from flask import Blueprint, redirect, url_for, render_template, flash
from flask_login import current_user
from app.forms import RegistrationForm
from app.models import Contributor
from .. import db
reg = Blueprint(
"reg", __name__, template_folder="templates"
)
@reg.route("/register", methods=["GET", "POST"])
def register():
if current_user.is_authenticated:
return redirect(url_for('proute.index'))
form = RegistrationForm()
if form.validate_on_submit():
my_sql = "SELECT setval('contributor_id_seq', "
my_sql += "(SELECT MAX(id) FROM contributor))"
db.engine.execute(my_sql)
db.session.commit()
contributor = Contributor(
name=form.username.data,
num_photos=0,
email=form.email.data,
)
contributor.set_password(form.password.data)
db.session.add(contributor)
db.session.commit()
flash("Congratulations, you are now a registered user!")
return redirect(url_for('auths.login'))
return render_template('register.html', title='Register', form=form)