From 167ec9a5ae9c9fa52d51cf3076bace49bfba1a29 Mon Sep 17 00:00:00 2001 From: kyungjunleeme Date: Mon, 14 Jul 2025 18:37:50 +0900 Subject: [PATCH 1/3] CHG: get_connection in redshift_sql --- .../src/airflow/providers/amazon/aws/hooks/redshift_sql.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/providers/amazon/src/airflow/providers/amazon/aws/hooks/redshift_sql.py b/providers/amazon/src/airflow/providers/amazon/aws/hooks/redshift_sql.py index 76e71eafb4b7b..a4e3a76250a15 100644 --- a/providers/amazon/src/airflow/providers/amazon/aws/hooks/redshift_sql.py +++ b/providers/amazon/src/airflow/providers/amazon/aws/hooks/redshift_sql.py @@ -74,7 +74,7 @@ def get_ui_field_behaviour(cls) -> dict: @cached_property def conn(self): - return self.get_connection(self.redshift_conn_id) + return self.get_connection(self.get_conn_id()) def _get_conn_params(self) -> dict[str, str | int]: """Retrieve connection parameters.""" From 6bd5109dcc06cf78a0b110a83e740e91ff20c5ff Mon Sep 17 00:00:00 2001 From: kyungjunleeme Date: Mon, 14 Jul 2025 18:45:12 +0900 Subject: [PATCH 2/3] CHG: get_connection in exasolhook --- providers/exasol/src/airflow/providers/exasol/hooks/exasol.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/providers/exasol/src/airflow/providers/exasol/hooks/exasol.py b/providers/exasol/src/airflow/providers/exasol/hooks/exasol.py index b6f807d227f90..e920e7e54b001 100644 --- a/providers/exasol/src/airflow/providers/exasol/hooks/exasol.py +++ b/providers/exasol/src/airflow/providers/exasol/hooks/exasol.py @@ -59,8 +59,7 @@ def __init__(self, *args, **kwargs) -> None: self.schema = kwargs.pop("schema", None) def get_conn(self) -> ExaConnection: - conn_id = self.get_conn_id() - conn = self.get_connection(conn_id) + conn = self.get_connection(self.get_conn_id()) conn_args = { "dsn": f"{conn.host}:{conn.port}", "user": conn.login, From 58df219c042cc183b265956002eaf3fb7a0f4f4d Mon Sep 17 00:00:00 2001 From: kyungjunleeme Date: Mon, 14 Jul 2025 18:47:16 +0900 Subject: [PATCH 3/3] CHG: get_connection in sqlite --- providers/sqlite/src/airflow/providers/sqlite/hooks/sqlite.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/providers/sqlite/src/airflow/providers/sqlite/hooks/sqlite.py b/providers/sqlite/src/airflow/providers/sqlite/hooks/sqlite.py index c0d10f6664513..06a52efc2c6b5 100644 --- a/providers/sqlite/src/airflow/providers/sqlite/hooks/sqlite.py +++ b/providers/sqlite/src/airflow/providers/sqlite/hooks/sqlite.py @@ -46,8 +46,7 @@ def get_conn(self) -> sqlite3.dbapi2.Connection: def get_uri(self) -> str: """Override DbApiHook get_uri method for get_sqlalchemy_engine().""" - conn_id = self.get_conn_id() - airflow_conn = self.get_connection(conn_id) + airflow_conn = self.get_connection(self.get_conn_id()) if airflow_conn.conn_type is None: airflow_conn.conn_type = self.conn_type airflow_uri = unquote(airflow_conn.get_uri())