45 lines
1.6 KiB
Python
45 lines
1.6 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 keystoneauth1 import exceptions
|
|
|
|
|
|
class ServiceProviders(object):
|
|
"""Helper methods for dealing with Service Providers."""
|
|
|
|
@classmethod
|
|
def from_token(cls, token):
|
|
if 'token' not in token:
|
|
raise ValueError('Token format does not support service'
|
|
'providers.')
|
|
|
|
return cls(token['token'].get('service_providers', []))
|
|
|
|
def __init__(self, service_providers):
|
|
|
|
def normalize(service_providers_list):
|
|
return dict((sp['id'], sp) for sp in service_providers_list
|
|
if 'id' in sp)
|
|
self._service_providers = normalize(service_providers)
|
|
|
|
def _get_service_provider(self, sp_id):
|
|
try:
|
|
return self._service_providers[sp_id]
|
|
except KeyError:
|
|
raise exceptions.ServiceProviderNotFound(sp_id)
|
|
|
|
def get_sp_url(self, sp_id):
|
|
return self._get_service_provider(sp_id).get('sp_url')
|
|
|
|
def get_auth_url(self, sp_id):
|
|
return self._get_service_provider(sp_id).get('auth_url')
|