mirror of
https://github.com/GSA/notifications-admin.git
synced 2026-02-09 21:04:14 -05:00
539950d772
parts of the initial setup/login stages were throwing 500s if user not already in process (ie: user directly navigated to url): * /resend-email-verification * /text-not-received * /send-new-code * verify
50 lines
1.7 KiB
Python
50 lines
1.7 KiB
Python
from flask import (
|
|
render_template,
|
|
redirect,
|
|
session,
|
|
url_for
|
|
)
|
|
|
|
from app import user_api_client
|
|
from app.main import main
|
|
from app.main.forms import TextNotReceivedForm
|
|
from app.utils import redirect_to_sign_in
|
|
|
|
|
|
@main.route('/resend-email-verification')
|
|
@redirect_to_sign_in
|
|
def resend_email_verification():
|
|
user = user_api_client.get_user_by_email(session['user_details']['email'])
|
|
user_api_client.send_verify_email(user.id, user.email_address)
|
|
return render_template('views/resend-email-verification.html', email=user.email_address)
|
|
|
|
|
|
@main.route('/text-not-received', methods=['GET', 'POST'])
|
|
@redirect_to_sign_in
|
|
def check_and_resend_text_code():
|
|
user = user_api_client.get_user_by_email(session['user_details']['email'])
|
|
|
|
if user.state == 'active':
|
|
# this is a verified user and therefore redirect to page to request resend without edit mobile
|
|
return render_template('views/verification-not-received.html')
|
|
|
|
form = TextNotReceivedForm(mobile_number=user.mobile_number)
|
|
if form.validate_on_submit():
|
|
user_api_client.send_verify_code(user.id, 'sms', to=form.mobile_number.data)
|
|
user.mobile_number = form.mobile_number.data
|
|
user_api_client.update_user(user)
|
|
return redirect(url_for('.verify'))
|
|
|
|
return render_template('views/text-not-received.html', form=form)
|
|
|
|
|
|
@main.route('/send-new-code', methods=['GET'])
|
|
@redirect_to_sign_in
|
|
def check_and_resend_verification_code():
|
|
user = user_api_client.get_user_by_email(session['user_details']['email'])
|
|
user_api_client.send_verify_code(user.id, 'sms', user.mobile_number)
|
|
if user.state == 'pending':
|
|
return redirect(url_for('main.verify'))
|
|
else:
|
|
return redirect(url_for('main.two_factor'))
|