Refactor gov user check into a decorator

We quite often use it in the same way as `@user_has_permissions`.
This commit is contained in:
Chris Hill-Scott
2018-12-12 13:10:46 +00:00
parent 9a70f6a7f4
commit a8b916b57f
6 changed files with 17 additions and 23 deletions

View File

@@ -1,13 +1,6 @@
import json
from flask import (
abort,
current_app,
redirect,
render_template,
session,
url_for,
)
from flask import current_app, redirect, render_template, session, url_for
from flask_login import current_user, login_required
from notifications_utils.url_safe_token import check_token
@@ -21,7 +14,7 @@ from app.main.forms import (
ConfirmPasswordForm,
TwoFactorForm,
)
from app.utils import is_gov_user
from app.utils import user_is_gov_user
NEW_EMAIL = 'new-email'
NEW_MOBILE = 'new-mob'
@@ -56,11 +49,9 @@ def user_profile_name():
@main.route("/user-profile/email", methods=['GET', 'POST'])
@login_required
@user_is_gov_user
def user_profile_email():
if not current_user.is_gov_user:
abort(403)
def _is_email_already_in_use(email):
return user_api_client.is_email_already_in_use(email)
form = ChangeEmailForm(_is_email_already_in_use,