@@ -690,16 +690,7 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
690
690
[ "airflow" , "operators" , "branch_operator" , "BaseBranchOperator" ] => {
691
691
Replacement :: Name ( "airflow.operators.branch.BaseBranchOperator" )
692
692
}
693
- [ "airflow" , "operators" , " dummy" , "EmptyOperator" ] => {
694
- Replacement :: Name ( "airflow.operators.empty.EmptyOperator" )
695
- }
696
- [ "airflow" , "operators" , "dummy" , "DummyOperator" ] => {
697
- Replacement :: Name ( "airflow.operators.empty.EmptyOperator" )
698
- }
699
- [ "airflow" , "operators" , "dummy_operator" , "EmptyOperator" ] => {
700
- Replacement :: Name ( "airflow.operators.empty.EmptyOperator" )
701
- }
702
- [ "airflow" , "operators" , "dummy_operator" , "DummyOperator" ] => {
693
+ [ "airflow" , "operators" , "dummy" | "dummy_operator" , "EmptyOperator" | "DummyOperator" ] => {
703
694
Replacement :: Name ( "airflow.operators.empty.EmptyOperator" )
704
695
}
705
696
[ "airflow" , "operators" , "email_operator" , "EmailOperator" ] => {
@@ -728,24 +719,21 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
728
719
}
729
720
730
721
// airflow.sensors
731
- [ "airflow" , "sensors" , "external_task" , "ExternalTaskSensorLink" ] => {
732
- Replacement :: Name ( "airflow.sensors.external_task.ExternalDagLink" )
733
- }
734
722
[ "airflow" , "sensors" , "base_sensor_operator" , "BaseSensorOperator" ] => {
735
723
Replacement :: Name ( "airflow.sensors.base.BaseSensorOperator" )
736
724
}
737
725
[ "airflow" , "sensors" , "date_time_sensor" , "DateTimeSensor" ] => {
738
726
Replacement :: Name ( "airflow.sensors.date_time.DateTimeSensor" )
739
727
}
740
- [ "airflow" , "sensors" , "external_task_sensor" , "ExternalTaskMarker" ] => {
728
+ [ "airflow" , "sensors" , "external_task" | " external_task_sensor", "ExternalTaskMarker" ] => {
741
729
Replacement :: Name ( "airflow.sensors.external_task.ExternalTaskMarker" )
742
730
}
743
- [ "airflow" , "sensors" , "external_task_sensor" , "ExternalTaskSensor" ] => {
744
- Replacement :: Name ( "airflow.sensors.external_task.ExternalTaskSensor" )
745
- }
746
- [ "airflow" , "sensors" , "external_task_sensor" , "ExternalTaskSensorLink" ] => {
731
+ [ "airflow" , "sensors" , "external_task" | "external_task_sensor" , "ExternalTaskSensorLink" ] => {
747
732
Replacement :: Name ( "airflow.sensors.external_task.ExternalDagLink" )
748
733
}
734
+ [ "airflow" , "sensors" , "external_task" | "external_task_sensor" , "ExternalTaskSensor" ] => {
735
+ Replacement :: Name ( "airflow.sensors.external_task.ExternalTaskSensor" )
736
+ }
749
737
[ "airflow" , "sensors" , "time_delta_sensor" , "TimeDeltaSensor" ] => {
750
738
Replacement :: Name ( "airflow.sensors.time_delta.TimeDeltaSensor" )
751
739
}
@@ -764,10 +752,9 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
764
752
[ "airflow" , "utils" , "dates" , "days_ago" ] => {
765
753
Replacement :: Name ( "pendulum.today('UTC').add(days=-N, ...)" )
766
754
}
767
- [ "airflow" , "utils" , "dates" , "parse_execution_date" ] => Replacement :: None ,
768
- [ "airflow" , "utils" , "dates" , "round_time" ] => Replacement :: None ,
769
- [ "airflow" , "utils" , "dates" , "scale_time_units" ] => Replacement :: None ,
770
- [ "airflow" , "utils" , "dates" , "infer_time_unit" ] => Replacement :: None ,
755
+ [ "airflow" , "utils" , "dates" , "parse_execution_date" | "round_time" | "scale_time_units" | "infer_time_unit" ] => {
756
+ Replacement :: None
757
+ }
771
758
772
759
// airflow.utils.file
773
760
[ "airflow" , "utils" , "file" , "TemporaryDirectory" ] => Replacement :: None ,
@@ -784,12 +771,10 @@ fn check_name(checker: &mut Checker, expr: &Expr, range: TextRange) {
784
771
}
785
772
786
773
// airflow.utils.state
787
- [ "airflow" , "utils" , "state" , "SHUTDOWN" ] => Replacement :: None ,
788
- [ "airflow" , "utils" , "state" , "terminating_states" ] => Replacement :: None ,
774
+ [ "airflow" , "utils" , "state" , "SHUTDOWN" | "terminating_states" ] => Replacement :: None ,
789
775
790
776
// airflow.utils.trigger_rule
791
- [ "airflow" , "utils" , "trigger_rule" , "TriggerRule" , "DUMMY" ] => Replacement :: None ,
792
- [ "airflow" , "utils" , "trigger_rule" , "TriggerRule" , "NONE_FAILED_OR_SKIPPED" ] => {
777
+ [ "airflow" , "utils" , "trigger_rule" , "TriggerRule" , "DUMMY" | "NONE_FAILED_OR_SKIPPED" ] => {
793
778
Replacement :: None
794
779
}
795
780
0 commit comments