Add websocket notification for underfunded order

Author: cpacia

dht/storage.py

@@ -148,3 +148,11 @@ def get_ttl(self, keyword, key):
         cursor = self.db.cursor()
         cursor.execute('''SELECT birthday FROM dht WHERE keyword=? AND id=?''', (keyword.encode("hex"), key,))
         return self.ttl - (time.time() - cursor.fetchall()[0][0])
+
+    def get_db_size(self):
+        cursor = self.db.cursor()
+        cursor.execute('''PRAGMA page_count;''')
+        count = cursor.fetchone()[0]
+        cursor.execute('''PRAGMA page_size;''')
+        size = cursor.fetchone()[0]
+        return count * size

market/contracts.py

@@ -861,16 +861,28 @@ def on_tx_received(self, address_version, address_hash, height, block_hash, tx):
 
             # get the amount (in satoshi) the user is expected to pay
             amount_to_pay = int(float(self.contract["buyer_order"]["order"]["payment"]["amount"]) * 100000000)
+            val = 0
             if tx not in self.received_txs:  # make sure we aren't parsing the same tx twice.
                 outpoints = transaction.check_for_funding(
                     self.contract["buyer_order"]["order"]["payment"]["address"])
                 if outpoints is not None:
                     for outpoint in outpoints:
                         self.amount_funded += outpoint["value"]
+                        val += outpoint["value"]
                         self.received_txs.append(tx)
                         self.outpoints.append(outpoint)
                 if self.amount_funded >= amount_to_pay:  # if fully funded
                     self.payment_received()
+                else:
+                    order_id = digest(json.dumps(self.contract, indent=4)).encode("hex")
+                    notification_json = {
+                        "notification": {
+                            "type": "partial payment",
+                            "amount": val,
+                            "order_id": order_id
+                        }
+                    }
+                    self.notification_listener.push_ws(notification_json)
 
         except Exception as e:
             self.log.critical("Error processing bitcoin transaction: %s" % e.message)

market/listeners.py

@@ -108,3 +108,6 @@ def notify(self, guid, handle, notif_type, order_id, title, image_hash):
             }
         }
         self.ws.push(json.dumps(sanitize_html(notification_json), indent=4))
+
+    def push_ws(self, json_obj):
+        self.ws.push(json.dumps(sanitize_html(json_obj), indent=4))

openbazaard.py

@@ -37,7 +37,6 @@
 from twisted.python import log, logfile
 from txws import WebSocketFactory
 
-
 def run(*args):
     TESTNET = args[0]
     LOGLEVEL = args[1]