mirror of
https://github.com/astral-sh/ruff.git
synced 2025-10-01 06:11:43 +00:00
1a79722ee0
<!--
Thank you for contributing to Ruff! 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?
- 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 AIR311 rules
* `airflow.io.path.ObjectStoragePath` → `airflow.sdk.ObjectStoragePath`
* `airflow.io.storage.attach` → `airflow.sdk.io.attach`
* `airflow.models.dag.DAG` → `airflow.sdk.DAG`
* `airflow.models.DAG` → `airflow.sdk.DAG`
* `airflow.decorators.dag` → `airflow.sdk.dag`
* `airflow.decorators.task` → `airflow.sdk.task`
* `airflow.decorators.task_group` → `airflow.sdk.task_group`
* `airflow.decorators.setup` → `airflow.sdk.setup`
* `airflow.decorators.teardown` → `airflow.sdk.teardown`
## Test Plan
<!-- How was it tested? -->
The test case has been added to the button of the existing test
fixtures, confirmed to be correct and later reorgnaized
61 lines
1.3 KiB
Python
61 lines
1.3 KiB
Python
from __future__ import annotations
|
|
|
|
from airflow import Dataset as DatasetFromRoot
|
|
from airflow.datasets import (
|
|
Dataset,
|
|
DatasetAlias,
|
|
DatasetAll,
|
|
DatasetAny,
|
|
expand_alias_to_datasets,
|
|
)
|
|
from airflow.datasets.metadata import Metadata
|
|
from airflow.decorators import dag, setup, task, task_group, teardown
|
|
from airflow.io.path import ObjectStoragePath
|
|
from airflow.io.storage import attach
|
|
from airflow.models import DAG as DAGFromModel
|
|
from airflow.models.baseoperator import chain, chain_linear, cross_downstream
|
|
from airflow.models.baseoperatorlink import BaseOperatorLink
|
|
from airflow.models.dag import DAG as DAGFromDag
|
|
from airflow.timetables.datasets import DatasetOrTimeSchedule
|
|
from airflow.utils.dag_parsing_context import get_parsing_context
|
|
|
|
# airflow
|
|
DatasetFromRoot()
|
|
|
|
# airflow.datasets
|
|
Dataset()
|
|
DatasetAlias()
|
|
DatasetAll()
|
|
DatasetAny()
|
|
Metadata()
|
|
expand_alias_to_datasets()
|
|
|
|
# airflow.decorators
|
|
dag()
|
|
task()
|
|
task_group()
|
|
setup()
|
|
teardown()
|
|
|
|
# airflow.io
|
|
ObjectStoragePath()
|
|
attach()
|
|
|
|
# airflow.models
|
|
DAGFromModel()
|
|
|
|
# airflow.models.baseoperator
|
|
chain()
|
|
chain_linear()
|
|
cross_downstream()
|
|
|
|
# airflow.models.baseoperatolinker
|
|
BaseOperatorLink()
|
|
|
|
# airflow.models.dag
|
|
DAGFromDag()
|
|
# airflow.timetables.datasets
|
|
DatasetOrTimeSchedule()
|
|
|
|
# airflow.utils.dag_parsing_context
|
|
get_parsing_context()
|