a bit more refactoring
This commit is contained in:
parent
7a19532049
commit
1416f3b978
9 changed files with 172 additions and 121 deletions
|
|
@ -1,8 +1,8 @@
|
||||||
lib_LTLIBRARIES = libledger.la
|
lib_LTLIBRARIES = libledger.la
|
||||||
libledger_la_CXXFLAGS =
|
libledger_la_CXXFLAGS =
|
||||||
libledger_la_SOURCES = amount.cc balance.cc value.cc autoxact.cc \
|
libledger_la_SOURCES = amount.cc autoxact.cc balance.cc binary.cc \
|
||||||
binary.cc config.cc datetime.cc format.cc journal.cc option.cc \
|
config.cc datetime.cc derive.cc format.cc journal.cc option.cc \
|
||||||
parser.cc qif.cc quotes.cc textual.cc valexpr.cc walk.cc
|
parser.cc qif.cc quotes.cc textual.cc valexpr.cc value.cc walk.cc
|
||||||
if READ_GNUCASH
|
if READ_GNUCASH
|
||||||
libledger_la_SOURCES += gnucash.cc
|
libledger_la_SOURCES += gnucash.cc
|
||||||
endif
|
endif
|
||||||
|
|
|
||||||
147
derive.cc
Normal file
147
derive.cc
Normal file
|
|
@ -0,0 +1,147 @@
|
||||||
|
#include "derive.h"
|
||||||
|
#include "datetime.h"
|
||||||
|
#include "valexpr.h"
|
||||||
|
#include "error.h"
|
||||||
|
|
||||||
|
namespace ledger {
|
||||||
|
|
||||||
|
entry_t * derive_new_entry(journal_t& journal,
|
||||||
|
strings_list::iterator i,
|
||||||
|
strings_list::iterator end)
|
||||||
|
{
|
||||||
|
std::auto_ptr<entry_t> added(new entry_t);
|
||||||
|
|
||||||
|
entry_t * matching = NULL;
|
||||||
|
|
||||||
|
if (! parse_date((*i).c_str(), &added->date))
|
||||||
|
throw error("Bad date passed to 'entry'");
|
||||||
|
|
||||||
|
if (++i == end)
|
||||||
|
throw error("Too few arguments to 'entry'");
|
||||||
|
|
||||||
|
mask_t regexp(*i++);
|
||||||
|
|
||||||
|
for (entries_list::reverse_iterator j = journal.entries.rbegin();
|
||||||
|
j != journal.entries.rend();
|
||||||
|
j++)
|
||||||
|
if (regexp.match((*j)->payee)) {
|
||||||
|
matching = *j;
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
|
||||||
|
added->payee = matching ? matching->payee : regexp.pattern;
|
||||||
|
|
||||||
|
if (i == end)
|
||||||
|
throw error("Too few arguments to 'entry'");
|
||||||
|
|
||||||
|
if ((*i)[0] == '-' || std::isdigit((*i)[0])) {
|
||||||
|
if (! matching)
|
||||||
|
throw error("Could not determine the account to draw from");
|
||||||
|
|
||||||
|
transaction_t * m_xact, * xact, * first;
|
||||||
|
m_xact = matching->transactions.front();
|
||||||
|
|
||||||
|
first = xact = new transaction_t(m_xact->account, amount_t(*i++));
|
||||||
|
added->add_transaction(xact);
|
||||||
|
|
||||||
|
if (xact->amount.commodity().symbol.empty())
|
||||||
|
xact->amount.set_commodity(m_xact->amount.commodity());
|
||||||
|
|
||||||
|
m_xact = matching->transactions.back();
|
||||||
|
|
||||||
|
xact = new transaction_t(m_xact->account, - first->amount);
|
||||||
|
added->add_transaction(xact);
|
||||||
|
|
||||||
|
if (i != end) {
|
||||||
|
account_t * acct = journal.find_account_re(*i);
|
||||||
|
if (! acct)
|
||||||
|
acct = journal.find_account(*i);
|
||||||
|
if (acct)
|
||||||
|
added->transactions.back()->account = acct;
|
||||||
|
}
|
||||||
|
} else {
|
||||||
|
while (i != end) {
|
||||||
|
std::string& re_pat(*i++);
|
||||||
|
account_t * acct = NULL;
|
||||||
|
commodity_t * cmdty = NULL;
|
||||||
|
|
||||||
|
if (matching) {
|
||||||
|
mask_t acct_regex(re_pat);
|
||||||
|
|
||||||
|
for (transactions_list::const_iterator x
|
||||||
|
= matching->transactions.begin();
|
||||||
|
x != matching->transactions.end();
|
||||||
|
x++) {
|
||||||
|
if (acct_regex.match((*x)->account->fullname())) {
|
||||||
|
acct = (*x)->account;
|
||||||
|
cmdty = &(*x)->amount.commodity();
|
||||||
|
break;
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
if (! acct) {
|
||||||
|
acct = journal.find_account_re(re_pat);
|
||||||
|
if (acct && acct->transactions.size() > 0)
|
||||||
|
cmdty = &acct->transactions.back()->amount.commodity();
|
||||||
|
}
|
||||||
|
|
||||||
|
if (! acct)
|
||||||
|
acct = journal.find_account(re_pat);
|
||||||
|
|
||||||
|
if (i == end) {
|
||||||
|
added->add_transaction(new transaction_t(acct));
|
||||||
|
goto done;
|
||||||
|
}
|
||||||
|
|
||||||
|
transaction_t * xact = new transaction_t(acct, amount_t(*i++));
|
||||||
|
if (! xact->amount.commodity())
|
||||||
|
xact->amount.set_commodity(*cmdty);
|
||||||
|
|
||||||
|
added->add_transaction(xact);
|
||||||
|
}
|
||||||
|
|
||||||
|
if (! matching) {
|
||||||
|
throw error("Could not determine the account to draw from");
|
||||||
|
} else {
|
||||||
|
transaction_t * xact
|
||||||
|
= new transaction_t(matching->transactions.back()->account);
|
||||||
|
added->add_transaction(xact);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
done:
|
||||||
|
if (! run_hooks(journal.entry_finalize_hooks, *added))
|
||||||
|
return NULL;
|
||||||
|
|
||||||
|
return added.release();
|
||||||
|
}
|
||||||
|
|
||||||
|
} // namespace ledger
|
||||||
|
|
||||||
|
#ifdef USE_BOOST_PYTHON
|
||||||
|
|
||||||
|
#include <boost/python.hpp>
|
||||||
|
#include <boost/python/detail/api_placeholder.hpp>
|
||||||
|
|
||||||
|
using namespace boost::python;
|
||||||
|
using namespace ledger;
|
||||||
|
|
||||||
|
entry_t * py_derive_new_entry(journal_t& journal, list args)
|
||||||
|
{
|
||||||
|
strings_list strs;
|
||||||
|
|
||||||
|
int l = len(args);
|
||||||
|
for (int i = 0; i < l; i++)
|
||||||
|
strs.push_back(extract<std::string>(args[i]));
|
||||||
|
|
||||||
|
return derive_new_entry(journal, strs.begin(), strs.end());
|
||||||
|
}
|
||||||
|
|
||||||
|
void export_derive()
|
||||||
|
{
|
||||||
|
def("derive_new_entry", py_derive_new_entry,
|
||||||
|
return_value_policy<manage_new_object>());
|
||||||
|
}
|
||||||
|
|
||||||
|
#endif // USE_BOOST_PYTHON
|
||||||
14
derive.h
Normal file
14
derive.h
Normal file
|
|
@ -0,0 +1,14 @@
|
||||||
|
#ifndef _DERIVE_H
|
||||||
|
#define _DERIVE_H
|
||||||
|
|
||||||
|
#include "journal.h"
|
||||||
|
|
||||||
|
namespace ledger {
|
||||||
|
|
||||||
|
entry_t * derive_new_entry(journal_t& journal,
|
||||||
|
strings_list::iterator begin,
|
||||||
|
strings_list::iterator end);
|
||||||
|
|
||||||
|
} // namespace ledger
|
||||||
|
|
||||||
|
#endif // _DERIVE_H
|
||||||
113
journal.cc
113
journal.cc
|
|
@ -426,117 +426,6 @@ bool journal_t::remove_entry(entry_t * entry)
|
||||||
return true;
|
return true;
|
||||||
}
|
}
|
||||||
|
|
||||||
entry_t * journal_t::derive_entry(strings_list::iterator i,
|
|
||||||
strings_list::iterator end)
|
|
||||||
{
|
|
||||||
std::auto_ptr<entry_t> added(new entry_t);
|
|
||||||
|
|
||||||
entry_t * matching = NULL;
|
|
||||||
|
|
||||||
if (! parse_date((*i).c_str(), &added->date))
|
|
||||||
throw error("Bad date passed to 'entry'");
|
|
||||||
|
|
||||||
if (++i == end)
|
|
||||||
throw error("Too few arguments to 'entry'");
|
|
||||||
|
|
||||||
mask_t regexp(*i++);
|
|
||||||
|
|
||||||
for (entries_list::reverse_iterator j = entries.rbegin();
|
|
||||||
j != entries.rend();
|
|
||||||
j++)
|
|
||||||
if (regexp.match((*j)->payee)) {
|
|
||||||
matching = *j;
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
|
|
||||||
added->payee = matching ? matching->payee : regexp.pattern;
|
|
||||||
|
|
||||||
if (i == end)
|
|
||||||
throw error("Too few arguments to 'entry'");
|
|
||||||
|
|
||||||
if ((*i)[0] == '-' || std::isdigit((*i)[0])) {
|
|
||||||
if (! matching)
|
|
||||||
throw error("Could not determine the account to draw from");
|
|
||||||
|
|
||||||
transaction_t * m_xact, * xact, * first;
|
|
||||||
m_xact = matching->transactions.front();
|
|
||||||
|
|
||||||
first = xact = new transaction_t(m_xact->account, amount_t(*i++));
|
|
||||||
added->add_transaction(xact);
|
|
||||||
|
|
||||||
if (xact->amount.commodity().symbol.empty())
|
|
||||||
xact->amount.set_commodity(m_xact->amount.commodity());
|
|
||||||
|
|
||||||
m_xact = matching->transactions.back();
|
|
||||||
|
|
||||||
xact = new transaction_t(m_xact->account, - first->amount);
|
|
||||||
added->add_transaction(xact);
|
|
||||||
|
|
||||||
if (i != end) {
|
|
||||||
account_t * acct = find_account_re(*i);
|
|
||||||
if (! acct)
|
|
||||||
acct = find_account(*i);
|
|
||||||
if (acct)
|
|
||||||
added->transactions.back()->account = acct;
|
|
||||||
}
|
|
||||||
} else {
|
|
||||||
while (i != end) {
|
|
||||||
std::string& re_pat(*i++);
|
|
||||||
account_t * acct = NULL;
|
|
||||||
commodity_t * cmdty = NULL;
|
|
||||||
|
|
||||||
if (matching) {
|
|
||||||
mask_t acct_regex(re_pat);
|
|
||||||
|
|
||||||
for (transactions_list::const_iterator x
|
|
||||||
= matching->transactions.begin();
|
|
||||||
x != matching->transactions.end();
|
|
||||||
x++) {
|
|
||||||
if (acct_regex.match((*x)->account->fullname())) {
|
|
||||||
acct = (*x)->account;
|
|
||||||
cmdty = &(*x)->amount.commodity();
|
|
||||||
break;
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if (! acct) {
|
|
||||||
acct = find_account_re(re_pat);
|
|
||||||
if (acct && acct->transactions.size() > 0)
|
|
||||||
cmdty = &acct->transactions.back()->amount.commodity();
|
|
||||||
}
|
|
||||||
|
|
||||||
if (! acct)
|
|
||||||
acct = find_account(re_pat);
|
|
||||||
|
|
||||||
if (i == end) {
|
|
||||||
added->add_transaction(new transaction_t(acct));
|
|
||||||
goto done;
|
|
||||||
}
|
|
||||||
|
|
||||||
transaction_t * xact = new transaction_t(acct, amount_t(*i++));
|
|
||||||
if (! xact->amount.commodity())
|
|
||||||
xact->amount.set_commodity(*cmdty);
|
|
||||||
|
|
||||||
added->add_transaction(xact);
|
|
||||||
}
|
|
||||||
|
|
||||||
if (! matching) {
|
|
||||||
throw error("Could not determine the account to draw from");
|
|
||||||
} else {
|
|
||||||
transaction_t * xact
|
|
||||||
= new transaction_t(matching->transactions.back()->account);
|
|
||||||
added->add_transaction(xact);
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
done:
|
|
||||||
if (! run_hooks(entry_finalize_hooks, *added))
|
|
||||||
return NULL;
|
|
||||||
|
|
||||||
return added.release();
|
|
||||||
}
|
|
||||||
|
|
||||||
bool journal_t::valid() const
|
bool journal_t::valid() const
|
||||||
{
|
{
|
||||||
if (! master->valid())
|
if (! master->valid())
|
||||||
|
|
@ -798,8 +687,6 @@ void export_journal()
|
||||||
.def("remove_entry_finalize_hook", py_remove_entry_finalize_hook)
|
.def("remove_entry_finalize_hook", py_remove_entry_finalize_hook)
|
||||||
.def("run_entry_finalize_hooks", py_run_entry_finalize_hooks)
|
.def("run_entry_finalize_hooks", py_run_entry_finalize_hooks)
|
||||||
#endif
|
#endif
|
||||||
.def("derive_entry", &journal_t::derive_entry,
|
|
||||||
return_value_policy<manage_new_object>())
|
|
||||||
|
|
||||||
.def("valid", &journal_t::valid)
|
.def("valid", &journal_t::valid)
|
||||||
;
|
;
|
||||||
|
|
|
||||||
|
|
@ -251,9 +251,6 @@ class journal_t
|
||||||
bool add_entry(entry_t * entry);
|
bool add_entry(entry_t * entry);
|
||||||
bool remove_entry(entry_t * entry);
|
bool remove_entry(entry_t * entry);
|
||||||
|
|
||||||
entry_t * derive_entry(strings_list::iterator begin,
|
|
||||||
strings_list::iterator end);
|
|
||||||
|
|
||||||
bool valid() const;
|
bool valid() const;
|
||||||
};
|
};
|
||||||
|
|
||||||
|
|
|
||||||
3
ledger.h
3
ledger.h
|
|
@ -29,6 +29,9 @@
|
||||||
#include <textual.h>
|
#include <textual.h>
|
||||||
#include <autoxact.h>
|
#include <autoxact.h>
|
||||||
#include <binary.h>
|
#include <binary.h>
|
||||||
|
#ifdef READ_GNUCASH
|
||||||
|
#include <gnucash.h>
|
||||||
|
#endif
|
||||||
#include <qif.h>
|
#include <qif.h>
|
||||||
|
|
||||||
#include <error.h>
|
#include <error.h>
|
||||||
|
|
|
||||||
3
main.cc
3
main.cc
|
|
@ -11,6 +11,7 @@
|
||||||
#include "format.h"
|
#include "format.h"
|
||||||
#include "walk.h"
|
#include "walk.h"
|
||||||
#include "quotes.h"
|
#include "quotes.h"
|
||||||
|
#include "derive.h"
|
||||||
#include "option.h"
|
#include "option.h"
|
||||||
#include "config.h"
|
#include "config.h"
|
||||||
#include "debug.h"
|
#include "debug.h"
|
||||||
|
|
@ -239,7 +240,7 @@ int parse_and_report(int argc, char * argv[], char * envp[])
|
||||||
|
|
||||||
std::auto_ptr<entry_t> new_entry;
|
std::auto_ptr<entry_t> new_entry;
|
||||||
if (command == "e") {
|
if (command == "e") {
|
||||||
new_entry.reset(journal->derive_entry(arg, args.end()));
|
new_entry.reset(derive_new_entry(*journal, arg, args.end()));
|
||||||
if (! new_entry.get())
|
if (! new_entry.get())
|
||||||
return 1;
|
return 1;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
2
main.py
2
main.py
|
|
@ -118,7 +118,7 @@ config.process_options(command, args);
|
||||||
|
|
||||||
new_entry = None
|
new_entry = None
|
||||||
if command == "e":
|
if command == "e":
|
||||||
new_entry = journal.derive_entry (args)
|
new_entry = derive_new_entry (journal, args)
|
||||||
if new_entry is None:
|
if new_entry is None:
|
||||||
sys.exit (1)
|
sys.exit (1)
|
||||||
|
|
||||||
|
|
|
||||||
|
|
@ -29,6 +29,7 @@ void export_format();
|
||||||
void export_valexpr();
|
void export_valexpr();
|
||||||
void export_datetime();
|
void export_datetime();
|
||||||
void export_autoxact();
|
void export_autoxact();
|
||||||
|
void export_derive();
|
||||||
|
|
||||||
void initialize_ledger_for_python()
|
void initialize_ledger_for_python()
|
||||||
{
|
{
|
||||||
|
|
@ -50,6 +51,7 @@ void initialize_ledger_for_python()
|
||||||
export_valexpr();
|
export_valexpr();
|
||||||
export_datetime();
|
export_datetime();
|
||||||
export_autoxact();
|
export_autoxact();
|
||||||
|
export_derive();
|
||||||
|
|
||||||
module_initialized = true;
|
module_initialized = true;
|
||||||
}
|
}
|
||||||
|
|
|
||||||
Loading…
Add table
Reference in a new issue