Merge pull request #26 from alphagov/persis_verify_codes

Persis verify codes

app/main/views/two_factor.py

@@ -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 TwoFactorForm
 
 
@@ -17,6 +17,7 @@ def process_two_factor():
 
     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='sms')
         login_user(user)
         return redirect('/dashboard')
     else: