@@ -451,68 +451,66 @@ fn check_method(checker: &Checker, call_expr: &ExprCall) {
451
451
452
452
let replacement = match qualname. segments ( ) {
453
453
[ "airflow" , "datasets" , "manager" , "DatasetManager" ] => match attr. as_str ( ) {
454
- "register_dataset_change" => Some ( Replacement :: Name ( "register_asset_change" ) ) ,
455
- "create_datasets" => Some ( Replacement :: Name ( "create_assets" ) ) ,
456
- "notify_dataset_created" => Some ( Replacement :: Name ( "notify_asset_created" ) ) ,
457
- "notify_dataset_changed" => Some ( Replacement :: Name ( "notify_asset_changed" ) ) ,
458
- "notify_dataset_alias_created" => Some ( Replacement :: Name ( "notify_asset_alias_created" ) ) ,
459
- _ => None ,
454
+ "register_dataset_change" => Replacement :: Name ( "register_asset_change" ) ,
455
+ "create_datasets" => Replacement :: Name ( "create_assets" ) ,
456
+ "notify_dataset_created" => Replacement :: Name ( "notify_asset_created" ) ,
457
+ "notify_dataset_changed" => Replacement :: Name ( "notify_asset_changed" ) ,
458
+ "notify_dataset_alias_created" => Replacement :: Name ( "notify_asset_alias_created" ) ,
459
+ _ => return ,
460
460
} ,
461
461
[ "airflow" , "lineage" , "hook" , "HookLineageCollector" ] => match attr. as_str ( ) {
462
- "create_dataset" => Some ( Replacement :: Name ( "create_asset" ) ) ,
463
- "add_input_dataset" => Some ( Replacement :: Name ( "add_input_asset" ) ) ,
464
- "add_output_dataset" => Some ( Replacement :: Name ( "add_output_asset" ) ) ,
465
- "collected_datasets" => Some ( Replacement :: Name ( "collected_assets" ) ) ,
466
- _ => None ,
462
+ "create_dataset" => Replacement :: Name ( "create_asset" ) ,
463
+ "add_input_dataset" => Replacement :: Name ( "add_input_asset" ) ,
464
+ "add_output_dataset" => Replacement :: Name ( "add_output_asset" ) ,
465
+ "collected_datasets" => Replacement :: Name ( "collected_assets" ) ,
466
+ _ => return ,
467
467
} ,
468
468
[ "airflow" , "providers" , "amazon" , "auth_manager" , "aws_auth_manager" , "AwsAuthManager" ] => {
469
469
match attr. as_str ( ) {
470
- "is_authorized_dataset" => Some ( Replacement :: Name ( "is_authorized_asset" ) ) ,
471
- _ => None ,
470
+ "is_authorized_dataset" => Replacement :: Name ( "is_authorized_asset" ) ,
471
+ _ => return ,
472
472
}
473
473
}
474
474
[ "airflow" , "providers_manager" , "ProvidersManager" ] => match attr. as_str ( ) {
475
- "initialize_providers_dataset_uri_resources" => Some ( Replacement :: Name (
476
- "initialize_providers_asset_uri_resources" ,
477
- ) ) ,
478
- _ => None ,
475
+ "initialize_providers_dataset_uri_resources" => {
476
+ Replacement :: Name ( "initialize_providers_asset_uri_resources" )
477
+ }
478
+ _ => return ,
479
479
} ,
480
480
[ "airflow" , "secrets" , "local_filesystem" , "LocalFilesystemBackend" ] => match attr. as_str ( )
481
481
{
482
- "get_connections" => Some ( Replacement :: Name ( "get_connection" ) ) ,
483
- _ => None ,
482
+ "get_connections" => Replacement :: Name ( "get_connection" ) ,
483
+ _ => return ,
484
484
} ,
485
485
[ "airflow" , "datasets" , ..] | [ "airflow" , "Dataset" ] => match attr. as_str ( ) {
486
- "iter_datasets" => Some ( Replacement :: Name ( "iter_assets" ) ) ,
487
- "iter_dataset_aliases" => Some ( Replacement :: Name ( "iter_asset_aliases" ) ) ,
488
- _ => None ,
486
+ "iter_datasets" => Replacement :: Name ( "iter_assets" ) ,
487
+ "iter_dataset_aliases" => Replacement :: Name ( "iter_asset_aliases" ) ,
488
+ _ => return ,
489
489
} ,
490
490
segments => {
491
491
if is_airflow_secret_backend ( segments) {
492
492
match attr. as_str ( ) {
493
- "get_conn_uri" => Some ( Replacement :: Name ( "get_conn_value" ) ) ,
494
- "get_connections" => Some ( Replacement :: Name ( "get_connection" ) ) ,
495
- _ => None ,
493
+ "get_conn_uri" => Replacement :: Name ( "get_conn_value" ) ,
494
+ "get_connections" => Replacement :: Name ( "get_connection" ) ,
495
+ _ => return ,
496
496
}
497
497
} else if is_airflow_hook ( segments) {
498
498
match attr. as_str ( ) {
499
- "get_connections" => Some ( Replacement :: Name ( "get_connection" ) ) ,
500
- _ => None ,
499
+ "get_connections" => Replacement :: Name ( "get_connection" ) ,
500
+ _ => return ,
501
501
}
502
502
} else {
503
- None
503
+ return ;
504
504
}
505
505
}
506
506
} ;
507
- if let Some ( replacement) = replacement {
508
- checker. report_diagnostic ( Diagnostic :: new (
509
- Airflow3Removal {
510
- deprecated : attr. to_string ( ) ,
511
- replacement,
512
- } ,
513
- attr. range ( ) ,
514
- ) ) ;
515
- }
507
+ checker. report_diagnostic ( Diagnostic :: new (
508
+ Airflow3Removal {
509
+ deprecated : attr. to_string ( ) ,
510
+ replacement,
511
+ } ,
512
+ attr. range ( ) ,
513
+ ) ) ;
516
514
}
517
515
518
516
/// Check whether a removed Airflow name is used.
0 commit comments