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)