from flask import Flask
from flask import (request, session, render_template, redirect, url_for)
from flask import Blueprint

from werkzeug.security import check_password_hash

from . import incr

bp = Blueprint('register', __name__, url_prefix='/reg')

# Come at the app without a form, you're new.  The 
@bp.route("/reg")
@incr.inccnt
def gen():
    # Give the page.
    return render_template('regpage.html', count=session['count'])

@bp.route("/reg",methods=["POST"])
@incr.inccnt
def process():
    # Get the creds.
    userid = request.form['userid']
    username = request.form['username']
    pwd1 = request.form['pwd1']
    pwd2 = request.form['pwd2']

    # Check the fields are filled out.  Javascript should have done this check,
    # but we can't depend on that, so we have to check, too.  But we don't have
    # to polite.
    if userid == '' or username == '' or pwd1 == '' or pwd2 == '':
        return render_template('regpage.html', count=session['count'],
                               userid=userid, username=username,
                               err='Incomplete form')

    # Create the account.
    from . import db
    err = db.add_user(userid, username, pwd1, pwd2)
    if err != None:
        return render_template('regpage.html', count=session['count'],
                               userid=userid, username=username, err=err)

    return redirect(url_for("gen.count"))