sfos/ledger
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Added Python interface for post_t

Browse source
This commit is contained in:
John Wiegley 2009-11-03 02:25:25 -05:00
parent ed0e01812a
commit a462b93e30
2 changed files with 130 additions and 13 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
src/py_item.cc
View file

@ -96,7 +96,7 @@ void export_item()
scope().attr("ITEM_GENERATED") = ITEM_GENERATED;
scope().attr("ITEM_TEMP") = ITEM_TEMP;
class_< item_t > ("Item", init<uint_least8_t>())
class_< item_t > ("JournalItem", init<uint_least8_t>())
#if 1
.def("flags", &supports_flags<>::flags)
.def("has_flags", &supports_flags<>::has_flags)

141
src/py_post.cc
View file

@ -32,33 +32,150 @@
#include <system.hh>
#include "pyinterp.h"
#include "post.h"
#include "xact.h"
namespace ledger {
using namespace boost::python;
#define EXC_TRANSLATOR(type) \
void exc_translate_ ## type(const type& err) { \
PyErr_SetString(PyExc_ArithmeticError, err.what()); \
namespace {
bool py_has_tag_1s(post_t& post, const string& tag) {
return post.has_tag(tag);
}
bool py_has_tag_1m(post_t& post, const mask_t& tag_mask) {
return post.has_tag(tag_mask);
}
bool py_has_tag_2m(post_t& post, const mask_t& tag_mask,
const boost::optional<mask_t>& value_mask) {
return post.has_tag(tag_mask, value_mask);
}
//EXC_TRANSLATOR(post_error)
boost::optional<string> py_get_tag_1s(post_t& post, const string& tag) {
return post.get_tag(tag);
}
boost::optional<string> py_get_tag_1m(post_t& post, const mask_t& tag_mask) {
return post.get_tag(tag_mask);
}
boost::optional<string> py_get_tag_2m(post_t& post, const mask_t& tag_mask,
const boost::optional<mask_t>& value_mask) {
return post.get_tag(tag_mask, value_mask);
}
post_t::xdata_t& py_xdata(post_t& post) {
return post.xdata();
}
account_t * py_reported_account(post_t& post) {
return post.reported_account();
}
} // unnamed namespace
void export_post()
{
#if 0
class_< post_t > ("Post")
;
scope().attr("POST_EXT_RECEIVED") = POST_EXT_RECEIVED;
scope().attr("POST_EXT_HANDLED") = POST_EXT_HANDLED;
scope().attr("POST_EXT_DISPLAYED") = POST_EXT_DISPLAYED;
scope().attr("POST_EXT_DIRECT_AMT") = POST_EXT_DIRECT_AMT;
scope().attr("POST_EXT_SORT_CALC") = POST_EXT_SORT_CALC;
scope().attr("POST_EXT_COMPOUND") = POST_EXT_COMPOUND;
scope().attr("POST_EXT_VISITED") = POST_EXT_VISITED;
scope().attr("POST_EXT_MATCHES") = POST_EXT_MATCHES;
scope().attr("POST_EXT_CONSIDERED") = POST_EXT_CONSIDERED;
class_< post_t::xdata_t > ("PostingXData")
#if 1
.def("flags", &supports_flags<uint_least16_t>::flags)
.def("has_flags", &supports_flags<uint_least16_t>::has_flags)
.def("set_flags", &supports_flags<uint_least16_t>::set_flags)
.def("clear_flags", &supports_flags<uint_least16_t>::clear_flags)
.def("add_flags", &supports_flags<uint_least16_t>::add_flags)
.def("drop_flags", &supports_flags<uint_least16_t>::drop_flags)
#endif
//register_optional_to_python<amount_t>();
.add_property("visited_value",
make_getter(&post_t::xdata_t::visited_value),
make_setter(&post_t::xdata_t::visited_value))
.add_property("compound_value",
make_getter(&post_t::xdata_t::compound_value),
make_setter(&post_t::xdata_t::compound_value))
.add_property("total",
make_getter(&post_t::xdata_t::total),
make_setter(&post_t::xdata_t::total))
.add_property("count",
make_getter(&post_t::xdata_t::count),
make_setter(&post_t::xdata_t::count))
.add_property("date",
make_getter(&post_t::xdata_t::date),
make_setter(&post_t::xdata_t::date))
.add_property("datetime",
make_getter(&post_t::xdata_t::datetime),
make_setter(&post_t::xdata_t::datetime))
.add_property("account",
make_getter(&post_t::xdata_t::account),
make_setter(&post_t::xdata_t::account))
.add_property("sort_values",
make_getter(&post_t::xdata_t::sort_values),
make_setter(&post_t::xdata_t::sort_values))
;
//implicitly_convertible<string, amount_t>();
scope().attr("POST_VIRTUAL") = POST_VIRTUAL;
scope().attr("POST_MUST_BALANCE") = POST_MUST_BALANCE;
scope().attr("POST_CALCULATED") = POST_CALCULATED;
scope().attr("POST_COST_CALCULATED") = POST_COST_CALCULATED;
#define EXC_TRANSLATE(type) \
register_exception_translator<type>(&exc_translate_ ## type);
class_< post_t, bases<item_t> > ("Posting")
//.def(init<account_t *>())
//EXC_TRANSLATE(post_error);
.add_property("xact",
make_getter(&post_t::xact,
return_value_policy<reference_existing_object>()),
make_setter(&post_t::xact,
with_custodian_and_ward<1, 2>()))
.add_property("account",
make_getter(&post_t::account,
return_value_policy<reference_existing_object>()),
make_setter(&post_t::account,
with_custodian_and_ward<1, 2>()))
.add_property("amount",
make_getter(&post_t::amount),
make_setter(&post_t::amount))
.add_property("cost",
make_getter(&post_t::cost),
make_setter(&post_t::cost))
.add_property("assigned_amount",
make_getter(&post_t::assigned_amount),
make_setter(&post_t::assigned_amount))
.def("has_tag", py_has_tag_1s)
.def("has_tag", py_has_tag_1m)
.def("has_tag", py_has_tag_2m)
.def("get_tag", py_get_tag_1s)
.def("get_tag", py_get_tag_1m)
.def("get_tag", py_get_tag_2m)
.def("date", &post_t::date)
.def("effective_date", &post_t::effective_date)
.def("must_balance", &post_t::must_balance)
.def("lookup", &post_t::lookup)
.def("valid", &post_t::valid)
.def("has_xdata", &post_t::has_xdata)
.def("clear_xdata", &post_t::clear_xdata)
.def("xdata", py_xdata,
return_value_policy<reference_existing_object>())
.def("add_to_value", &post_t::add_to_value)
.def("set_reported_account", &post_t::set_reported_account)
.def("reported_account", py_reported_account,
return_value_policy<reference_existing_object>())
;
}
} // namespace ledger