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.""" 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, 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())