notifications-admin/app/main/views/register.py

from datetime import (
    datetime,
    timedelta
)

from flask import (
    render_template,
    redirect,
    session,
    abort,
    url_for,
    flash
)

from flask.ext.login import current_user

from app.main import main

from app.main.forms import (
    RegisterUserForm,
    RegisterUserFromInviteForm
)

from app import user_api_client


@main.route('/register', methods=['GET', 'POST'])
def register():
    if current_user and current_user.is_authenticated():
        return redirect(url_for('main.choose_service'))

    form = RegisterUserForm()
    if form.validate_on_submit():
        registered = _do_registration(form, send_sms=False)
        if registered:
            return redirect(url_for('main.registration_continue'))
        else:
            flash('There was an error registering your account')
            return render_template('views/register.html', form=form), 400

    return render_template('views/register.html', form=form)


@main.route('/register-from-invite', methods=['GET', 'POST'])
def register_from_invite():
    form = RegisterUserFromInviteForm()
    invited_user = session.get('invited_user')
    if not invited_user:
        abort(404)

    if form.validate_on_submit():
        if form.service.data != invited_user['service'] or form.email_address.data != invited_user['email_address']:
            abort(400)
        registered = _do_registration(form, send_email=False)
        if registered:
            return redirect(url_for('main.verify'))
        else:
            flash('There was an error registering your account')

    form.service.data = invited_user['service']
    form.email_address.data = invited_user['email_address']

    return render_template('views/register-from-invite.html', email_address=invited_user['email_address'], form=form)


def _do_registration(form, service=None, send_sms=True, send_email=True):
    if user_api_client.is_email_unique(form.email_address.data):
        user = user_api_client.register_user(form.name.data,
                                             form.email_address.data,
                                             form.mobile_number.data,
                                             form.password.data)

        # TODO possibly there should be some exception handling
        # for sending sms and email codes.
        # How do we report to the user there is a problem with
        # sending codes apart from service unavailable?
        # at the moment i believe http 500 is fine.

        if send_email:
            user_api_client.send_verify_email(user.id, user.email_address)

        if send_sms:
            user_api_client.send_verify_code(user.id, 'sms', user.mobile_number)
        session['expiry_date'] = str(datetime.now() + timedelta(hours=1))
        session['user_details'] = {"email": user.email_address, "id": user.id}
        return True
    else:
        return False


@main.route('/registration-continue')
def registration_continue():
    return render_template('views/registration-continue.html')