问将Airflow BigQueryInsertJobOperator和BigQueryGetDataOperator优先级更改为Batch

EN

execute_insert_query = BigQueryExecuteQueryOperator(
    task_id="execute_insert_query",
    sql=INSERT_ROWS_QUERY,
    use_legacy_sql=False,
    location=location,
    priority='BATCH',
)

class MyBigQueryInsertJobOperator(BigQueryInsertJobOperator):
    def __init__(
        self,
        priority: str = 'INTERACTIVE',
        **kwargs,
    ) -> None:
        super().__init__(**kwargs)
        self.priority = priority

    def _submit_job(
        self,
        hook: BigQueryHook,
        job_id: str,
    ) -> BigQueryJob:
        # Submit a new job
        job = hook.insert_job(
            configuration=self.configuration,
            project_id=self.project_id,
            location=self.location,
            job_id=job_id,
            priority=self.priority,
        )
        # Start the job and wait for it to complete and get the result.
        job.result()
        return job