Skip to content

Commit 8480c78

Browse files
authored
Move DagIsPaused exception to standard provider (#49500)
* Move DagIsPaused exception to standard provider * remove from core
1 parent 6b6867d commit 8480c78

File tree

3 files changed

+13
-14
lines changed

3 files changed

+13
-14
lines changed

airflow-core/src/airflow/exceptions.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -246,17 +246,6 @@ class AirflowTimetableInvalid(AirflowException):
246246
"""Raise when a DAG has an invalid timetable."""
247247

248248

249-
class DagIsPaused(AirflowException):
250-
"""Raise when a dag is paused and something tries to run it."""
251-
252-
def __init__(self, dag_id: str) -> None:
253-
super().__init__(dag_id)
254-
self.dag_id = dag_id
255-
256-
def __str__(self) -> str:
257-
return f"Dag {self.dag_id} is paused"
258-
259-
260249
class DagNotFound(AirflowNotFoundException):
261250
"""Raise when a DAG is not available in the system."""
262251

providers/standard/src/airflow/providers/standard/operators/trigger_dagrun.py

Lines changed: 11 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -60,14 +60,24 @@
6060
from airflow.utils.context import Context
6161

6262
if AIRFLOW_V_3_0_PLUS:
63-
from airflow.exceptions import DagIsPaused
6463
from airflow.sdk import BaseOperatorLink
6564
from airflow.sdk.execution_time.xcom import XCom
6665
else:
6766
from airflow.models import XCom # type: ignore[no-redef]
6867
from airflow.models.baseoperatorlink import BaseOperatorLink # type: ignore[no-redef]
6968

7069

70+
class DagIsPaused(AirflowException):
71+
"""Raise when a dag is paused and something tries to run it."""
72+
73+
def __init__(self, dag_id: str) -> None:
74+
super().__init__(dag_id)
75+
self.dag_id = dag_id
76+
77+
def __str__(self) -> str:
78+
return f"Dag {self.dag_id} is paused"
79+
80+
7181
class TriggerDagRunLink(BaseOperatorLink):
7282
"""
7383
Operator link for TriggerDagRunOperator.

providers/standard/tests/unit/standard/operators/test_trigger_dagrun.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -30,7 +30,7 @@
3030
from airflow.models.dagrun import DagRun
3131
from airflow.models.log import Log
3232
from airflow.models.taskinstance import TaskInstance
33-
from airflow.providers.standard.operators.trigger_dagrun import TriggerDagRunOperator
33+
from airflow.providers.standard.operators.trigger_dagrun import DagIsPaused, TriggerDagRunOperator
3434
from airflow.providers.standard.triggers.external_task import DagStateTrigger
3535
from airflow.utils import timezone
3636
from airflow.utils.session import create_session
@@ -41,7 +41,7 @@
4141
from tests_common.test_utils.version_compat import AIRFLOW_V_3_0_PLUS
4242

4343
if AIRFLOW_V_3_0_PLUS:
44-
from airflow.exceptions import DagIsPaused, DagRunTriggerException
44+
from airflow.exceptions import DagRunTriggerException
4545

4646
pytestmark = pytest.mark.db_test
4747

0 commit comments

Comments
 (0)