Turn on redirects for check_and_resend_verification_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:44 +01:00
parent c3b7481e11
commit 1dd8b08042
2 changed files with 17 additions and 6 deletions

View File

@@ -39,10 +39,11 @@ def check_and_resend_text_code():
def check_and_resend_verification_code():
user = User.from_email_address(session['user_details']['email'])
user.send_verify_code()
redirect_url = request.args.get('next')
if user.state == 'pending':
return redirect(url_for('main.verify'))
return redirect(url_for('main.verify', next=redirect_url))
else:
return redirect(url_for('main.two_factor'))
return redirect(url_for('main.two_factor', next=redirect_url))
@main.route('/email-not-received', methods=['GET'])