Class OAuth2Provider

shared/auth/OAuth2Provider.py:144–168

Factory Class For OIDC Clients implementation. Depending on the setting set in settings.OIDC_PROVIDER_NAME this class will instantiatiate a different client oidc provider implementation for users to manage login, registration, token refresh, etc...

class OAuth2Provider(metaclass = Singleton):
    """
        Factory Class For OIDC Clients implementation.
        Depending on the setting set in settings.OIDC_PROVIDER_NAME
        this class will instantiatiate a different client oidc provider implementation for
        users to manage login, registration, token refresh, etc...
    """
    oidc_client: OAuth2ClientBase

    def __init__(self):
        from shared.auth.KeycloakDiffgramClient import KeycloakDiffgramClient
        from shared.auth.CognitoDiffgramClient import CognitoDiffgramClient

        provider = settings.OAUTH2_PROVIDER_NAME

        if not provider:
            raise ValueError("No DIFFGRAM_STATIC_STORAGE_PROVIDER env var set. valid values are [gcp, aws, azure]")

        if provider == 'keycloak':
            self.oidc_client = KeycloakDiffgramClient()
        if provider == 'cognito':
            self.oidc_client = CognitoDiffgramClient()

    def get_client(self):
        return self.oidc_client