[airflow]: extend names moved from core to provider (AIR303) (#15196)

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

@@ -6,38 +6,268 @@ from airflow.auth.managers.fab.api.auth.backend import (
 from airflow.auth.managers.fab.fab_auth_manager import FabAuthManager
 from airflow.auth.managers.fab.security_manager import override as fab_override
 from airflow.config_templates.default_celery import DEFAULT_CELERY_CONFIG
-from airflow.executors.celery_executor import app
+from airflow.executors.celery_executor import CeleryExecutor, app
+from airflow.executors.celery_kubernetes_executor import CeleryKubernetesExecutor
+from airflow.executors.dask_executor import DaskExecutor
 from airflow.executors.kubernetes_executor_types import (
     ALL_NAMESPACES,
     POD_EXECUTOR_DONE_KEY,
 )
 from airflow.hooks.dbapi import ConnectorProtocol, DbApiHook
-from airflow.hooks.hive_hooks import HIVE_QUEUE_PRIORITIES
+from airflow.hooks.dbapi_hook import DbApiHook as DbApiHook2
+from airflow.hooks.docker_hook import DockerHook
+from airflow.hooks.druid_hook import DruidDbApiHook, DruidHook
+from airflow.hooks.hive_hooks import (
+    HIVE_QUEUE_PRIORITIES,
+    HiveCliHook,
+    HiveMetastoreHook,
+    HiveServer2Hook,
+)
+from airflow.hooks.http_hook import HttpHook
+from airflow.hooks.jdbc_hook import JdbcHook, jaydebeapi
+from airflow.hooks.mssql_hook import MsSqlHook
+from airflow.hooks.mysql_hook import MySqlHook
+from airflow.hooks.oracle_hook import OracleHook
+from airflow.hooks.pig_hook import PigCliHook
+from airflow.hooks.postgres_hook import PostgresHook
+from airflow.hooks.presto_hook import PrestoHook
+from airflow.hooks.S3_hook import S3Hook, provide_bucket_name
+from airflow.hooks.samba_hook import SambaHook
+from airflow.hooks.slack_hook import SlackHook
+from airflow.hooks.sqlite_hook import SqliteHook
+from airflow.hooks.webhdfs_hook import WebHDFSHook
+from airflow.hooks.zendesk_hook import ZendeskHook
 from airflow.macros.hive import closest_ds_partition, max_partition
+from airflow.operators.check_operator import (
+    CheckOperator,
+    IntervalCheckOperator,
+    SQLCheckOperator,
+    SQLIntervalCheckOperator,
+    SQLThresholdCheckOperator,
+    SQLValueCheckOperator,
+    ThresholdCheckOperator,
+    ValueCheckOperator,
+)
+from airflow.operators.docker_operator import DockerOperator
+from airflow.operators.druid_check_operator import DruidCheckOperator
+from airflow.operators.gcs_to_s3 import GCSToS3Operator
+from airflow.operators.google_api_to_s3_transfer import (
+    GoogleApiToS3Operator,
+    GoogleApiToS3Transfer,
+)
+from airflow.operators.hive_operator import HiveOperator
+from airflow.operators.hive_stats_operator import HiveStatsCollectionOperator
+from airflow.operators.hive_to_druid import HiveToDruidOperator, HiveToDruidTransfer
+from airflow.operators.hive_to_mysql import HiveToMySqlOperator, HiveToMySqlTransfer
+from airflow.operators.hive_to_samba_operator import HiveToSambaOperator
+from airflow.operators.http_operator import SimpleHttpOperator
+from airflow.operators.jdbc_operator import JdbcOperator
+from airflow.operators.mssql_operator import MsSqlOperator
+from airflow.operators.mssql_to_hive import MsSqlToHiveOperator, MsSqlToHiveTransfer
+from airflow.operators.mysql_operator import MySqlOperator
+from airflow.operators.mysql_to_hive import MySqlToHiveOperator, MySqlToHiveTransfer
+from airflow.operators.oracle_operator import OracleOperator
+from airflow.operators.papermill_operator import PapermillOperator
+from airflow.operators.pig_operator import PigOperator
+from airflow.operators.postgres_operator import Mapping, PostgresOperator
+from airflow.operators.presto_check_operator import (
+    PrestoCheckOperator,
+    PrestoIntervalCheckOperator,
+    PrestoValueCheckOperator,
+    SQLCheckOperator as SQLCheckOperator2,
+    SQLIntervalCheckOperator as SQLIntervalCheckOperator2,
+    SQLValueCheckOperator as SQLValueCheckOperator2,
+)
+from airflow.operators.presto_to_mysql import (
+    PrestoToMySqlOperator,
+    PrestoToMySqlTransfer,
+)
+from airflow.operators.redshift_to_s3_operator import (
+    RedshiftToS3Operator,
+    RedshiftToS3Transfer,
+)
+from airflow.operators.s3_file_transform_operator import S3FileTransformOperator
+from airflow.operators.s3_to_hive_operator import S3ToHiveOperator, S3ToHiveTransfer
+from airflow.operators.s3_to_redshift_operator import (
+    S3ToRedshiftOperator,
+    S3ToRedshiftTransfer,
+)
+from airflow.operators.slack_operator import SlackAPIOperator, SlackAPIPostOperator
+from airflow.operators.sql import (
+    BaseSQLOperator,
+    BranchSQLOperator,
+    SQLCheckOperator as SQLCheckOperator3,
+    SQLColumnCheckOperator as SQLColumnCheckOperator2,
+    SQLIntervalCheckOperator as SQLIntervalCheckOperator3,
+    SQLTableCheckOperator,
+    SQLThresholdCheckOperator as SQLThresholdCheckOperator2,
+    SQLValueCheckOperator as SQLValueCheckOperator3,
+    _convert_to_float_if_possible,
+    parse_boolean,
+)
+from airflow.operators.sqlite_operator import SqliteOperator
+from airflow.sensors.hive_partition_sensor import HivePartitionSensor
+from airflow.sensors.http_sensor import HttpSensor
+from airflow.sensors.metastore_partition_sensor import MetastorePartitionSensor
+from airflow.sensors.named_hive_partition_sensor import NamedHivePartitionSensor
+from airflow.sensors.s3_key_sensor import S3KeySensor
+from airflow.sensors.sql import SqlSensor
+from airflow.sensors.sql_sensor import SqlSensor2
+from airflow.sensors.web_hdfs_sensor import WebHdfsSensor
 from airflow.www.security import FabAirflowSecurityManagerOverride
 
+# apache-airflow-providers-amazon
+provide_bucket_name()
+GCSToS3Operator()
+GoogleApiToS3Operator()
+GoogleApiToS3Transfer()
+RedshiftToS3Operator()
+RedshiftToS3Transfer()
+S3FileTransformOperator()
+S3Hook()
+S3KeySensor()
+S3ToRedshiftOperator()
+S3ToRedshiftTransfer()
+
+# apache-airflow-providers-celery
+DEFAULT_CELERY_CONFIG
+app
+CeleryExecutor()
+CeleryKubernetesExecutor()
+
+# apache-airflow-providers-common-sql
+_convert_to_float_if_possible()
+parse_boolean()
+BaseSQLOperator()
+BranchSQLOperator()
+CheckOperator()
+ConnectorProtocol()
+DbApiHook()
+DbApiHook2()
+IntervalCheckOperator()
+PrestoCheckOperator()
+PrestoIntervalCheckOperator()
+PrestoValueCheckOperator()
+SQLCheckOperator()
+SQLCheckOperator2()
+SQLCheckOperator3()
+SQLColumnCheckOperator2()
+SQLIntervalCheckOperator()
+SQLIntervalCheckOperator2()
+SQLIntervalCheckOperator3()
+SQLTableCheckOperator()
+SQLThresholdCheckOperator()
+SQLThresholdCheckOperator2()
+SQLValueCheckOperator()
+SQLValueCheckOperator2()
+SQLValueCheckOperator3()
+SqlSensor()
+SqlSensor2()
+ThresholdCheckOperator()
+ValueCheckOperator()
+
+# apache-airflow-providers-daskexecutor
+DaskExecutor()
+
+# apache-airflow-providers-docker
+DockerHook()
+DockerOperator()
+
+# apache-airflow-providers-apache-druid
+DruidDbApiHook()
+DruidHook()
+DruidCheckOperator()
+
+# apache-airflow-providers-apache-hdfs
+WebHDFSHook()
+WebHdfsSensor()
+
+# apache-airflow-providers-apache-hive
+HIVE_QUEUE_PRIORITIES
+closest_ds_partition()
+max_partition()
+HiveCliHook()
+HiveMetastoreHook()
+HiveOperator()
+HivePartitionSensor()
+HiveServer2Hook()
+HiveStatsCollectionOperator()
+HiveToDruidOperator()
+HiveToDruidTransfer()
+HiveToSambaOperator()
+S3ToHiveOperator()
+S3ToHiveTransfer()
+MetastorePartitionSensor()
+NamedHivePartitionSensor()
+
+# apache-airflow-providers-http
+HttpHook()
+HttpSensor()
+SimpleHttpOperator()
+
+# apache-airflow-providers-jdbc
+jaydebeapi
+JdbcHook()
+JdbcOperator()
+
 # apache-airflow-providers-fab
 basic_auth, kerberos_auth
 auth_current_user
 backend_kerberos_auth
 fab_override
-
 FabAuthManager()
 FabAirflowSecurityManagerOverride()
 
-# apache-airflow-providers-celery
-DEFAULT_CELERY_CONFIG
-app
-
-# apache-airflow-providers-common-sql
-ConnectorProtocol()
-DbApiHook()
-
 # apache-airflow-providers-cncf-kubernetes
 ALL_NAMESPACES
 POD_EXECUTOR_DONE_KEY
 
-# apache-airflow-providers-apache-hive
+# apache-airflow-providers-microsoft-mssql
-HIVE_QUEUE_PRIORITIES
+MsSqlHook()
-closest_ds_partition()
+MsSqlOperator()
-max_partition()
+MsSqlToHiveOperator()
+MsSqlToHiveTransfer()
+
+# apache-airflow-providers-mysql
+HiveToMySqlOperator()
+HiveToMySqlTransfer()
+MySqlHook()
+MySqlOperator()
+MySqlToHiveOperator()
+MySqlToHiveTransfer()
+PrestoToMySqlOperator()
+PrestoToMySqlTransfer()
+
+# apache-airflow-providers-oracle
+OracleHook()
+OracleOperator()
+
+# apache-airflow-providers-papermill
+PapermillOperator()
+
+# apache-airflow-providers-apache-pig
+PigCliHook()
+PigOperator()
+
+# apache-airflow-providers-postgres
+Mapping
+PostgresHook()
+PostgresOperator()
+
+# apache-airflow-providers-presto
+PrestoHook()
+
+# apache-airflow-providers-samba
+SambaHook()
+
+# apache-airflow-providers-slack
+SlackHook()
+SlackAPIOperator()
+SlackAPIPostOperator()
+
+# apache-airflow-providers-sqlite
+SqliteHook()
+SqliteOperator()
+
+# apache-airflow-providers-zendesk
+ZendeskHook()