109526036: Fix bug.

If one of the codes was invalid and one was valid on the verify page the valid code would be marked as used.
This commit is contained in:
Rebecca Law
2015-12-10 16:34:29 +00:00
parent 2b4097dd2d
commit bbecc03531
3 changed files with 11 additions and 3 deletions

View File

@@ -2,7 +2,7 @@ from flask import render_template, redirect, jsonify, session
from flask_login import login_user
from app.main import main
from app.main.dao import users_dao
from app.main.dao import users_dao, verify_codes_dao
from app.main.forms import VerifyForm
@@ -16,6 +16,8 @@ def process_verify():
form = VerifyForm()
if form.validate_on_submit():
user = users_dao.get_user_by_id(session['user_id'])
verify_codes_dao.use_code_for_user_and_type(user_id=user.id, code_type='email')
verify_codes_dao.use_code_for_user_and_type(user_id=user.id, code_type='sms')
users_dao.activate_user(user.id)
login_user(user)
return redirect('/add-service')