mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-08-03 05:11:03 -04:00
c858869a52
Refactored the forms so that fields like email_address can be used in multiple forms. Refactored form validation so that a query function is passed into the form to be run, this way the form is not exposed to the dao layer and the query is more efficient. This PR still requires some frontend attention. Will work with Chris to update the templates.
38 lines
1.5 KiB
Python
38 lines
1.5 KiB
Python
from datetime import datetime, timedelta
|
|
|
|
from flask import render_template, redirect, session
|
|
|
|
from app.main import main
|
|
from app.main.dao import users_dao
|
|
from app.main.forms import RegisterUserForm
|
|
from app.main.views import send_sms_code, send_email_code
|
|
from app.models import User
|
|
|
|
|
|
# TODO how do we handle duplicate unverifed email addresses?
|
|
# malicious or otherwise.
|
|
@main.route('/register', methods=['GET', 'POST'])
|
|
def register():
|
|
form = RegisterUserForm(users_dao.get_user_by_email)
|
|
|
|
if form.validate_on_submit():
|
|
user = User(name=form.name.data,
|
|
email_address=form.email_address.data,
|
|
mobile_number=form.mobile_number.data,
|
|
password=form.password.data,
|
|
created_at=datetime.now(),
|
|
role_id=1)
|
|
users_dao.insert_user(user)
|
|
# 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=form.mobile_number.data)
|
|
send_email_code(user_id=user.id, email=form.email_address.data)
|
|
session['expiry_date'] = str(datetime.now() + timedelta(hours=1))
|
|
session['user_email'] = user.email_address
|
|
return redirect('/verify')
|
|
|
|
return render_template('views/register.html', form=form)
|