1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
|
#include "reconcile.h"
#include "walk.h"
namespace ledger {
#define xact_next(x) ((transaction_t *)transaction_xdata(*x).ptr)
#define xact_next_ptr(x) ((transaction_t **)&transaction_xdata(*x).ptr)
static bool search_for_balance(amount_t& amount,
transaction_t ** prev, transaction_t * next)
{
for (; next; next = xact_next(next)) {
transaction_t * temp = *prev;
*prev = next;
amount -= next->amount;
if (! amount ||
search_for_balance(amount, xact_next_ptr(next), xact_next(next)))
return true;
amount += next->amount;
*prev = temp;
}
return false;
}
void reconcile_transactions::push_to_handler(transaction_t * first)
{
for (; first; first = xact_next(first))
item_handler<transaction_t>::operator()(*first);
item_handler<transaction_t>::flush();
}
void reconcile_transactions::flush()
{
value_t cleared_balance;
value_t pending_balance;
transaction_t * first = NULL;
transaction_t ** last_ptr = &first;
bool found_pending = false;
for (transactions_list::iterator x = xacts.begin();
x != xacts.end();
x++) {
if (! cutoff || std::difftime((*x)->date(), cutoff) < 0) {
switch ((*x)->state) {
case transaction_t::CLEARED:
cleared_balance += (*x)->amount;
break;
case transaction_t::UNCLEARED:
case transaction_t::PENDING:
pending_balance += (*x)->amount;
*last_ptr = *x;
last_ptr = xact_next_ptr(*x);
break;
}
}
}
if (cleared_balance.type >= value_t::BALANCE)
throw error("Cannot reconcile accounts with multiple commodities");
cleared_balance.cast(value_t::AMOUNT);
balance.cast(value_t::AMOUNT);
commodity_t& cb_comm = ((amount_t *) cleared_balance.data)->commodity();
commodity_t& b_comm = ((amount_t *) balance.data)->commodity();
balance -= cleared_balance;
if (balance.type >= value_t::BALANCE)
throw error(std::string("Reconcile balance is not of the same commodity ('") +
b_comm.symbol + "' != '" + cb_comm.symbol + "')");
// If the amount to reconcile is the same as the pending balance,
// then assume an exact match and return the results right away.
amount_t to_reconcile = *((amount_t *) balance.data);
pending_balance.cast(value_t::AMOUNT);
if (to_reconcile == *((amount_t *) pending_balance.data) ||
search_for_balance(to_reconcile, &first, first)) {
push_to_handler(first);
} else {
throw error("Could not reconcile account!");
}
}
} // namespace ledger
#ifdef USE_BOOST_PYTHON
#include <boost/python.hpp>
using namespace boost::python;
using namespace ledger;
void export_reconcile()
{
class_< reconcile_transactions, bases<item_handler<transaction_t> > >
("ReconcileTransactions",
init<item_handler<transaction_t> *, const value_t&, time_t>()
[with_custodian_and_ward<1, 2>()])
.def("flush", &reconcile_transactions::flush)
.def("__call__", &reconcile_transactions::operator())
;
}
#endif // USE_BOOST_PYTHON
|