Airflow BigQueryOperator: how to save query result in a partitioned Table?

后端 未结 4 2055
长情又很酷
长情又很酷 2021-01-01 00:23

I have a simple DAG

from airflow import DAG
from airflow.contrib.operators.bigquery_operator import BigQueryOperator

with DAG(dag_id=\'my_dags.my_dag\') as         


        
4条回答
  •  一个人的身影
    2021-01-01 01:15

    Using BigQueryOperator you can pass time_partitioning parameter which will create ingestion-time partitioned tables

    bq_cmd = BigQueryOperator (
                task_id=                    "task_id",
                sql=                        [query],
                destination_dataset_table=  destination_tbl,
                use_legacy_sql=             False,
                write_disposition=          'WRITE_TRUNCATE',
                time_partitioning=          {'time_partitioning_type':'DAY'},
                allow_large_results=        True,
                trigger_rule=               'all_success',
                query_params=               query_params,
                dag=                        dag
            )
    

提交回复
热议问题