diff --git a/dags/test_secrets_backend.py b/dags/test_secrets_backend.py index f8409669f8b02486f0f49743df0876f560f452f4..1cfe57412d002634d058e7bf8ec46d90cc33c3fd 100644 --- a/dags/test_secrets_backend.py +++ b/dags/test_secrets_backend.py @@ -11,8 +11,12 @@ default_args = { @dag(default_args=default_args, schedule_interval=None, start_date=days_ago(2), tags=['example']) def test_secrets_backend(): @task() - def get_print_and_return_conenction(): - conn = BaseHook.get_connection('860355e9-975f-4253-9421-1815e20c879b') + def get_print_and_return_conenction(**kwargs): + oid = '860355e9-975f-4253-9421-1815e20c879b' + params = kwargs['params'] + if 'oid' in params: + oid = params['oid'] + conn = BaseHook.get_connection(oid) print(conn.get_extra()) get_print_and_return_conenction()