diff --git a/src/dags/providers/ibm/ibm_credentials.py b/src/dags/providers/ibm/ibm_credentials.py index fe2cd2c7a744afe9233dc6d3c91c6c917efbc0f9..5435a45a3894d0ebcb42d2a13f2fd8c2f609cd5f 100644 --- a/src/dags/providers/ibm/ibm_credentials.py +++ b/src/dags/providers/ibm/ibm_credentials.py @@ -48,19 +48,23 @@ class IBMCredentials(BaseCredentials): self._client_secret = os.getenv("client_secret") self._username = os.getenv("username") self._password = os.getenv("password") - self._scope = os.getenv("scope") + def _generate_token(self) -> str: - keycloak_openid = KeycloakOpenID(server_url=uri, + if self._client_id is None: + self._populate_ad_credentials() + + keycloak_openid = KeycloakOpenID(server_url=os.getenv("KEYCLOACK_URI"), client_id=self._client_id, - realm_name=realm, + realm_name=os.getenv("REALM_NAME"), client_secret_key=self._client_secret) token = keycloak_openid.token(self._username, self._password) refresh_token = keycloak_openid.refresh_token(token['refresh_token']) access_token = refresh_token['access_token'] + return access_token @retry(stop=stop_after_attempt(RETRIES))