Wallet.(balance/total_sent/total_received/fee_per_kb)

Author: samuraisam

python/bindings.c

@@ -1178,7 +1178,46 @@ PyObject *b_WalletGetChangeAddress(b_Wallet *self, void *closure) {
     return (PyObject *)addrObj;
 }
 
+PyObject *b_WalletGetBalance(b_Wallet *self, void *closure) {
+    return Py_BuildValue("K", BRWalletBalance(self->ob_fval));
+}
+
+PyObject *b_WalletGetTotalSent(b_Wallet *self, void *closure) {
+    return Py_BuildValue("K", BRWalletTotalSent(self->ob_fval));
+}
+
+PyObject *b_WalletGetTotalReceived(b_Wallet *self, void *closure) {
+    return Py_BuildValue("K", BRWalletTotalReceived(self->ob_fval));
+}
+
+PyObject *b_WalletGetFeePerKB(b_Wallet *self, void *closure) {
+    return Py_BuildValue("K", BRWalletFeePerKb(self->ob_fval));
+}
+
+int b_WalletSetFeePerKB(b_Wallet *self, PyObject *value, void *closure) {
+    if (!PyLong_Check(value)) {
+        PyErr_SetString(PyExc_TypeError, "fee per kb must be a number");
+        return -1;;
+    }
+    BRWalletSetFeePerKb(self->ob_fval, PyLong_AsUnsignedLongLong(value));
+    return 0;
+}
+
 static PyGetSetDef b_WalletGetSetters[] = {
+    // props
+    {"balance", (getter)b_WalletGetBalance, NULL,
+     "gets the total balance int he wallet, not including transactions known to be invalid",
+     NULL},
+    {"total_sent", (getter)b_WalletGetTotalSent, NULL,
+     "gets the total amount sent not including change addresses",
+     NULL},
+    {"total_received", (getter)b_WalletGetTotalReceived, NULL,
+     "gets the total amount received not including change addresses",
+     NULL},
+    {"fee_per_kb", (getter)b_WalletGetFeePerKB, (setter)b_WalletSetFeePerKB,
+     "fee-per-kb size to use when creating a transaction, in satoshis",
+     NULL},
+    // callbacks
     {"on_balance_changed",
      (getter)b_WalletGetBalanceChanged, (setter)b_WalletSetBalanceChanged,
      "callback fired when sync is started",

python/tests.py

@@ -229,3 +229,24 @@ def test_get_change_address(self):
         wallet = self._get_wallet()
         self.assertNotEqual(wallet.change_address, None)
         self.assertEqual(wallet.change_address, wallet.change_address)
+
+    def test_get_balance(self):
+        wallet = self._get_wallet()
+        self.assertEqual(wallet.balance, 0)
+
+    def test_get_total_sent(self):
+        wallet = self._get_wallet()
+        self.assertEqual(wallet.total_sent, 0)
+
+    def test_get_total_received(self):
+        wallet = self._get_wallet()
+        self.assertEqual(wallet.total_received, 0)
+
+    def test_get_fee(self):
+        wallet = self._get_wallet()
+        self.assertGreater(wallet.fee_per_kb, 1)
+
+    def test_set_fee(self):
+        wallet = self._get_wallet()
+        wallet.fee_per_kb = 123
+        self.assertEqual(wallet.fee_per_kb, 123)