Skip to content

Commit 659d94f

Browse files
TaragolisTaragolis
authored
Use airflow.exceptions.AirflowException in providers (#34511)
* Use airflow.exceptions.AirflowException in providers * Use airflow.exceptions.AirflowException in providers system tests * Use airflow.exceptions.AirflowException in providers tests
1 parent fc0dd62 commit 659d94f

File tree

82 files changed

+83
-88
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

82 files changed

+83
-88
lines changed

‎airflow/providers/amazon/aws/hooks/quicksight.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
from botocore.exceptions import ClientError
2424

25-
from airflow import AirflowException
25+
from airflow.exceptions import AirflowException
2626
from airflow.providers.amazon.aws.hooks.base_aws import AwsBaseHook
2727
from airflow.providers.amazon.aws.hooks.sts import StsHook
2828

‎airflow/providers/amazon/aws/operators/athena.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
from functools import cached_property
2121
from typing import TYPE_CHECKING, Any, Sequence
2222

23-
from airflow import AirflowException
2423
from airflow.configuration import conf
24+
from airflow.exceptions import AirflowException
2525
from airflow.models import BaseOperator
2626
from airflow.providers.amazon.aws.hooks.athena import AthenaHook
2727
from airflow.providers.amazon.aws.triggers.athena import AthenaTrigger

‎airflow/providers/amazon/aws/operators/eventbridge.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -19,7 +19,7 @@
1919
from functools import cached_property
2020
from typing import TYPE_CHECKING, Sequence
2121

22-
from airflow import AirflowException
22+
from airflow.exceptions import AirflowException
2323
from airflow.models import BaseOperator
2424
from airflow.providers.amazon.aws.hooks.eventbridge import EventBridgeHook
2525
from airflow.utils.helpers import prune_dict

‎airflow/providers/amazon/aws/operators/glue.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@
2222
from functools import cached_property
2323
from typing import TYPE_CHECKING, Sequence
2424

25-
from airflow import AirflowException
2625
from airflow.configuration import conf
26+
from airflow.exceptions import AirflowException
2727
from airflow.models import BaseOperator
2828
from airflow.providers.amazon.aws.hooks.glue import GlueJobHook
2929
from airflow.providers.amazon.aws.hooks.s3 import S3Hook

‎airflow/providers/amazon/aws/operators/glue_crawler.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,8 +20,8 @@
2020
from functools import cached_property
2121
from typing import TYPE_CHECKING, Sequence
2222

23-
from airflow import AirflowException
2423
from airflow.configuration import conf
24+
from airflow.exceptions import AirflowException
2525
from airflow.providers.amazon.aws.triggers.glue_crawler import GlueCrawlerCompleteTrigger
2626

2727
if TYPE_CHECKING:

‎airflow/providers/amazon/aws/operators/lambda_function.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,8 +22,8 @@
2222
from functools import cached_property
2323
from typing import TYPE_CHECKING, Any, Sequence
2424

25-
from airflow import AirflowException
2625
from airflow.configuration import conf
26+
from airflow.exceptions import AirflowException
2727
from airflow.models import BaseOperator
2828
from airflow.providers.amazon.aws.hooks.lambda_function import LambdaHook
2929
from airflow.providers.amazon.aws.triggers.lambda_function import LambdaCreateFunctionCompleteTrigger

‎airflow/providers/amazon/aws/triggers/ecs.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,7 @@
2222

2323
from botocore.exceptions import ClientError, WaiterError
2424

25-
from airflow import AirflowException
25+
from airflow.exceptions import AirflowException
2626
from airflow.providers.amazon.aws.hooks.ecs import EcsHook
2727
from airflow.providers.amazon.aws.hooks.logs import AwsLogsHook
2828
from airflow.providers.amazon.aws.triggers.base import AwsBaseWaiterTrigger

‎airflow/providers/amazon/aws/triggers/sagemaker.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -25,7 +25,7 @@
2525

2626
from botocore.exceptions import WaiterError
2727

28-
from airflow import AirflowException
28+
from airflow.exceptions import AirflowException
2929
from airflow.providers.amazon.aws.hooks.sagemaker import SageMakerHook
3030
from airflow.providers.amazon.aws.utils.waiter_with_logging import async_wait
3131
from airflow.triggers.base import BaseTrigger, TriggerEvent

‎airflow/providers/apache/beam/operators/beam.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -30,8 +30,8 @@
3030
from functools import partial
3131
from typing import IO, TYPE_CHECKING, Any, Callable, Sequence
3232

33-
from airflow import AirflowException
3433
from airflow.configuration import conf
34+
from airflow.exceptions import AirflowException
3535
from airflow.models import BaseOperator
3636
from airflow.providers.apache.beam.hooks.beam import BeamHook, BeamRunnerType
3737
from airflow.providers.apache.beam.triggers.beam import BeamPipelineTrigger

‎airflow/providers/apache/hdfs/hooks/hdfs.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -16,7 +16,7 @@
1616
# under the License.
1717
from __future__ import annotations
1818

19-
from airflow import AirflowException
19+
from airflow.exceptions import AirflowException
2020
from airflow.hooks.base import BaseHook
2121

2222
_EXCEPTION_MESSAGE = """The old HDFS Hooks have been removed in 4.0.0 version of the apache.hdfs provider.

0 commit comments

Comments
 (0)