Skip to content

Commit cbdfb3f

Browse files
committed
fixup! refactor(AIR302): combine similiar cases
1 parent 95e8efc commit cbdfb3f

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

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

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -62,6 +62,7 @@ impl Violation for Airflow3MovedToProvider {
6262
fn fix_title(&self) -> Option<String> {
6363
let Airflow3MovedToProvider { replacement, .. } = self;
6464
match replacement {
65+
ProviderReplacement::None => {None}
6566
ProviderReplacement::ProviderName {
6667
name,
6768
provider,
@@ -79,7 +80,6 @@ impl Violation for Airflow3MovedToProvider {
7980
} => {
8081
Some(format!("Install `apache-airflow-provider-{provider}>={version}` and use `{module}.{name}` instead."))
8182
} ,
82-
_ => None,
8383
}
8484
}
8585
}
@@ -188,7 +188,7 @@ fn check_names_moved_to_provider(checker: &Checker, expr: &Expr, ranged: TextRan
188188
version: "1.0.0"
189189
},
190190
["airflow", "operators", "check_operator", "CheckOperator" | "SQLCheckOperator"]
191-
| ["airflow", "operators", "druid_check_operator", "DruidCheckOperator"]
191+
| ["airflow", "operators", "druid_check_operator", "DruidCheckOperator"]
192192
| ["airflow", "operators", "presto_check_operator", "SQLCheckOperator" | "PrestoCheckOperator"]
193193
| ["airflow", "operators", "sql", "SQLCheckOperator"] => ProviderReplacement::ProviderName {
194194
name: "airflow.providers.common.sql.operators.sql.SQLCheckOperator",
@@ -239,7 +239,7 @@ fn check_names_moved_to_provider(checker: &Checker, expr: &Expr, ranged: TextRan
239239
provider: "common-sql",
240240
version: "1.0.0"
241241
},
242-
["airflow", "operators", "jdbc_operator", "JdbcOperator"]
242+
["airflow", "operators", "jdbc_operator", "JdbcOperator"]
243243
| ["airflow", "operators", "mssql_operator", "MsSqlOperator"]
244244
| ["airflow", "operators", "mysql_operator", "MySqlOperator"]
245245
| ["airflow", "operators", "oracle_operator", "OracleOperator"]

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

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -65,6 +65,7 @@ impl Violation for Airflow3SuggestedToMoveToProvider {
6565
fn fix_title(&self) -> Option<String> {
6666
let Airflow3SuggestedToMoveToProvider { replacement, .. } = self;
6767
match replacement {
68+
ProviderReplacement::None => {None}
6869
ProviderReplacement::ProviderName {
6970
name,
7071
provider,
@@ -82,7 +83,6 @@ impl Violation for Airflow3SuggestedToMoveToProvider {
8283
} => {
8384
Some(format!("Install `apache-airflow-provider-{provider}>={version}` and use `{module}.{name}` instead."))
8485
},
85-
_ => None,
8686
}
8787
}
8888
}

0 commit comments

Comments
 (0)