[airflow] Extend AIR311 rules (#17422)

<!--
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

Author: Lee-W

crates/ruff_linter/resources/test/fixtures/airflow/AIR311_names.py

@@ -9,11 +9,17 @@
     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
@@ -24,6 +30,20 @@
 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()
@@ -32,6 +52,8 @@
 # airflow.models.baseoperatolinker
 BaseOperatorLink()
 
+# airflow.models.dag
+DAGFromDag()
 # airflow.timetables.datasets
 DatasetOrTimeSchedule()
 

crates/ruff_linter/src/rules/airflow/rules/suggested_to_update_3_0.rs

@@ -210,6 +210,25 @@ fn check_name(checker: &Checker, expr: &Expr, range: TextRange) {
             "expand_alias_to_datasets" => Replacement::Name("airflow.sdk.expand_alias_to_assets"),
             _ => return,
         },
+
+        // airflow.decorators
+        ["airflow", "decorators", rest @ ("dag" | "task" | "task_group" | "setup" | "teardown")] => {
+            Replacement::SourceModuleMoved {
+                module: "airflow.sdk",
+                name: (*rest).to_string(),
+            }
+        }
+
+        // airflow.io
+        ["airflow", "io", "path", "ObjectStoragePath"] => Replacement::SourceModuleMoved {
+            module: "airflow.sdk",
+            name: "ObjectStoragePath".to_string(),
+        },
+        ["airflow", "io", "storage", "attach"] => Replacement::SourceModuleMoved {
+            module: "airflow.sdk.io",
+            name: "attach".to_string(),
+        },
+
         // airflow.models.baseoperator
         ["airflow", "models", "baseoperator", rest] => match *rest {
             "chain" | "chain_linear" | "cross_downstream" => Replacement::SourceModuleMoved {
@@ -221,6 +240,11 @@ fn check_name(checker: &Checker, expr: &Expr, range: TextRange) {
             }
             _ => return,
         },
+        // airflow.model..DAG
+        ["airflow", "models", .., "DAG"] => Replacement::SourceModuleMoved {
+            module: "airflow.sdk",
+            name: "DAG".to_string(),
+        },
         // airflow.timetables
         ["airflow", "timetables", "datasets", "DatasetOrTimeSchedule"] => {
             Replacement::Name("airflow.timetables.assets.AssetOrTimeSchedule")