Replace BaseHook to Task SDK for opensearch (#52851)

Author: kyungjunleeme

providers/opensearch/src/airflow/providers/opensearch/hooks/opensearch.py

@@ -27,11 +27,7 @@
     from opensearchpy import Connection as OpenSearchConnectionClass
 
 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.opensearch.version_compat import BaseHook
 from airflow.utils.strings import to_boolean
 
 

providers/opensearch/src/airflow/providers/opensearch/version_compat.py

@@ -32,14 +32,23 @@ def get_base_airflow_version_tuple() -> tuple[int, int, int]:
     return airflow_version.major, airflow_version.minor, airflow_version.micro
 
 
-AIRFLOW_V_3_0_PLUS = get_base_airflow_version_tuple() >= (3, 0, 0)
+AIRFLOW_V_3_0_PLUS: bool = 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_0_PLUS:
     from airflow.sdk import BaseOperator
 else:
     from airflow.models import BaseOperator
 
+if AIRFLOW_V_3_1_PLUS:
+    from airflow.sdk import BaseHook
+else:
+    from airflow.hooks.base import BaseHook  # type: ignore[attr-defined,no-redef]
+
+
 __all__ = [
     "AIRFLOW_V_3_0_PLUS",
+    "AIRFLOW_V_3_1_PLUS",
+    "BaseHook",
     "BaseOperator",
 ]

providers/opensearch/tests/unit/opensearch/conftest.py

@@ -21,11 +21,7 @@
 import pytest
 
 from airflow.models import Connection
-
-try:
-    from airflow.sdk import BaseHook
-except ImportError:
-    from airflow.hooks.base import BaseHook  # type: ignore[attr-defined,no-redef]
+from airflow.providers.opensearch.version_compat import BaseHook
 
 try:
     from opensearchpy import OpenSearch