Skip to content
Snippets Groups Projects
Commit 294b427a authored by Christian Boettcher's avatar Christian Boettcher
Browse files

get datacat connection data from airflow config

parent c36bcc9d
No related branches found
No related tags found
No related merge requests found
Pipeline #86128 failed
...@@ -74,15 +74,19 @@ class DatacatSecretsBackend(BaseSecretsBackend): ...@@ -74,15 +74,19 @@ class DatacatSecretsBackend(BaseSecretsBackend):
# TODO init with datacat elevated login to prevent secrets acces from jobs # TODO init with datacat elevated login to prevent secrets acces from jobs
def __init__(self, **kwargs):
super().__init__(**kwargs)
self.url = kwargs["url"]
self.user = kwargs["user"]
self.password = kwargs["password"]
def get_connection(self, conn_id: str): def get_connection(self, conn_id: str):
"""returns a Connection object created from the <conenction_type>/<conn_id> object in the datacatalog""" """returns a Connection object created from the <conenction_type>/<conn_id> object in the datacatalog"""
# only for testing: check that a specific oid has been requested # only for testing: check that a specific oid has been requested
if conn_id != "860355e9-975f-4253-9421-1815e20c879b": if conn_id != "860355e9-975f-4253-9421-1815e20c879b":
return None return None
secrets_conn = DataCatConnectionWithSecrets(self.url, self.user, self.password)
hook = DataCatalogHook("datacatalog")
secrets_conn = DataCatConnectionWithSecrets(hook.connection.url, hook.connection.user, hook.connection._password)
data = secrets_conn.get_object(connection_backend_type, conn_id) data = secrets_conn.get_object(connection_backend_type, conn_id)
secrets = secrets_conn.get_all_secret_key_value(connection_backend_type, conn_id) secrets = secrets_conn.get_all_secret_key_value(connection_backend_type, conn_id)
conn = get_connection_from_entry(data, secrets, connection_backend_type, conn_id) conn = get_connection_from_entry(data, secrets, connection_backend_type, conn_id)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment