气流:使用BigQueryCheckOperator模板化的参数

Question

我想使用气流中的BigQueryCheckOperator，其中查询需要将执行日期作为参数。我写了下面的操作符检查当前日期，但我不明白如何使用日期的execution_date。

BigQueryCheckOperator(
task_id='mytask',
retry_delay=10,
retries=2,
sql="""
SELECT
COUNT(*)
FROM
mydataset.mytable AS tb
WHERE
DATE(tb.date) = '{check_date}'
""".format(
check_date=(datetime.now()).strftime("%Y-%m-%d")
),
bigquery_conn_id='my_google_cloud_conn',
dag=dag)

你能帮我吗?

Answer 1

您可以使用Jinja模板和变量= {{ds}}

BigQueryCheckOperator(
task_id='mytask',
retry_delay=10,
retries=2,
sql="""
SELECT
COUNT(*)
FROM
mydataset.mytable AS tb
WHERE
DATE(tb.date) = DATE('{{ ds }}')""",
bigquery_conn_id='my_google_cloud_conn',
dag=dag)