entrouvert
/
django-mellon
16
0
Fork 0
Code Issues Pull Requests Releases Activity

misc: pass the authentication request to the adapter object (#50833)

This commit is contained in:
Benjamin Dauvergne 2021-02-03 21:21:35 +01:00
parent 7cd78e96ab
commit d106f3da90
3 changed files with 7 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mellon/adapters.py
View File

@ -58,6 +58,9 @@ def display_truncated_list(l, max_length=10):
class DefaultAdapter(object): class DefaultAdapter(object):
def __init__(self, request=None):
self.request = request
def get_idp(self, entity_id): def get_idp(self, entity_id):
'''Find the first IdP definition matching entity_id''' '''Find the first IdP definition matching entity_id'''
for idp in self.get_idps(): for idp in self.get_idps():

2
mellon/backends.py
View File

@ -34,7 +34,7 @@ class SAMLBackend(ModelBackend):
if not idp: if not idp:
logger.debug('unknown idp %s', saml_attributes['issuer']) logger.debug('unknown idp %s', saml_attributes['issuer'])
return None return None
adapters = utils.get_adapters(idp) adapters = utils.get_adapters(idp, request=request)
for adapter in adapters: for adapter in adapters:
if not hasattr(adapter, 'authorize'): if not hasattr(adapter, 'authorize'):
continue continue

5
mellon/utils.py
View File

@ -180,11 +180,12 @@ def import_object(path):
@to_list @to_list
def get_adapters(idp={}): def get_adapters(idp={}, **kwargs):
idp = idp or {} idp = idp or {}
adapters = tuple(idp.get('ADAPTER', ())) + tuple(app_settings.ADAPTER) adapters = tuple(idp.get('ADAPTER', ())) + tuple(app_settings.ADAPTER)
for adapter in adapters: for adapter in adapters:
yield import_object(adapter)() klass = import_object(adapter)
yield klass(**kwargs)
def get_values(saml_attributes, name): def get_values(saml_attributes, name):