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

from datetime import datetime, timedelta

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

from client.errors import HTTPError

from app.main import main
from app.main.dao import users_dao
from app.main.forms import RegisterUserForm

from app import user_api_client

# TODO how do we handle duplicate unverifed email addresses?
# malicious or otherwise.
from app.notify_client.sender import send_sms_code, send_email_code


@main.route('/register', methods=['GET', 'POST'])
def register():
    form = RegisterUserForm(users_dao.get_user_by_email)

    if form.validate_on_submit():
        try:
            user = user_api_client.register_user(form.name.data,
                                                 form.email_address.data,
                                                 form.mobile_number.data,
                                                 form.password.data)
        except HTTPError as e:
            if e.status_code == 404:
                abort(404)
            else:
                raise e

        # 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.
        send_sms_code(user_id=user.id, mobile_number=user.mobile_number)
        send_email_code(user_id=user.id, email=user.email_address)
        session['expiry_date'] = str(datetime.now() + timedelta(hours=1))
        session['user_details'] = {"email": user.email_address, "id": user.id}
        return redirect('/verify')

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