Skip to content

Commit b1eb36b

Browse files
committed
feat(AIR303): initial air303
1 parent edce559 commit b1eb36b

File tree

2 files changed

+16
-0
lines changed

2 files changed

+16
-0
lines changed

crates/ruff_linter/src/checkers/ast/analyze/expression.rs

+3
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,9 @@ pub(crate) fn expression(expr: &Expr, checker: &mut Checker) {
223223
if checker.enabled(Rule::Airflow3Removal) {
224224
airflow::rules::removed_in_3(checker, expr);
225225
}
226+
if checker.enabled(Rule::Airflow3Removal) {
227+
airflow::rules::moved_to_provider_in_3(checker, expr);
228+
}
226229

227230
// Ex) List[...]
228231
if checker.any_enabled(&[

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

+13
Original file line numberDiff line numberDiff line change
@@ -204,3 +204,16 @@ pub(crate) fn removed_in_3(checker: &mut Checker, expr: &Expr) {
204204
_ => {}
205205
}
206206
}
207+
208+
/// AIR303
209+
pub(crate) fn moved_to_provider_in_3(checker: &mut Checker, expr: &Expr) {
210+
if !checker.semantic().seen_module(Modules::AIRFLOW) {
211+
return;
212+
}
213+
214+
match expr {
215+
Expr::Attribute(ExprAttribute { attr: ranged, .. }) => removed_name(checker, expr, ranged),
216+
ranged @ Expr::Name(_) => removed_name(checker, expr, ranged),
217+
_ => {}
218+
}
219+
}

0 commit comments

Comments
 (0)