From d8075cd04c9b1671dcbabe93c30163d0737d8098 Mon Sep 17 00:00:00 2001 From: Andrey Anshin Date: Mon, 20 Nov 2023 18:11:54 +0400 Subject: [PATCH] Remove backcompat with Airflow 2.3/2.4 in providers (#35727) * Remove backcompat with Airflow 2.3/2.4 in providers * Revert changes in sql.py --- airflow/providers/google/cloud/hooks/gcs.py | 9 +-------- .../google/cloud/secrets/secret_manager.py | 14 ++++++-------- .../providers/microsoft/azure/secrets/key_vault.py | 14 ++++++-------- 3 files changed, 13 insertions(+), 24 deletions(-) diff --git a/airflow/providers/google/cloud/hooks/gcs.py b/airflow/providers/google/cloud/hooks/gcs.py index d8bb36037fa10..1a74d09d55cbd 100644 --- a/airflow/providers/google/cloud/hooks/gcs.py +++ b/airflow/providers/google/cloud/hooks/gcs.py @@ -45,6 +45,7 @@ from airflow.providers.google.cloud.utils.helpers import normalize_directory_path from airflow.providers.google.common.consts import CLIENT_INFO from airflow.providers.google.common.hooks.base_google import GoogleBaseAsyncHook, GoogleBaseHook +from airflow.typing_compat import ParamSpec from airflow.utils import timezone from airflow.version import version @@ -54,14 +55,6 @@ from aiohttp import ClientSession from google.api_core.retry import Retry -try: - # Airflow 2.3 doesn't have this yet - from airflow.typing_compat import ParamSpec -except ImportError: - try: - from typing import ParamSpec # type: ignore[no-redef, attr-defined] - except ImportError: - from typing_extensions import ParamSpec RT = TypeVar("RT") T = TypeVar("T", bound=Callable) diff --git a/airflow/providers/google/cloud/secrets/secret_manager.py b/airflow/providers/google/cloud/secrets/secret_manager.py index fd8b8e33e28df..a40c6bfbe5fb4 100644 --- a/airflow/providers/google/cloud/secrets/secret_manager.py +++ b/airflow/providers/google/cloud/secrets/secret_manager.py @@ -28,7 +28,6 @@ from airflow.providers.google.cloud.utils.credentials_provider import get_credentials_and_project_id from airflow.secrets import BaseSecretsBackend from airflow.utils.log.logging_mixin import LoggingMixin -from airflow.version import version as airflow_version log = logging.getLogger(__name__) @@ -154,13 +153,12 @@ def get_conn_uri(self, conn_id: str) -> str | None: :param conn_id: the connection id :return: deserialized Connection """ - if _parse_version(airflow_version) >= (2, 3): - warnings.warn( - f"Method `{self.__class__.__name__}.get_conn_uri` is deprecated and will be removed " - "in a future release. Please use method `get_conn_value` instead.", - AirflowProviderDeprecationWarning, - stacklevel=2, - ) + warnings.warn( + f"Method `{self.__class__.__name__}.get_conn_uri` is deprecated and will be removed " + "in a future release. Please use method `get_conn_value` instead.", + AirflowProviderDeprecationWarning, + stacklevel=2, + ) return self.get_conn_value(conn_id) def get_variable(self, key: str) -> str | None: diff --git a/airflow/providers/microsoft/azure/secrets/key_vault.py b/airflow/providers/microsoft/azure/secrets/key_vault.py index 794788206c137..bfa9117b111dd 100644 --- a/airflow/providers/microsoft/azure/secrets/key_vault.py +++ b/airflow/providers/microsoft/azure/secrets/key_vault.py @@ -38,7 +38,6 @@ from airflow.providers.microsoft.azure.utils import get_sync_default_azure_credential from airflow.secrets import BaseSecretsBackend from airflow.utils.log.logging_mixin import LoggingMixin -from airflow.version import version as airflow_version def _parse_version(val): @@ -170,13 +169,12 @@ def get_conn_uri(self, conn_id: str) -> str | None: :param conn_id: the connection id :return: deserialized Connection """ - if _parse_version(airflow_version) >= (2, 3): - warnings.warn( - f"Method `{self.__class__.__name__}.get_conn_uri` is deprecated and will be removed " - "in a future release. Please use method `get_conn_value` instead.", - AirflowProviderDeprecationWarning, - stacklevel=2, - ) + warnings.warn( + f"Method `{self.__class__.__name__}.get_conn_uri` is deprecated and will be removed " + "in a future release. Please use method `get_conn_value` instead.", + AirflowProviderDeprecationWarning, + stacklevel=2, + ) return self.get_conn_value(conn_id) def get_variable(self, key: str) -> str | None: