Added Python interface for post_t

1 files changed, 129 insertions, 12 deletions

diff --git a/src/py_post.cc b/src/py_post.cc
index b1d62464..b4849c3d 100644
--- a/src/py_post.cc
+++ b/src/py_post.cc
@@ -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);
+  }
+
+  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();
   }
 
-//EXC_TRANSLATOR(post_error)
+  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))
+    ;
+
+  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;
+
+  class_< post_t, bases<item_t> > ("Posting")
+    //.def(init<account_t *>())
+
+    .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)
 
-  //implicitly_convertible<string, amount_t>();
+    .def("date", &post_t::date)
+    .def("effective_date", &post_t::effective_date)
 
-#define EXC_TRANSLATE(type) \
-  register_exception_translator<type>(&exc_translate_ ## type);
+    .def("must_balance", &post_t::must_balance)
 
-  //EXC_TRANSLATE(post_error);
+    .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