run mbt only in current month

ShubhamBansal1997 · Dec 2, 2019 · 6beadf2 · 6beadf2
1 parent d780be1
commit 6beadf2
Showing 1 changed file with 7 additions and 8 deletions.
diff --git a/dags/dashboard_subdags.py b/dags/dashboard_subdags.py
@@ -162,13 +162,6 @@ def generate_uuid():
         dag=monthly_dag
     )
 
-    create_mbt = BashOperator(
-        task_id='create_mbt_for_month',
-        bash_command=run_query_template,
-        params={'query': 'create_mbt_for_month'},
-        dag=monthly_dag
-    )
-
     get_agg_id >> create_aggregation_record >> daily_attendance
     daily_attendance >> stage_1_tasks
     daily_attendance >> update_months_table
@@ -184,7 +177,6 @@ def generate_uuid():
     agg_ccs_record >> agg_awc_table
     agg_awc_table >> ls_tasks
     ls_tasks >> agg_ls_table
-    agg_awc_table >> create_mbt
 
     if interval == 0:
         aggregate_awc_daily = BashOperator(
@@ -194,5 +186,12 @@ def generate_uuid():
             dag=monthly_dag
         )
         agg_awc_table >> aggregate_awc_daily
+        create_mbt = BashOperator(
+            task_id='create_mbt_for_month',
+            bash_command=run_query_template,
+            params={'query': 'create_mbt_for_month'},
+            dag=monthly_dag
+        )
+        agg_awc_table >> create_mbt
 
     return monthly_dag