push session and request

[electrum-server.git] / modules / python_bitcoin / __init__.py

diff --git a/modules/python_bitcoin/__init__.py b/modules/python_bitcoin/__init__.py
index c89d377..cd9a4db 100644 (file)
--- a/modules/python_bitcoin/__init__.py
+++ b/modules/python_bitcoin/__init__.py
@@ -106,7 +106,7 @@ class NumblocksSubscribe:
 
     def subscribe(self, session, request):
         last = self.latest.get()
-        session.push_response({"id": request["id"], "result": last})
+        self.push_response(session,{"id": request["id"], "result": last})
         with self.lock:
             self.subscribed.append((session, request))
 
@@ -121,7 +121,7 @@ class NumblocksSubscribe:
         self.latest.set(latest)
         subscribed = self.spring_clean()
         for session, request in subscribed:
-            session.push_response({"id": request["id"], "result": latest})
+            self.push_response(session,{"id": request["id"], "result": latest})
         self.backend.blockchain.subscribe_reorganize(self.reorganize)
 
     def spring_clean(self):
@@ -141,7 +141,7 @@ class AddressGetHistory:
             bitcoin.bind(self.respond, session, request, bitcoin._1))
 
     def respond(self, session, request, result):
-        session.push_response({"id": request["id"], "result": result})
+        self.push_response(session,{"id": request["id"], "result": result})
 
 class LibbitcoinProcessor(stratum.Processor):
 
@@ -154,15 +154,36 @@ class LibbitcoinProcessor(stratum.Processor):
     def stop(self):
         self.backend.stop()
 
-    def process(self, session):
-        request = session.pop_request()
+    def process(self, session, request):
+
         print "New request (lib)", request
         if request["method"] == "numblocks.subscribe":
             self.numblocks_subscribe.subscribe(session, request)
         elif request["method"] == "address.get_history":
             self.address_get_history.get(session, request)
+        elif request["method"] == "server.banner":
+            self.push_response(session, {"id": request["id"],
+                "result": "libbitcoin using python-bitcoin bindings"})
+        elif request["method"] == "transaction.broadcast":
+            self.broadcast_transaction(session, request)
         # Execute and when ready, you call
-        # session.push_response(response)
+        # self.push_response(session,response)
+
+    def broadcast_transaction(self, session, request):
+        raw_tx = bitcoin.data_chunk(str(request["params"]))
+        exporter = bitcoin.satoshi_exporter()
+        try:
+            tx = exporter.load_transaction(raw_tx)
+        except RuntimeError:
+            response = {"id": request["id"], "result": None,
+                "error": {"message": 
+                    "Exception while parsing the transaction data.",
+                    "code": -4}}
+        else:
+            self.backend.protocol.broadcast_transaction(tx)
+            tx_hash = str(bitcoin.hash_transaction(tx))
+            response = {"id": request["id"], "result": tx_hash}
+        self.push_response(session,response)
 
 def run(stratum):
     print "Warning: pre-alpha prototype. Full of bugs."