def setUpClass(cls):
super(AggregationScriptTestBase, cls).setUpClass()
_setup_ucr_tables()
update_child_table(TEST_DOMAIN)
update_child_history_table(TEST_DOMAIN)
update_woman_table(TEST_DOMAIN)
update_woman_history_table(TEST_DOMAIN)
update_ccs_record_table(TEST_DOMAIN)
for month in range(1, 3):
update_agg_awc_table(TEST_DOMAIN, date(2019, month, 1))
update_agg_village_table(TEST_DOMAIN, date(2019, month, 1))
def post(self, request, *args, **kwargs):
date_param = self.request.POST.get('date')
if not date_param:
messages.error(request, 'Date is required')
return redirect(self.urlname, domain=self.domain)
date = force_to_date(date_param)
update_child_table(self.domain)
update_child_history_table(self.domain)
update_ccs_record_table(self.domain)
update_woman_table(self.domain)
update_woman_history_table(self.domain)
update_agg_awc_table(self.domain, date)
update_agg_village_table(self.domain, date)
messages.success(request, 'Aggregation task has run.')
return redirect(self.urlname, domain=self.domain)