mirror of https://github.com/astral-sh/ruff
[airflow] `BashOperator` has been moved to `airflow.providers.standard.operators.bash.BashOperator` (AIR302) (#15922)
## Summary Extend AIR302 with * `airflow.operators.bash.BashOperator → airflow.providers.standard.operators.bash.BashOperator` * change existing rules `airflow.operators.bash_operator.BashOperator → airflow.operators.bash.BashOperator` to `airflow.operators.bash_operator.BashOperator → airflow.providers.standard.operators.bash.BashOperator` ## Test Plan a test fixture has been updated
This commit is contained in:
parent
ba02294af3
commit
de4d9979eb
|
|
@ -1,3 +1,5 @@
|
||||||
|
from __future__ import annotations
|
||||||
|
|
||||||
from airflow import (
|
from airflow import (
|
||||||
PY36,
|
PY36,
|
||||||
PY37,
|
PY37,
|
||||||
|
|
@ -6,6 +8,8 @@ from airflow import (
|
||||||
PY310,
|
PY310,
|
||||||
PY311,
|
PY311,
|
||||||
PY312,
|
PY312,
|
||||||
|
)
|
||||||
|
from airflow import (
|
||||||
Dataset as DatasetFromRoot,
|
Dataset as DatasetFromRoot,
|
||||||
)
|
)
|
||||||
from airflow.api_connexion.security import requires_access, requires_access_dataset
|
from airflow.api_connexion.security import requires_access, requires_access_dataset
|
||||||
|
|
@ -41,7 +45,8 @@ from airflow.lineage.hook import DatasetLineageInfo
|
||||||
from airflow.listeners.spec.dataset import on_dataset_changed, on_dataset_created
|
from airflow.listeners.spec.dataset import on_dataset_changed, on_dataset_created
|
||||||
from airflow.metrics.validators import AllowListValidator, BlockListValidator
|
from airflow.metrics.validators import AllowListValidator, BlockListValidator
|
||||||
from airflow.operators import dummy_operator
|
from airflow.operators import dummy_operator
|
||||||
from airflow.operators.bash_operator import BashOperator
|
from airflow.operators.bash import BashOperator
|
||||||
|
from airflow.operators.bash_operator import BashOperator as LegacyBashOperator
|
||||||
from airflow.operators.branch_operator import BaseBranchOperator
|
from airflow.operators.branch_operator import BaseBranchOperator
|
||||||
from airflow.operators.dagrun_operator import TriggerDagRunLink, TriggerDagRunOperator
|
from airflow.operators.dagrun_operator import TriggerDagRunLink, TriggerDagRunOperator
|
||||||
from airflow.operators.dummy import DummyOperator, EmptyOperator
|
from airflow.operators.dummy import DummyOperator, EmptyOperator
|
||||||
|
|
@ -76,6 +81,8 @@ from airflow.sensors.external_task import (
|
||||||
from airflow.sensors.external_task_sensor import (
|
from airflow.sensors.external_task_sensor import (
|
||||||
ExternalTaskMarker,
|
ExternalTaskMarker,
|
||||||
ExternalTaskSensor,
|
ExternalTaskSensor,
|
||||||
|
)
|
||||||
|
from airflow.sensors.external_task_sensor import (
|
||||||
ExternalTaskSensorLink as ExternalTaskSensorLinkFromExternalTaskSensor,
|
ExternalTaskSensorLink as ExternalTaskSensorLinkFromExternalTaskSensor,
|
||||||
)
|
)
|
||||||
from airflow.sensors.time_delta_sensor import TimeDeltaSensor
|
from airflow.sensors.time_delta_sensor import TimeDeltaSensor
|
||||||
|
|
@ -164,8 +171,9 @@ AllowListValidator(), BlockListValidator()
|
||||||
dummy_operator.EmptyOperator()
|
dummy_operator.EmptyOperator()
|
||||||
dummy_operator.DummyOperator()
|
dummy_operator.DummyOperator()
|
||||||
|
|
||||||
# airflow.operators.bash_operator
|
# airflow.operators.bash / airflow.operators.bash_operator
|
||||||
BashOperator()
|
BashOperator()
|
||||||
|
LegacyBashOperator()
|
||||||
|
|
||||||
# airflow.operators.branch_operator
|
# airflow.operators.branch_operator
|
||||||
BaseBranchOperator()
|
BaseBranchOperator()
|
||||||
|
|
|
||||||
|
|
@ -684,8 +684,8 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
|
||||||
["airflow", "operators", "subdag", ..] => {
|
["airflow", "operators", "subdag", ..] => {
|
||||||
Replacement::Message("The whole `airflow.subdag` module has been removed.")
|
Replacement::Message("The whole `airflow.subdag` module has been removed.")
|
||||||
}
|
}
|
||||||
["airflow", "operators", "bash_operator", "BashOperator"] => {
|
["airflow", "operators", "bash" | "bash_operator", "BashOperator"] => {
|
||||||
Replacement::Name("airflow.operators.bash.BashOperator")
|
Replacement::Name("airflow.providers.standard.operators.bash.BashOperator")
|
||||||
}
|
}
|
||||||
["airflow", "operators", "branch_operator", "BaseBranchOperator"] => {
|
["airflow", "operators", "branch_operator", "BaseBranchOperator"] => {
|
||||||
Replacement::Name("airflow.operators.branch.BaseBranchOperator")
|
Replacement::Name("airflow.operators.branch.BaseBranchOperator")
|
||||||
|
|
|
||||||
File diff suppressed because it is too large
Load Diff
Loading…
Reference in New Issue