add merchant script
This commit is contained in:
parent
e70c20803f
commit
6bf18d8607
|
@ -0,0 +1,19 @@
|
|||
[main]
|
||||
host = hostname of the machine where you run this program
|
||||
port = choose a port number
|
||||
password = choose a password
|
||||
|
||||
[db]
|
||||
instance = the hostname of your sql server
|
||||
name = the database name
|
||||
user = your database username
|
||||
password = your database password
|
||||
|
||||
[electrum]
|
||||
server = the electrum server you will use
|
||||
mpk = the master public key of your wallet (in hexadecimal)
|
||||
|
||||
[callback]
|
||||
received = URL where we POST json data when payment has been received
|
||||
expired = URL where we POST json data if payment has expired
|
||||
password = password sent in the json data, for authentication
|
|
@ -0,0 +1,228 @@
|
|||
#!/usr/bin/env python
|
||||
#
|
||||
# Electrum - lightweight Bitcoin client
|
||||
# Copyright (C) 2011 thomasv@gitorious
|
||||
#
|
||||
# This program is free software: you can redistribute it and/or modify
|
||||
# it under the terms of the GNU General Public License as published by
|
||||
# the Free Software Foundation, either version 3 of the License, or
|
||||
# (at your option) any later version.
|
||||
#
|
||||
# This program is distributed in the hope that it will be useful,
|
||||
# but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||||
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||||
# GNU General Public License for more details.
|
||||
#
|
||||
# You should have received a copy of the GNU General Public License
|
||||
# along with this program. If not, see <http://www.gnu.org/licenses/>.
|
||||
|
||||
|
||||
import time, thread, sys, socket, os
|
||||
import urllib2,json
|
||||
import MySQLdb as mdb
|
||||
import Queue
|
||||
from electrum import Wallet, TcpStratumInterface
|
||||
|
||||
import ConfigParser
|
||||
config = ConfigParser.ConfigParser()
|
||||
config.read("merchant.conf")
|
||||
|
||||
db_instance = config.get('db','instance')
|
||||
db_user = config.get('db','user')
|
||||
db_password = config.get('db','password')
|
||||
db_name = config.get('db','name')
|
||||
|
||||
electrum_server = config.get('electrum','server')
|
||||
mpk = config.get('electrum','mpk')
|
||||
|
||||
my_password = config.get('main','password')
|
||||
my_host = config.get('main','host')
|
||||
my_port = config.getint('main','port')
|
||||
|
||||
cb_received = config.get('callback','received')
|
||||
cb_expired = config.get('callback','expired')
|
||||
cb_password = config.get('callback','password')
|
||||
|
||||
wallet = Wallet()
|
||||
wallet.master_public_key = mpk.decode('hex')
|
||||
|
||||
|
||||
|
||||
omg_addresses = {}
|
||||
|
||||
def electrum_input_thread(in_queue, i):
|
||||
while True:
|
||||
addr, amount = in_queue.get(True,1000000000)
|
||||
if addr in omg_addresses:
|
||||
continue
|
||||
else:
|
||||
print "subscribing to ", addr
|
||||
omg_addresses[addr] = amount
|
||||
i.send([('blockchain.address.subscribe',[addr])])
|
||||
|
||||
|
||||
def electrum_output_thread(out_queue, i):
|
||||
while True:
|
||||
r = i.responses.get(True, 100000000000)
|
||||
method = r.get('method')
|
||||
|
||||
if method == 'blockchain.address.subscribe':
|
||||
addr = r.get('params')[0]
|
||||
i.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) )
|
||||
|
||||
|
||||
stopping = False
|
||||
|
||||
def do_stop():
|
||||
global stopping
|
||||
stopping = True
|
||||
|
||||
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));")
|
||||
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
|
||||
addr = wallet.get_new_address(i, 0)
|
||||
out_queue.put( ('request',(i,addr,amount,expires_in) ))
|
||||
return addr
|
||||
|
||||
def get_mpk():
|
||||
return wallet.master_public_key.encode('hex')
|
||||
|
||||
|
||||
def server_thread(conn):
|
||||
from jsonrpclib.SimpleJSONRPCServer import SimpleJSONRPCServer
|
||||
server = SimpleJSONRPCServer(( my_host, my_port))
|
||||
server.register_function(process_request, 'request')
|
||||
server.register_function(get_mpk, 'mpk')
|
||||
server.register_function(do_stop, 'stop')
|
||||
server.serve_forever()
|
||||
|
||||
|
||||
def handle_command(cmd):
|
||||
import jsonrpclib
|
||||
server = jsonrpclib.Server('http://%s:%d'%(my_host, my_port))
|
||||
try:
|
||||
if cmd == 'mpk':
|
||||
out = server.mpk()
|
||||
elif cmd == 'stop':
|
||||
out = server.stop()
|
||||
elif cmd == 'create':
|
||||
conn = mdb.connect(db_instance, db_user, db_password, db_name);
|
||||
do_create(conn)
|
||||
out = "ok"
|
||||
else:
|
||||
out = "unknown command"
|
||||
except socket.error:
|
||||
print "Server not running"
|
||||
return 1
|
||||
|
||||
print out
|
||||
return 0
|
||||
|
||||
|
||||
if __name__ == '__main__':
|
||||
|
||||
if len(sys.argv) > 1:
|
||||
ret = handle_command(sys.argv[1])
|
||||
sys.exit(ret)
|
||||
|
||||
print "using database", db_name
|
||||
conn = mdb.connect(db_instance, db_user, db_password, db_name);
|
||||
|
||||
i = TcpStratumInterface(electrum_server, 50001)
|
||||
i.init_socket()
|
||||
i.start()
|
||||
|
||||
|
||||
# this process detects when addresses have paid
|
||||
in_queue = Queue.Queue()
|
||||
out_queue = Queue.Queue()
|
||||
thread.start_new_thread(electrum_input_thread, (in_queue,i))
|
||||
thread.start_new_thread(electrum_output_thread, (out_queue,i))
|
||||
|
||||
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;")
|
||||
data = cur.fetchall()
|
||||
for item in data:
|
||||
in_queue.put(item)
|
||||
|
||||
try:
|
||||
cmd, params = out_queue.get(True, 10)
|
||||
except Queue.Empty:
|
||||
cmd = ''
|
||||
|
||||
if cmd == 'payment':
|
||||
# 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)
|
||||
cur.execute(sql)
|
||||
|
||||
|
||||
# set paid=0 for expired requests
|
||||
cur.execute("""UPDATE electrum_payments set paid=0 WHERE expires_at < CURRENT_TIMESTAMP and paid is NULL;""")
|
||||
|
||||
# do callback for addresses that received payment
|
||||
cur.execute("""SELECT id, address, paid from electrum_payments WHERE paid is not NULL and processed is NULL;""")
|
||||
data = cur.fetchall()
|
||||
for item in data:
|
||||
print "callback:", item
|
||||
id = int(item[0])
|
||||
address = item[1]
|
||||
paid = int(item[2])
|
||||
headers = {'content-type':'application/json'}
|
||||
data_json = { 'id':id, 'address':address, 'btc_auth':cb_password }
|
||||
data_json = json.dumps(data_json)
|
||||
url = cb_received if paid else cb_expired
|
||||
req = urllib2.Request(url, data_json, headers)
|
||||
try:
|
||||
response_stream = urllib2.urlopen(req)
|
||||
cur.execute("UPDATE electrum_payments SET processed=1 WHERE id=%d;"%(id))
|
||||
except urllib2.HTTPError:
|
||||
print "cannot do callback", data_json
|
||||
|
||||
conn.commit()
|
||||
|
||||
|
||||
conn.close()
|
||||
|
||||
|
Loading…
Reference in New Issue