Airflow 2.0中无法导入BigQueryOperator的问题求助

问题原因 #

在较新版本的Airflow（2.2+）及对应的Google Cloud Provider包（>=3.0.0）中，原BigQueryOperator已被标记为废弃并移除，替换为更细分的操作类，因此会出现「模块中找不到该类」的报错。

解决方案 #

将代码中的BigQueryOperator替换为BigQueryExecuteQueryOperator，导入路径和代码调整如下：

from airflow.providers.google.cloud.operators.bigquery import BigQueryExecuteQueryOperator

bq_to_bq  = BigQueryExecuteQueryOperator(
        task_id                     = "bq_to_bq",
        sql                         = "SELECT count(*) as count FROM `raw_bikesharing.stations`",
        destination_dataset_table   = 'dwh_bikesharing.temporary_stations_count',
        write_disposition           = 'WRITE_TRUNCATE',
        create_disposition          = 'CREATE_IF_NEEDED',
        use_legacy_sql              = False,
        priority                    = 'BATCH'
 )

额外检查项 #

• 确认Cloud Composer环境的Airflow版本：在Composer控制台的环境详情页可查看当前Airflow版本，确保使用的操作类与版本匹配。
• 若需兼容旧版本代码，可降低Google Cloud Provider包版本，但不推荐此做法——废弃类后续会被完全移除，存在兼容性风险。

内容的提问来源于stack exchange，提问作者Jesse Mensah