darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-02-05 10:53:28 -05:00
Code Issues Packages Projects Releases Wiki Activity

[WIP]: fixing unit tests

Browse Source
This commit is contained in:
Rebecca Law
2016-03-18 10:49:22 +00:00
parent 7eb4bcc592
commit d003dc4aa9
8 changed files with 140 additions and 134 deletions
Download Patch File Download Diff File Expand all files Collapse all files

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

@@ -7,8 +7,7 @@ from flask import (
request
)
from flask_login import login_user
from flask_login import login_user, current_user
from app.main import main
from app.main.dao import users_dao, services_dao
from app.main.forms import TwoFactorForm
@@ -43,6 +42,8 @@ def two_factor():
if next_url and _is_safe_redirect_url(next_url):
return redirect(next_url)
if current_user.platform_admin:
return redirect(url_for('main.show_all_services'))
if len(services) == 1:
return redirect(url_for('main.service_dashboard', service_id=services[0]['id']))
else: