mirror of
https://github.com/astral-sh/ruff.git
synced 2025-09-30 13:51:16 +00:00
[airflow
] Add external_task.{ExternalTaskMarker, ExternalTaskSensor}
for AIR302
(#16014)
## Summary Apply suggestions similar to https://github.com/astral-sh/ruff/pull/15922#discussion_r1940697704 ## Test Plan a test fixture has been updated
This commit is contained in:
parent
b1c61cb2ee
commit
618bfaf884
3 changed files with 589 additions and 607 deletions
|
@ -75,15 +75,10 @@ from airflow.secrets.local_filesystem import LocalFilesystemBackend, load_connec
|
|||
from airflow.security.permissions import RESOURCE_DATASET
|
||||
from airflow.sensors.base_sensor_operator import BaseSensorOperator
|
||||
from airflow.sensors.date_time_sensor import DateTimeSensor
|
||||
from airflow.sensors.external_task import (
|
||||
ExternalTaskSensorLink as ExternalTaskSensorLinkFromExternalTask,
|
||||
)
|
||||
from airflow.sensors.external_task_sensor import (
|
||||
ExternalTaskMarker,
|
||||
ExternalTaskSensor,
|
||||
)
|
||||
from airflow.sensors.external_task_sensor import (
|
||||
ExternalTaskSensorLink as ExternalTaskSensorLinkFromExternalTaskSensor,
|
||||
ExternalTaskSensorLink,
|
||||
)
|
||||
from airflow.sensors.time_delta_sensor import TimeDeltaSensor
|
||||
from airflow.timetables.datasets import DatasetOrTimeSchedule
|
||||
|
@ -249,11 +244,13 @@ BaseSensorOperator()
|
|||
DateTimeSensor()
|
||||
|
||||
# airflow.sensors.external_task
|
||||
ExternalTaskSensorLinkFromExternalTask()
|
||||
|
||||
# airflow.sensors.external_task_sensor
|
||||
ExternalTaskSensorLink()
|
||||
ExternalTaskMarker()
|
||||
ExternalTaskSensor()
|
||||
|
||||
# airflow.sensors.external_task_sensor
|
||||
ExternalTaskMarkerFromExternalTaskSensor()
|
||||
ExternalTaskSensorFromExternalTaskSensor()
|
||||
ExternalTaskSensorLinkFromExternalTaskSensor()
|
||||
|
||||
# airflow.sensors.time_delta_sensor
|
||||
|
|
|
@ -690,16 +690,7 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
|
|||
["airflow", "operators", "branch_operator", "BaseBranchOperator"] => {
|
||||
Replacement::Name("airflow.operators.branch.BaseBranchOperator")
|
||||
}
|
||||
["airflow", "operators", " dummy", "EmptyOperator"] => {
|
||||
Replacement::Name("airflow.operators.empty.EmptyOperator")
|
||||
}
|
||||
["airflow", "operators", "dummy", "DummyOperator"] => {
|
||||
Replacement::Name("airflow.operators.empty.EmptyOperator")
|
||||
}
|
||||
["airflow", "operators", "dummy_operator", "EmptyOperator"] => {
|
||||
Replacement::Name("airflow.operators.empty.EmptyOperator")
|
||||
}
|
||||
["airflow", "operators", "dummy_operator", "DummyOperator"] => {
|
||||
["airflow", "operators", "dummy" | "dummy_operator", "EmptyOperator" | "DummyOperator"] => {
|
||||
Replacement::Name("airflow.operators.empty.EmptyOperator")
|
||||
}
|
||||
["airflow", "operators", "email_operator", "EmailOperator"] => {
|
||||
|
@ -728,24 +719,21 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
|
|||
}
|
||||
|
||||
// airflow.sensors
|
||||
["airflow", "sensors", "external_task", "ExternalTaskSensorLink"] => {
|
||||
Replacement::Name("airflow.sensors.external_task.ExternalDagLink")
|
||||
}
|
||||
["airflow", "sensors", "base_sensor_operator", "BaseSensorOperator"] => {
|
||||
Replacement::Name("airflow.sensors.base.BaseSensorOperator")
|
||||
}
|
||||
["airflow", "sensors", "date_time_sensor", "DateTimeSensor"] => {
|
||||
Replacement::Name("airflow.sensors.date_time.DateTimeSensor")
|
||||
}
|
||||
["airflow", "sensors", "external_task_sensor", "ExternalTaskMarker"] => {
|
||||
["airflow", "sensors", "external_task" | "external_task_sensor", "ExternalTaskMarker"] => {
|
||||
Replacement::Name("airflow.sensors.external_task.ExternalTaskMarker")
|
||||
}
|
||||
["airflow", "sensors", "external_task_sensor", "ExternalTaskSensor"] => {
|
||||
Replacement::Name("airflow.sensors.external_task.ExternalTaskSensor")
|
||||
}
|
||||
["airflow", "sensors", "external_task_sensor", "ExternalTaskSensorLink"] => {
|
||||
["airflow", "sensors", "external_task" | "external_task_sensor", "ExternalTaskSensorLink"] => {
|
||||
Replacement::Name("airflow.sensors.external_task.ExternalDagLink")
|
||||
}
|
||||
["airflow", "sensors", "external_task" | "external_task_sensor", "ExternalTaskSensor"] => {
|
||||
Replacement::Name("airflow.sensors.external_task.ExternalTaskSensor")
|
||||
}
|
||||
["airflow", "sensors", "time_delta_sensor", "TimeDeltaSensor"] => {
|
||||
Replacement::Name("airflow.sensors.time_delta.TimeDeltaSensor")
|
||||
}
|
||||
|
@ -764,10 +752,9 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
|
|||
["airflow", "utils", "dates", "days_ago"] => {
|
||||
Replacement::Name("pendulum.today('UTC').add(days=-N, ...)")
|
||||
}
|
||||
["airflow", "utils", "dates", "parse_execution_date"] => Replacement::None,
|
||||
["airflow", "utils", "dates", "round_time"] => Replacement::None,
|
||||
["airflow", "utils", "dates", "scale_time_units"] => Replacement::None,
|
||||
["airflow", "utils", "dates", "infer_time_unit"] => Replacement::None,
|
||||
["airflow", "utils", "dates", "parse_execution_date" | "round_time" | "scale_time_units" | "infer_time_unit"] => {
|
||||
Replacement::None
|
||||
}
|
||||
|
||||
// airflow.utils.file
|
||||
["airflow", "utils", "file", "TemporaryDirectory"] => Replacement::None,
|
||||
|
@ -784,12 +771,10 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
|
|||
}
|
||||
|
||||
// airflow.utils.state
|
||||
["airflow", "utils", "state", "SHUTDOWN"] => Replacement::None,
|
||||
["airflow", "utils", "state", "terminating_states"] => Replacement::None,
|
||||
["airflow", "utils", "state", "SHUTDOWN" | "terminating_states"] => Replacement::None,
|
||||
|
||||
// airflow.utils.trigger_rule
|
||||
["airflow", "utils", "trigger_rule", "TriggerRule", "DUMMY"] => Replacement::None,
|
||||
["airflow", "utils", "trigger_rule", "TriggerRule", "NONE_FAILED_OR_SKIPPED"] => {
|
||||
["airflow", "utils", "trigger_rule", "TriggerRule", "DUMMY" | "NONE_FAILED_OR_SKIPPED"] => {
|
||||
Replacement::None
|
||||
}
|
||||
|
||||
|
|
File diff suppressed because it is too large
Load diff
Loading…
Add table
Add a link
Reference in a new issue