yujunjun/OpenMetadata
1
0
Fork 0
mirror of https://github.com/open-metadata/OpenMetadata.git synced 2025-11-03 20:19:31 +00:00
Code Issues Packages Projects Releases Wiki Activity

added custom oidc to airflow provider (#5598)

Browse Source 
Co-authored-by: Onkar Ravgan <onkarravgan@Onkars-MacBook-Pro.local>
This commit is contained in:
Onkar Ravgan 2022-06-25 09:12:19 +05:30 committed by GitHub
parent 95bca61719
commit 932d1adfe5
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
ingestion/src/airflow_provider_openmetadata/lineage/config/providers.py
View File

@ -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"),
)