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)
