80 lines
2.8 KiB
Python
80 lines
2.8 KiB
Python
|
# Licensed under the Apache License, Version 2.0 (the "License"); you may
|
||
|
# not use this file except in compliance with the License. You may obtain
|
||
|
# a copy of the License at
|
||
|
#
|
||
|
# http://www.apache.org/licenses/LICENSE-2.0
|
||
|
#
|
||
|
# Unless required by applicable law or agreed to in writing, software
|
||
|
# distributed under the License is distributed on an "AS IS" BASIS, WITHOUT
|
||
|
# WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. See the
|
||
|
# License for the specific language governing permissions and limitations
|
||
|
# under the License.
|
||
|
|
||
|
from keystoneclient import base
|
||
|
|
||
|
|
||
|
SAML2_ENDPOINT = '/auth/OS-FEDERATION/saml2'
|
||
|
ECP_ENDPOINT = '/auth/OS-FEDERATION/saml2/ecp'
|
||
|
|
||
|
|
||
|
class SamlManager(base.Manager):
|
||
|
"""Manager class for creating SAML assertions."""
|
||
|
|
||
|
def create_saml_assertion(self, service_provider, token_id):
|
||
|
"""Create a SAML assertion from a token.
|
||
|
|
||
|
Equivalent Identity API call:
|
||
|
POST /auth/OS-FEDERATION/saml2
|
||
|
|
||
|
:param service_provider: Service Provider resource.
|
||
|
:type service_provider: string
|
||
|
:param token_id: Token to transform to SAML assertion.
|
||
|
:type token_id: string
|
||
|
|
||
|
:returns: SAML representation of token_id
|
||
|
:rtype: string
|
||
|
"""
|
||
|
headers, body = self._create_common_request(service_provider, token_id)
|
||
|
resp, body = self.client.post(SAML2_ENDPOINT, json=body,
|
||
|
headers=headers)
|
||
|
return self._prepare_return_value(resp, resp.text)
|
||
|
|
||
|
def create_ecp_assertion(self, service_provider, token_id):
|
||
|
"""Create an ECP wrapped SAML assertion from a token.
|
||
|
|
||
|
Equivalent Identity API call:
|
||
|
POST /auth/OS-FEDERATION/saml2/ecp
|
||
|
|
||
|
:param service_provider: Service Provider resource.
|
||
|
:type service_provider: string
|
||
|
:param token_id: Token to transform to SAML assertion.
|
||
|
:type token_id: string
|
||
|
|
||
|
:returns: SAML representation of token_id, wrapped in ECP envelope
|
||
|
:rtype: string
|
||
|
"""
|
||
|
headers, body = self._create_common_request(service_provider, token_id)
|
||
|
resp, body = self.client.post(ECP_ENDPOINT, json=body,
|
||
|
headers=headers)
|
||
|
return self._prepare_return_value(resp, resp.text)
|
||
|
|
||
|
def _create_common_request(self, service_provider, token_id):
|
||
|
headers = {'Content-Type': 'application/json'}
|
||
|
body = {
|
||
|
'auth': {
|
||
|
'identity': {
|
||
|
'methods': ['token'],
|
||
|
'token': {
|
||
|
'id': token_id
|
||
|
}
|
||
|
},
|
||
|
'scope': {
|
||
|
'service_provider': {
|
||
|
'id': base.getid(service_provider)
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
}
|
||
|
|
||
|
return (headers, body)
|