def _delete_ledgers_for_case(case_id): transactions = LedgerAccessorSQL.get_ledger_transactions_for_case( case_id) form_ids = {tx.form_id for tx in transactions} for form_id in form_ids: LedgerAccessorSQL.delete_ledger_transactions_for_form( [case_id], form_id) LedgerAccessorSQL.delete_ledger_values(case_id)
def hard_rebuild_ledgers(case_id, section_id=None, entry_id=None): transactions = LedgerAccessorSQL.get_ledger_transactions_for_case(case_id, section_id, entry_id) if not transactions: LedgerAccessorSQL.delete_ledger_values(case_id, section_id, entry_id) return ledger_value = LedgerAccessorSQL.get_ledger_value(case_id, section_id, entry_id) ledger_value = LedgerProcessorSQL._rebuild_ledger_value_from_transactions(ledger_value, transactions) LedgerAccessorSQL.save_ledger_values([ledger_value]) publish_ledger_v2_saved(ledger_value)
def hard_rebuild_ledgers(domain, case_id, section_id, entry_id): transactions = LedgerAccessorSQL.get_ledger_transactions_for_case(case_id, section_id, entry_id) if not transactions: LedgerAccessorSQL.delete_ledger_values(case_id, section_id, entry_id) publish_ledger_v2_deleted(domain, case_id, section_id, entry_id) return ledger_value = LedgerAccessorSQL.get_ledger_value(case_id, section_id, entry_id) ledger_value = LedgerProcessorSQL._rebuild_ledger_value_from_transactions(ledger_value, transactions) LedgerAccessorSQL.save_ledger_values([ledger_value]) publish_ledger_v2_saved(ledger_value)
def delete_all_ledgers(domain): if should_use_sql_backend(domain): for case_id in CaseAccessorSQL.get_case_ids_in_domain(domain): transactions = LedgerAccessorSQL.get_ledger_transactions_for_case(case_id) form_ids = {tx.form_id for tx in transactions} for form_id in form_ids: LedgerAccessorSQL.delete_ledger_transactions_for_form([case_id], form_id) LedgerAccessorSQL.delete_ledger_values(case_id) else: from casexml.apps.stock.models import StockReport from casexml.apps.stock.models import StockTransaction stock_report_ids = StockReport.objects.filter(domain=domain).values_list('id', flat=True) StockReport.objects.filter(domain=domain).delete() StockTransaction.objects.filter(report_id__in=stock_report_ids).delete()
def test_delete_ledger_values_case_section(self): from corehq.apps.commtrack.tests.util import get_single_balance_block form_id = self._submit_ledgers([ get_single_balance_block(self.case_one.case_id, product_id, 10) for product_id in [self.product_a._id, self.product_b._id] ]) ledger_values = LedgerAccessorSQL.get_ledger_values_for_case(self.case_one.case_id) self.assertEqual(2, len(ledger_values)) LedgerAccessorSQL.delete_ledger_transactions_for_form([self.case_one.case_id], form_id) deleted = LedgerAccessorSQL.delete_ledger_values(self.case_one.case_id, 'stock') self.assertEqual(2, deleted) ledger_values = LedgerAccessorSQL.get_ledger_values_for_case(self.case_one.case_id) self.assertEqual(0, len(ledger_values))
def test_delete_ledger_values_case_section_1(self): from corehq.apps.commtrack.tests import get_single_balance_block form_id = self._submit_ledgers([ get_single_balance_block(self.case_one.case_id, self.product_a._id, 10, section_id=section_id) for section_id in ['stock', 'consumption'] ]) ledger_values = LedgerAccessorSQL.get_ledger_values_for_case(self.case_one.case_id) self.assertEqual(2, len(ledger_values)) LedgerAccessorSQL.delete_ledger_transactions_for_form([self.case_one.case_id], form_id) deleted = LedgerAccessorSQL.delete_ledger_values(self.case_one.case_id, 'stock') self.assertEqual(1, deleted) ledger_values = LedgerAccessorSQL.get_ledger_values_for_case(self.case_one.case_id) self.assertEqual(1, len(ledger_values)) self.assertEqual('consumption', ledger_values[0].section_id)
def test_delete_ledger_values_case_section(self): from corehq.apps.commtrack.tests.util import get_single_balance_block form_id = self._submit_ledgers([ get_single_balance_block(self.case_one.case_id, product_id, 10) for product_id in [self.product_a._id, self.product_b._id] ]) ledger_values = LedgerAccessorSQL.get_ledger_values_for_case( self.case_one.case_id) self.assertEqual(2, len(ledger_values)) LedgerAccessorSQL.delete_ledger_transactions_for_form( [self.case_one.case_id], form_id) deleted = LedgerAccessorSQL.delete_ledger_values( self.case_one.case_id, 'stock') self.assertEqual(2, deleted) ledger_values = LedgerAccessorSQL.get_ledger_values_for_case( self.case_one.case_id) self.assertEqual(0, len(ledger_values))
def _delete_ledgers_for_case(case_id): transactions = LedgerAccessorSQL.get_ledger_transactions_for_case(case_id) form_ids = {tx.form_id for tx in transactions} for form_id in form_ids: LedgerAccessorSQL.delete_ledger_transactions_for_form([case_id], form_id) LedgerAccessorSQL.delete_ledger_values(case_id)
def delete_ledger_values(self, case_id): LedgerAccessorSQL.delete_ledger_values(case_id)
def test_delete_ledger_values_raise_error_case(self): with self.assertRaisesRegexp(LedgerSaveError, '.*still has transactions.*'): LedgerAccessorSQL.delete_ledger_values(self.case.case_id)
def test_delete_ledger_values_raise_error_case_section_entry(self): with self.assertRaisesRegexp(LedgerSaveError, '.*still has transactions.*'): LedgerAccessorSQL.delete_ledger_values(self.case.case_id, 'stock', self.product._id)