From 932d1adfe5dd6b5f6a96c3dfee71ff4eba4647e0 Mon Sep 17 00:00:00 2001 From: Onkar Ravgan Date: Sat, 25 Jun 2022 09:12:19 +0530 Subject: [PATCH] added custom oidc to airflow provider (#5598) Co-authored-by: Onkar Ravgan --- .../lineage/config/providers.py | 15 +++++++++++++++ 1 file changed, 15 insertions(+) diff --git a/ingestion/src/airflow_provider_openmetadata/lineage/config/providers.py b/ingestion/src/airflow_provider_openmetadata/lineage/config/providers.py index 8f247f8e775..6d88afa646a 100644 --- a/ingestion/src/airflow_provider_openmetadata/lineage/config/providers.py +++ b/ingestion/src/airflow_provider_openmetadata/lineage/config/providers.py @@ -25,6 +25,9 @@ from metadata.generated.schema.security.client.auth0SSOClientConfig import ( from metadata.generated.schema.security.client.azureSSOClientConfig import ( AzureSSOClientConfig, ) +from metadata.generated.schema.security.client.customOidcSSOClientConfig import ( + CustomOIDCSSOClientConfig, +) from metadata.generated.schema.security.client.googleSSOClientConfig import ( GoogleSSOClientConfig, ) @@ -104,3 +107,15 @@ def load_om_auth() -> OpenMetadataJWTClientConfig: Load config for Azure Auth """ return OpenMetadataJWTClientConfig(jwtToken=conf.get(LINEAGE, "jwt_token")) + + +@provider_config_registry.add(AuthProvider.custom_oidc.value) +def load_custom_oidc_auth() -> CustomOIDCSSOClientConfig: + """ + Load config for Custom OIDC Auth + """ + return CustomOIDCSSOClientConfig( + clientId=conf.get(LINEAGE, "client_id"), + secretKey=conf.get(LINEAGE, "secret_key"), + tokenEndpoint=conf.get(LINEAGE, "token_endpoint"), + )