57 lines
2.2 KiB
Python
57 lines
2.2 KiB
Python
# authentic2 - versatile identity manager
|
|
# Copyright (C) 2010-2020 Entr'ouvert
|
|
#
|
|
# This program is free software: you can redistribute it and/or modify it
|
|
# under the terms of the GNU Affero General Public License as published
|
|
# by the Free Software Foundation, either version 3 of the License, or
|
|
# (at your option) any later version.
|
|
#
|
|
# This program is distributed in the hope that it will be useful,
|
|
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
# GNU Affero General Public License for more details.
|
|
#
|
|
# You should have received a copy of the GNU Affero General Public License
|
|
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
|
|
|
import django.apps
|
|
from django import template
|
|
from mellon.utils import get_idp
|
|
|
|
|
|
class AppConfig(django.apps.AppConfig):
|
|
|
|
name = 'authentic2_auth_saml'
|
|
|
|
def ready(self):
|
|
from django.db.models.signals import pre_save
|
|
|
|
from authentic2.custom_user.models import DeletedUser
|
|
|
|
pre_save.connect(self.pre_save_deleted_user, sender=DeletedUser)
|
|
|
|
def pre_save_deleted_user(self, sender, instance, **kwargs):
|
|
'''Delete and copy UserSamlIdentifier to old_data'''
|
|
from mellon.models import UserSAMLIdentifier
|
|
|
|
saml_accounts = UserSAMLIdentifier.objects.filter(user__uuid=instance.old_uuid).order_by('id')
|
|
for saml_account in saml_accounts:
|
|
instance.old_data = instance.old_data or {}
|
|
instance.old_data.setdefault('saml_accounts', []).append(
|
|
{
|
|
'issuer': saml_account.issuer,
|
|
'name_id': saml_account.name_id,
|
|
}
|
|
)
|
|
|
|
def a2_hook_manager_user_data(self, view, user):
|
|
user_saml_identifiers = user.saml_identifiers.all()
|
|
if not user_saml_identifiers:
|
|
return ['']
|
|
for user_saml_identifier in user_saml_identifiers:
|
|
user_saml_identifier.idp = get_idp(user_saml_identifier.issuer)
|
|
context = {'user_saml_identifiers': user_saml_identifiers}
|
|
return [
|
|
template.loader.get_template('authentic2_auth_saml/manager_user_sidebar.html').render(context)
|
|
]
|