From 214f28b03ed732f568adb9d7ab437f7068f58494 Mon Sep 17 00:00:00 2001 From: Wei Lee Date: Tue, 29 Oct 2024 22:34:07 +0800 Subject: [PATCH] fix(providers/amazon): alias is_authorized_dataset to is_authorized_asset (#43470) --- .../amazon/aws/auth_manager/aws_auth_manager.py | 13 ++++++++++++- 1 file changed, 12 insertions(+), 1 deletion(-) diff --git a/providers/src/airflow/providers/amazon/aws/auth_manager/aws_auth_manager.py b/providers/src/airflow/providers/amazon/aws/auth_manager/aws_auth_manager.py index face67c38fb57..414907961ce71 100644 --- a/providers/src/airflow/providers/amazon/aws/auth_manager/aws_auth_manager.py +++ b/providers/src/airflow/providers/amazon/aws/auth_manager/aws_auth_manager.py @@ -17,6 +17,7 @@ from __future__ import annotations import argparse +import warnings from collections import defaultdict from functools import cached_property from typing import TYPE_CHECKING, Container, Sequence, cast @@ -24,7 +25,7 @@ from flask import session, url_for from airflow.cli.cli_config import CLICommand, DefaultHelpParser, GroupCommand -from airflow.exceptions import AirflowOptionalProviderFeatureException +from airflow.exceptions import AirflowOptionalProviderFeatureException, AirflowProviderDeprecationWarning from airflow.providers.amazon.aws.auth_manager.avp.entities import AvpEntities from airflow.providers.amazon.aws.auth_manager.avp.facade import ( AwsAuthManagerAmazonVerifiedPermissionsFacade, @@ -166,6 +167,16 @@ def is_authorized_asset( method=method, entity_type=AvpEntities.ASSET, user=user or self.get_user(), entity_id=asset_uri ) + def is_authorized_dataset( + self, *, method: ResourceMethod, details: AssetDetails | None = None, user: BaseUser | None = None + ) -> bool: + warnings.warn( + "is_authorized_dataset will be renamed as is_authorized_asset in Airflow 3 and will be removed when the minimum Airflow version is set to 3.0 for the amazon provider", + AirflowProviderDeprecationWarning, + stacklevel=2, + ) + return self.is_authorized_asset(method=method, user=user) + def is_authorized_pool( self, *, method: ResourceMethod, details: PoolDetails | None = None, user: BaseUser | None = None ) -> bool: