# Generated via https://github.com/mozilla/bigquery-etl/blob/master/bigquery_etl/query_scheduling/generate_airflow_dags.py

from airflow import DAG
from airflow.operators.sensors import ExternalTaskSensor
import datetime
from utils.gcp import bigquery_etl_query, gke_command

default_args = {
    "owner": "test@example.org",
    "start_date": datetime.datetime(2020, 1, 1, 0, 0),
    "end_date": None,
    "email": ["test@example.org"],
    "depends_on_past": False,
    "retry_delay": datetime.timedelta(seconds=3600),
    "email_on_failure": True,
    "email_on_retry": True,
    "retries": 2,
}

with DAG(
    "bqetl_test_dag", default_args=default_args, schedule_interval="@daily"
) as dag:

    test__python_script_query__v1 = gke_command(
        task_id="test__python_script_query__v1",
        command=[
            "python",
            "sql/moz-fx-data-test-project/test/python_script_query_v1/query.py",
        ]
        + ["--date", "{{ds}}"],
        docker_image="mozilla/bigquery-etl:latest",
        owner="test@example.com",
        email=["test@example.com"],
    )
