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
Branches
Tags release-v0.0.2
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