Turn on redirects for check_and_resend_text_code

This is part of the work to make sure user is redirected
to the page they initially were meant to visit after
they sign in.
This commit is contained in:
Pea Tyczynska
2020-10-09 11:39:01 +01:00
parent d089424501
commit c3b7481e11
3 changed files with 22 additions and 7 deletions

View File

@@ -1,4 +1,4 @@
from flask import redirect, render_template, session, url_for
from flask import redirect, render_template, request, session, url_for
from app import user_api_client
from app.main import main
@@ -19,16 +19,17 @@ def resend_email_verification():
@redirect_to_sign_in
def check_and_resend_text_code():
user = User.from_email_address(session['user_details']['email'])
redirect_url = request.args.get('next')
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')
return render_template('views/verification-not-received.html', redirect_url=redirect_url)
form = TextNotReceivedForm(mobile_number=user.mobile_number)
if form.validate_on_submit():
user.send_verify_code(to=form.mobile_number.data)
user.update(mobile_number=form.mobile_number.data)
return redirect(url_for('.verify'))
return redirect(url_for('.verify', next=redirect_url))
return render_template('views/text-not-received.html', form=form)

View File

@@ -16,7 +16,7 @@
{{ govukButton({
"element": "a",
"text": "Resend security code",
"href": url_for('main.check_and_resend_verification_code')
"href": url_for('main.check_and_resend_verification_code', next=redirect_url)
}) }}
</p>