From 334878d75d89dcf0be5540c9ace81ff9acd3d27b Mon Sep 17 00:00:00 2001 From: Tim Jacomb Date: Sun, 3 Mar 2019 15:08:01 +0000 Subject: [PATCH] Use display name of role --- app/views.py | 10 ++++------ 1 file changed, 4 insertions(+), 6 deletions(-) diff --git a/app/views.py b/app/views.py index 9600efc..74e0ebc 100755 --- a/app/views.py +++ b/app/views.py @@ -344,9 +344,8 @@ def saml_authorized(): elif admin_group_name in user_groups: uplift_to_admin(user) elif admin_attribute_name or group_attribute_name: - user_role = Role.query.filter_by(name='User').first().id - if user.role_id != user_role: - user.role_id = user_role + if user.role.name != 'User': + user.role_id = Role.query.filter_by(name='User').first().id history = History(msg='Demoting {0} to user'.format(user.username), created_by='SAML Assertion') history.add() user.plain_text_password = None @@ -381,9 +380,8 @@ def handle_account(account_name): def uplift_to_admin(user): - admin_role = Role.query.filter_by(name='Administrator').first().id - if user.role_id != admin_role: - user.role_id = admin_role + if user.role.name != 'Administrator': + user.role_id = Role.query.filter_by(name='Administrator').first().id history = History(msg='Promoting {0} to administrator'.format(user.username), created_by='SAML Assertion') history.add()