#!/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():
        db.engine.execute("SELECT setval('contributor_id_seq', (SELECT MAX(id) FROM contributor))")
        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)