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 37 38 | from airflow import DAG
from airflow.utils.dates import days_ago
from template.dag_template import build_dag
dag_id = 'exchange-daily-revenue-report_v002'
emr_steps = """[
{
"step-name": "DailyRevenueReportJob",
"main-class": "com.viooh.smex.dailyrevenue.DailyRevenueReport",
"group-id":"com/viooh/smex",
"artifact": "daily-revenue-report"
}
]"""
# cluster level parameters (optional)
cluster_args = {
"master-instance-types": "m5.xlarge,m5.2xlarge,m5.4xlarge",
"core-instance-types": "m5.xlarge,m5.2xlarge,m5.4xlarge",
"task-instance-types": "m5.xlarge,m5.2xlarge,m5.4xlarge",
"core-instance-capacity": 2,
"task-instance-capacity": 0,
"emr-version": "emr-6.14.0"
}
# dag parameter
dag_args = {
'owner': 'data.engineers@viooh.com',
'start_date': days_ago(1)
}
dag = DAG(
dag_id,
schedule_interval='0 9 * * *', # cron expression
default_args=dag_args)
build_dag(emr_steps=emr_steps, dag=dag, cluster_args=cluster_args)
|