mirror of
https://github.com/astral-sh/ruff.git
synced 2025-10-27 02:16:54 +00:00
<!-- Thank you for contributing to Ruff/ty! To help us out with reviewing, please consider the following: - Does this pull request include a summary of the change? (See below.) - Does this pull request include a descriptive title? (Please prefix with `[ty]` for ty pull requests.) - Does this pull request include references to any relevant issues? --> ## Summary <!-- What's the purpose of the change? What does it do, and why? --> Extend the following rules. ### AIR311 * `airflow.sensors.base.BaseSensorOperator` → airflow.sdk.bases.sensor.BaseSensorOperator` * `airflow.sensors.base.PokeReturnValue` → airflow.sdk.bases.sensor.PokeReturnValue` * `airflow.sensors.base.poke_mode_only` → airflow.sdk.bases.sensor.poke_mode_only` * `airflow.decorators.base.DecoratedOperator` → airflow.sdk.bases.decorator.DecoratedOperator` * `airflow.models.param.Param` → airflow.sdk.definitions.param.Param` * `airflow.decorators.base.DecoratedMappedOperator` → `airflow.sdk.bases.decorator.DecoratedMappedOperator` * `airflow.decorators.base.DecoratedOperator` → `airflow.sdk.bases.decorator.DecoratedOperator` * `airflow.decorators.base.TaskDecorator` → `airflow.sdk.bases.decorator.TaskDecorator` * `airflow.decorators.base.get_unique_task_id` → `airflow.sdk.bases.decorator.get_unique_task_id` * `airflow.decorators.base.task_decorator_factory` → `airflow.sdk.bases.decorator.task_decorator_factory` ### AIR312 * `airflow.sensors.bash.BashSensor` → `airflow.providers.standard.sensor.bash.BashSensor` * `airflow.sensors.python.PythonSensor` → `airflow.providers.standard.sensors.python.PythonSensor` ## Test Plan <!-- How was it tested? --> update the test fixture accordingly in the second commit and reorg in the third
89 lines
2.1 KiB
Python
89 lines
2.1 KiB
Python
from __future__ import annotations
|
|
|
|
from airflow.hooks.filesystem import FSHook
|
|
from airflow.hooks.package_index import PackageIndexHook
|
|
from airflow.hooks.subprocess import SubprocessHook
|
|
from airflow.operators.bash import BashOperator
|
|
from airflow.operators.datetime import BranchDateTimeOperator
|
|
from airflow.operators.empty import EmptyOperator
|
|
from airflow.operators.latest_only import LatestOnlyOperator
|
|
from airflow.operators.trigger_dagrun import TriggerDagRunOperator
|
|
from airflow.operators.weekday import BranchDayOfWeekOperator
|
|
|
|
FSHook()
|
|
PackageIndexHook()
|
|
SubprocessHook()
|
|
|
|
BashOperator()
|
|
BranchDateTimeOperator()
|
|
TriggerDagRunOperator()
|
|
EmptyOperator()
|
|
|
|
LatestOnlyOperator()
|
|
BranchDayOfWeekOperator()
|
|
|
|
from airflow.operators.python import (
|
|
BranchPythonOperator,
|
|
PythonOperator,
|
|
PythonVirtualenvOperator,
|
|
ShortCircuitOperator,
|
|
)
|
|
from airflow.sensors.bash import BashSensor
|
|
from airflow.sensors.date_time import DateTimeSensor
|
|
|
|
BranchPythonOperator()
|
|
PythonOperator()
|
|
PythonVirtualenvOperator()
|
|
ShortCircuitOperator()
|
|
|
|
BashSensor()
|
|
DateTimeSensor()
|
|
from airflow.sensors.date_time import DateTimeSensorAsync
|
|
from airflow.sensors.external_task import (
|
|
ExternalTaskMarker,
|
|
ExternalTaskSensor,
|
|
)
|
|
from airflow.sensors.filesystem import FileSensor
|
|
from airflow.sensors.python import PythonSensor
|
|
|
|
BranchPythonOperator()
|
|
PythonOperator()
|
|
PythonVirtualenvOperator()
|
|
ShortCircuitOperator()
|
|
DateTimeSensorAsync()
|
|
ExternalTaskMarker()
|
|
ExternalTaskSensor()
|
|
FileSensor()
|
|
PythonSensor()
|
|
|
|
from airflow.sensors.time_sensor import (
|
|
TimeSensor,
|
|
TimeSensorAsync,
|
|
)
|
|
|
|
TimeSensor()
|
|
TimeSensorAsync()
|
|
|
|
from airflow.sensors.time_delta import (
|
|
TimeDeltaSensor,
|
|
TimeDeltaSensorAsync,
|
|
)
|
|
from airflow.sensors.weekday import DayOfWeekSensor
|
|
from airflow.triggers.external_task import (
|
|
DagStateTrigger,
|
|
WorkflowTrigger,
|
|
)
|
|
from airflow.triggers.file import FileTrigger
|
|
from airflow.triggers.temporal import (
|
|
DateTimeTrigger,
|
|
TimeDeltaTrigger,
|
|
)
|
|
|
|
TimeDeltaSensor()
|
|
TimeDeltaSensorAsync()
|
|
DayOfWeekSensor()
|
|
DagStateTrigger()
|
|
WorkflowTrigger()
|
|
FileTrigger()
|
|
DateTimeTrigger()
|
|
TimeDeltaTrigger()
|