darkhelm/notifications-admin
1
0
Fork 0
mirror of https://github.com/GSA/notifications-admin.git synced 2026-08-24 00:07:02 -04:00
Code Issues Packages Projects Releases Wiki Activity

Refactor government user check onto model

Browse Source
This commit is contained in:
Chris Hill-Scott
2018-12-12 12:29:08 +00:00
parent 37f73e48a6
commit 126db71de6
4 changed files with 11 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
app/main/views/add_service.py
View File

@@ -75,7 +75,7 @@ def add_service():
service_id = _add_invited_user_to_service(invited_user)
return redirect(url_for('main.service_dashboard', service_id=service_id))
if not is_gov_user(current_user.email_address):
if not current_user.is_gov_user:
abort(403)
form = CreateServiceForm()

2
app/main/views/choose_account.py
View File

@@ -26,7 +26,7 @@ def choose_account():
'views/choose-account.html',
organisations=orgs_and_services['organisations'],
services_without_organisations=orgs_and_services['services_without_organisations'],
can_add_service=is_gov_user(current_user.email_address)
can_add_service=current_user.is_gov_user,
)

4
app/main/views/user_profile.py
View File

@@ -33,7 +33,7 @@ NEW_MOBILE_PASSWORD_CONFIRMED = 'new-mob-password-confirmed'
def user_profile():
return render_template(
'views/user-profile.html',
can_see_edit=is_gov_user(current_user.email_address)
can_see_edit=current_user.is_gov_user,
)
@@ -58,7 +58,7 @@ def user_profile_name():
@login_required
def user_profile_email():
if not is_gov_user(current_user.email_address):
if not current_user.is_gov_user:
abort(403)
def _is_email_already_in_use(email):