37 lines
1.3 KiB
Python
37 lines
1.3 KiB
Python
from oauth.services.github import GithubOAuthService
|
|
from oauth.services.google import GoogleOAuthService
|
|
from oauth.oidc import OIDCLoginService
|
|
|
|
CUSTOM_LOGIN_SERVICES = {
|
|
'GITHUB_LOGIN_CONFIG': GithubOAuthService,
|
|
'GOOGLE_LOGIN_CONFIG': GoogleOAuthService,
|
|
}
|
|
|
|
PREFIX_BLACKLIST = ['ldap', 'jwt', 'keystone']
|
|
|
|
class OAuthLoginManager(object):
|
|
""" Helper class which manages all registered OAuth login services. """
|
|
def __init__(self, config, client=None):
|
|
self.services = []
|
|
|
|
# Register the endpoints for each of the OAuth login services.
|
|
for key in config.keys():
|
|
# All keys which end in _LOGIN_CONFIG setup a login service.
|
|
if key.endswith('_LOGIN_CONFIG'):
|
|
if key in CUSTOM_LOGIN_SERVICES:
|
|
custom_service = CUSTOM_LOGIN_SERVICES[key](config, key)
|
|
if custom_service.login_enabled(config):
|
|
self.services.append(custom_service)
|
|
else:
|
|
prefix = key.rstrip('_LOGIN_CONFIG').lower()
|
|
if prefix in PREFIX_BLACKLIST:
|
|
raise Exception('Cannot use reserved config name %s' % key)
|
|
|
|
self.services.append(OIDCLoginService(config, key, client=client))
|
|
|
|
def get_service(self, service_id):
|
|
for service in self.services:
|
|
if service.service_id() == service_id:
|
|
return service
|
|
|
|
return None |