This is an automated email from the ASF dual-hosted git repository.
potiuk 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 6654c150565 Move all BaseHook usages to version_compat in teradata
provider (#52858)
6654c150565 is described below
commit 6654c150565faa91d0159d071dbe0c39787479a6
Author: Phani Kumar <[email protected]>
AuthorDate: Fri Jul 4 16:45:46 2025 +0530
Move all BaseHook usages to version_compat in teradata provider (#52858)
* Move BaseHook to version_compat
* Move BaseHook to version_compat
---
providers/teradata/src/airflow/providers/teradata/hooks/ttu.py | 6 +-----
.../teradata/src/airflow/providers/teradata/version_compat.py | 8 +++++++-
2 files changed, 8 insertions(+), 6 deletions(-)
diff --git a/providers/teradata/src/airflow/providers/teradata/hooks/ttu.py
b/providers/teradata/src/airflow/providers/teradata/hooks/ttu.py
index 9128b6a88ef..1c47f4808f0 100644
--- a/providers/teradata/src/airflow/providers/teradata/hooks/ttu.py
+++ b/providers/teradata/src/airflow/providers/teradata/hooks/ttu.py
@@ -22,11 +22,7 @@ from abc import ABC
from typing import Any
from airflow.exceptions import AirflowException
-
-try:
- from airflow.sdk import BaseHook
-except ImportError:
- from airflow.hooks.base import BaseHook # type:
ignore[attr-defined,no-redef]
+from airflow.providers.teradata.version_compat import BaseHook
class TtuHook(BaseHook, ABC):
diff --git
a/providers/teradata/src/airflow/providers/teradata/version_compat.py
b/providers/teradata/src/airflow/providers/teradata/version_compat.py
index 4f8d5e32bca..f800a29bbb7 100644
--- a/providers/teradata/src/airflow/providers/teradata/version_compat.py
+++ b/providers/teradata/src/airflow/providers/teradata/version_compat.py
@@ -33,10 +33,16 @@ def get_base_airflow_version_tuple() -> tuple[int, int,
int]:
AIRFLOW_V_3_0_PLUS = get_base_airflow_version_tuple() >= (3, 0, 0)
+AIRFLOW_V_3_1_PLUS: bool = get_base_airflow_version_tuple() >= (3, 1, 0)
+
+if AIRFLOW_V_3_1_PLUS:
+ from airflow.sdk import BaseHook
+else:
+ from airflow.hooks.base import BaseHook # type:
ignore[attr-defined,no-redef]
if AIRFLOW_V_3_0_PLUS:
from airflow.sdk import BaseOperator
else:
from airflow.models import BaseOperator # type: ignore[no-redef]
-__all__ = ["AIRFLOW_V_3_0_PLUS", "BaseOperator"]
+__all__ = ["AIRFLOW_V_3_0_PLUS", "AIRFLOW_V_3_1_PLUS", "BaseHook",
"BaseOperator"]
