1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 | from airflow import DAG
from airflow.utils.dates import days_ago
from template.dag_template import build_dag
dag_id = 'compliance-daily-export_v002'
emr_steps = """[
{
"step-name": "CCD-Hour-Model-Export",
"main-class": "com.viooh.export.CassandraCCDHourToS3",
"artifact" :"compliance-daily-report"
}
]"""
# cluster level parameters (optional)
cluster_args = {
"master-instance-types": "m5.xlarge,m5.2xlarge",
"core-instance-types": "m5.xlarge,m5.2xlarge",
"task-instance-types": "m5.xlarge,m5.2xlarge",
"core-instance-capacity": 3,
"task-instance-capacity": 0
}
# dag parameter
dag_args = {
'owner': 'data.engineers@viooh.com',
'start_date': days_ago(1)
}
dag = DAG(
dag_id,
schedule_interval="1 0 * * *", # cron expression
default_args=dag_args)
build_dag(emr_steps=emr_steps, dag=dag, cluster_args=cluster_args)
|