saml2: improve attributes management

This commit is contained in:
Jérôme Schneider 2014-12-05 13:39:58 +01:00
parent 69d1dc3ead
commit 69917267cb
1 changed files with 7 additions and 1 deletions

View File

@ -270,7 +270,13 @@ class SAML2Auth(AuthForm):
env['beaker.session']['validated'] = True
attributes = saml2utils.get_attributes_from_assertion(login.assertion,
logger)
env['beaker.session']['attributes'] = attributes
clean_attributes = dict()
for k, v in attributes.iteritems():
if len(k) > 1:
clean_attributes[k[0]] = v
else:
clean_attributes[k] = v
env['beaker.session']['attributes'] = clean_attributes
env['beaker.session']['unique_id'] = login.nameIdentifier.content
env['beaker.session']['liberty_session'] = login.session.dump()
env['beaker.session'].save()