diff --git a/extra/modules/root.ptl b/extra/modules/root.ptl index ee4ddfe..b7f6489 100644 --- a/extra/modules/root.ptl +++ b/extra/modules/root.ptl @@ -27,16 +27,15 @@ from authentic.root import LoginError, RegistrationError import stores -def is_federated_with_msp(self): - if not self.lasso_proxy_dump: +def is_federated_with_msp(authentic_identity): + if not authentic_identity.lasso_proxy_dump: return False - identity = lasso.Identity.newFromDump(self.lasso_proxy_dump) + identity = lasso.Identity.newFromDump(authentic_identity.lasso_proxy_dump) msp_provider_id = get_cfg('adeline', {}).get('msp_idp') for p in identity.providerIds: if p == msp_provider_id: return True return False -identities.Identity.is_federated_with_msp = is_federated_with_msp def get_key_from_provider_id(provider_id): return provider_id.replace('://', '-').replace('/', '-') @@ -72,7 +71,7 @@ class AlternateRootDirectory(OldRootDirectory): except KeyError: get_session_manager().expire_session() else: - if identity.is_federated_with_msp(): + if is_federated_with_msp(identity): federated_with_msp = 'plop' # any value will do get_response().filter['federated_with_msp'] = federated_with_msp get_response().filter['msp_session_image_url'] = get_cfg('adeline', {}).get( @@ -87,7 +86,7 @@ class AlternateRootDirectory(OldRootDirectory): federated_with_msp = False if get_session() and get_session().user: identity = identities.get_store().get_identity(get_session().user) - if identity.is_federated_with_msp(): + if is_federated_with_msp(identity): federated_with_msp = True for klp, lp in get_cfg('providers', {}).items(): if lp['role'] == lasso.PROVIDER_ROLE_IDP: @@ -144,7 +143,7 @@ class AlternateRootDirectory(OldRootDirectory): '%s' % _('Our partners services') '