darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-09-09 01:03:13 -04:00
Code Issues Packages Projects Releases Wiki Activity

109526036:Mark the sms code as being used on the two-factor page

Browse Source
This commit is contained in:
Rebecca Law
2015-12-10 16:38:34 +00:00
parent bbecc03531
commit 1ce8170ce1
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
app/main/views/two_factor.py
View File

@@ -2,7 +2,7 @@ from flask import render_template, redirect, jsonify, session
from flask_login import login_user from flask_login import login_user
from app.main import main 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 from app.main.forms import TwoFactorForm
@@ -17,6 +17,7 @@ def process_two_factor():
if form.validate_on_submit(): if form.validate_on_submit():
user = users_dao.get_user_by_id(session['user_id']) 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) login_user(user)
return redirect('/dashboard') return redirect('/dashboard')
else: else: