import urllib2,json
import MySQLdb as mdb
import Queue
-from electrum import Wallet, Interface
+from electrum import Wallet, Interface, WalletVerifier, SimpleConfig, WalletSynchronizer
import ConfigParser
config = ConfigParser.ConfigParser()
cb_expired = config.get('callback','expired')
cb_password = config.get('callback','password')
-wallet = Wallet()
-wallet.master_public_key = config.get('electrum','mpk')
+wallet_config = SimpleConfig()
+master_public_key = config.get('electrum','mpk')
+wallet_config.set_key('master_public_key',master_public_key)
+wallet = Wallet(wallet_config)
+wallet.synchronize = lambda: None # prevent address creation by the wallet
omg_addresses = {}
-def electrum_input_thread(in_queue):
+def input_reader_thread(request_queue):
while True:
- addr, amount = in_queue.get(True,1000000000)
+ addr, amount, confirmations = request_queue.get(True,1000000000)
if addr in omg_addresses:
continue
else:
print "subscribing to ", addr
- omg_addresses[addr] = amount
- interface.send([('blockchain.address.subscribe',[addr])])
+ omg_addresses[addr] = {'requested':float(amount), 'confirmations':int(confirmations)}
+ if addr not in wallet.addresses:
+ with wallet.lock:
+ print "adding %s to wallet"%addr
+ wallet.addresses.append(addr)
+ wallet.history[addr] = []
+ synchronizer.subscribe_to_addresses([addr])
+ wallet.up_to_date = False
-def electrum_output_thread(out_queue):
- while True:
- r = interface.responses.get(True, 100000000000)
- method = r.get('method')
-
- if method == 'blockchain.address.subscribe':
- addr = r.get('params')[0]
- interface.send([('blockchain.address.get_history',[addr])])
-
- elif method == 'blockchain.address.get_history':
- addr = r.get('params')[0]
- #print "received history for", addr
- confirmed = unconfirmed = 0
- h = r.get('result')
- if h is None:
- continue
- for item in h:
- v = item['value']
- if v<0: continue
- if item['height']:
- confirmed += v
- else:
- unconfirmed += v
-
- s = (confirmed+unconfirmed)/1.e8
- print "balance for %s:"%addr, s
- amount = float(omg_addresses.get(addr))
- if s>=amount:
- out_queue.put( ('payment',addr) )
+
+
+def on_wallet_update():
+ print "updated_callback"
+ for addr in omg_addresses:
+ h = wallet.history.get(addr)
+
+ requested_amount = omg_addresses[addr].get('requested')
+ requested_confs = omg_addresses[addr].get('confirmations')
+
+ value = 0
+ for tx_hash, tx_height in h:
+
+ tx = wallet.transactions.get(tx_hash)
+ if not tx: continue
+ if verifier.get_confirmations(tx_hash) < requested_confs: continue
+ for o in tx.get('outputs'):
+ if o.get('address') == addr:
+ value += o.get('value')
+
+ s = (value)/1.e8
+ print "balance for %s:"%addr, s, requested_amount
+ if s>= requested_amount:
+ print "payment accepted", addr
+ out_queue.put( ('payment', addr))
stopping = False
def do_create(conn):
# creation
cur = conn.cursor()
- cur.execute("CREATE TABLE electrum_payments (id INT PRIMARY KEY, address VARCHAR(40), amount FLOAT, received_at TIMESTAMP, expires_at TIMESTAMP, paid INT(1), processed INT(1));")
+ cur.execute("CREATE TABLE electrum_payments (id INT PRIMARY KEY, address VARCHAR(40), amount FLOAT, confirmations INT(8), received_at TIMESTAMP, expires_at TIMESTAMP, paid INT(1), processed INT(1));")
conn.commit()
def process_request(i, amount, confirmations, expires_in, password):
print "process_request", i, amount, confirmations, expires_in
if password!=my_password:
print "wrong password ", password
- return
+ return
addr = wallet.get_new_address(i, 0)
out_queue.put( ('request',(i,addr,amount,expires_in) ))
return addr
interface = Interface({'server':"%s:%d:t"%(electrum_server, 50001)})
interface.start()
+ interface.send([('blockchain.numblocks.subscribe',[])])
+
+ wallet.interface = interface
+ interface.register_callback('updated', on_wallet_update)
+
+ verifier = WalletVerifier(interface, wallet_config)
+ wallet.set_verifier(verifier)
+
+ synchronizer = WalletSynchronizer(wallet, wallet_config)
+ synchronizer.start()
- verifier = WalletVerifier(interface, config)
verifier.start()
# this process detects when addresses have paid
- in_queue = Queue.Queue()
+ request_queue = Queue.Queue()
out_queue = Queue.Queue()
- thread.start_new_thread(electrum_input_thread, (in_queue,))
- thread.start_new_thread(electrum_output_thread, (out_queue,))
-
+ thread.start_new_thread(input_reader_thread, (request_queue,))
thread.start_new_thread(server_thread, (conn,))
-
while not stopping:
cur = conn.cursor()
- # get a list of addresses to watch
- cur.execute("SELECT address, amount FROM electrum_payments WHERE paid IS NULL;")
+ # read pending requests from table
+ cur.execute("SELECT address, amount, confirmations FROM electrum_payments WHERE paid IS NULL;")
data = cur.fetchall()
for item in data:
- in_queue.put(item)
+ request_queue.put(item)
try:
cmd, params = out_queue.get(True, 10)
cmd = ''
if cmd == 'payment':
+ addr = params
# set paid=1 for received payments
print "received payment from", addr
cur.execute("select id from electrum_payments where address='%s';"%addr)
id = cur.fetchone()[0]
cur.execute("update electrum_payments set paid=1 where id=%d;"%(id))
+
elif cmd == 'request':
- i, addr, amount, hours = params
- sql = "INSERT INTO electrum_payments (id, address, amount, received_at, expires_at, paid, processed)"\
- + " VALUES (%d, '%s', %f, CURRENT_TIMESTAMP, ADDTIME(CURRENT_TIMESTAMP, '0 %d:0:0'), NULL, NULL);"%(i, addr, amount, hours)
+ # add a new request to the table.
+ i, addr, amount, confs, hours = params
+ sql = "INSERT INTO electrum_payments (id, address, amount, confirmations, received_at, expires_at, paid, processed)"\
+ + " VALUES (%d, '%s', %f, %d, CURRENT_TIMESTAMP, ADDTIME(CURRENT_TIMESTAMP, '0 %d:0:0'), NULL, NULL);"%(i, addr, amount, confs, hours)
cur.execute(sql)
conn.close()
+ print "terminated"
+
git://git.novaco.in / electrum-nvc.git / blobdiff