This is an automated email from the ASF dual-hosted git repository.
jscheffl pushed a commit to branch main
in repository https://gitbox.apache.org/repos/asf/airflow.git
The following commit(s) were added to refs/heads/main by this push:
new f244427e661 Refactoring get con part dbapihook in providers (#53335)
f244427e661 is described below
commit f244427e6613928fd3c3ea62c4f261834b7f2521
Author: Kyungjun Lee <[email protected]>
AuthorDate: Tue Jul 15 03:13:10 2025 +0900
Refactoring get con part dbapihook in providers (#53335)
* CHG: get_connection in redshift_sql
* CHG: get_connection in exasolhook
* CHG: get_connection in sqlite
---
.../amazon/src/airflow/providers/amazon/aws/hooks/redshift_sql.py | 2 +-
providers/exasol/src/airflow/providers/exasol/hooks/exasol.py | 3 +--
providers/sqlite/src/airflow/providers/sqlite/hooks/sqlite.py | 3 +--
3 files changed, 3 insertions(+), 5 deletions(-)
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 76e71eafb4b..a4e3a76250a 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 @@ class RedshiftSQLHook(DbApiHook):
@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 b6f807d227f..e920e7e54b0 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 @@ class ExasolHook(DbApiHook):
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 c0d10f66645..06a52efc2c6 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 @@ class SqliteHook(DbApiHook):
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())
