refactoring interfaces
This commit is contained in:
parent
1603dc50d0
commit
f3128b95f9
|
@ -45,6 +45,14 @@ class Interface:
|
||||||
self.disconnected_event = threading.Event()
|
self.disconnected_event = threading.Event()
|
||||||
self.disconnected_event.clear()
|
self.disconnected_event.clear()
|
||||||
|
|
||||||
|
#only asynchrnous
|
||||||
|
self.addresses_waiting_for_status = []
|
||||||
|
self.addresses_waiting_for_history = []
|
||||||
|
self.tx_event = threading.Event()
|
||||||
|
self.up_to_date_event = threading.Event()
|
||||||
|
self.up_to_date_event.clear()
|
||||||
|
|
||||||
|
|
||||||
def send_tx(self, data):
|
def send_tx(self, data):
|
||||||
out = self.handler('transaction.broadcast', data )
|
out = self.handler('transaction.broadcast', data )
|
||||||
return out
|
return out
|
||||||
|
@ -56,42 +64,94 @@ class Interface:
|
||||||
pass
|
pass
|
||||||
|
|
||||||
|
|
||||||
|
def handle_json_response(self, c):
|
||||||
|
#print c
|
||||||
|
msg_id = c.get('id')
|
||||||
|
result = c.get('result')
|
||||||
|
error = c.get('error')
|
||||||
|
if msg_id is None:
|
||||||
|
print "error: message without ID"
|
||||||
|
return
|
||||||
|
|
||||||
|
method, params = self.messages[msg_id]
|
||||||
|
if error:
|
||||||
|
print "received error:", c, method, params
|
||||||
|
else:
|
||||||
|
self.handle_response(method, params, result)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
def handle_response(self, method, params, result):
|
||||||
|
|
||||||
|
if method == 'session.new':
|
||||||
|
self.session_id, self.message = ast.literal_eval( result )
|
||||||
|
self.was_updated = True
|
||||||
|
|
||||||
|
elif method == 'server.banner':
|
||||||
|
self.message = result
|
||||||
|
self.was_updated = True
|
||||||
|
|
||||||
|
elif method == 'session.poll':
|
||||||
|
blocks, changed_addresses = ast.literal_eval( result )
|
||||||
|
if blocks == -1: raise BaseException("session not found")
|
||||||
|
self.blocks = int(blocks)
|
||||||
|
if changed_addresses:
|
||||||
|
self.is_up_to_date = False
|
||||||
|
self.was_updated = True
|
||||||
|
for addr, status in changed_addresses.items():
|
||||||
|
apply(self.address_callback, (addr, status))
|
||||||
|
else:
|
||||||
|
self.is_up_to_date = True
|
||||||
|
|
||||||
|
elif method == 'server.peers':
|
||||||
|
self.servers = map( lambda x:x[1], result )
|
||||||
|
|
||||||
|
elif method == 'address.subscribe':
|
||||||
|
addr = params[-1]
|
||||||
|
if addr in self.addresses_waiting_for_status:
|
||||||
|
self.addresses_waiting_for_status.remove(addr)
|
||||||
|
apply(self.address_callback,(addr, result))
|
||||||
|
|
||||||
|
elif method == 'address.get_history':
|
||||||
|
addr = params[0]
|
||||||
|
if addr in self.addresses_waiting_for_history:
|
||||||
|
self.addresses_waiting_for_history.remove(addr)
|
||||||
|
apply(self.history_callback, (addr, result))
|
||||||
|
self.was_updated = True
|
||||||
|
|
||||||
|
elif method == 'transaction.broadcast':
|
||||||
|
self.tx_result = result
|
||||||
|
self.tx_event.set()
|
||||||
|
|
||||||
|
elif method == 'numblocks.subscribe':
|
||||||
|
self.blocks = result
|
||||||
|
if self.newblock_callback: apply(self.newblock_callback,(result,))
|
||||||
|
else:
|
||||||
|
print "received message:", c, method, params
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class PollingInterface(Interface):
|
class PollingInterface(Interface):
|
||||||
""" non-persistent connection. synchronous calls"""
|
""" non-persistent connection. synchronous calls"""
|
||||||
|
|
||||||
def start_session(self, addresses, version):
|
def start_session(self, addresses, version):
|
||||||
out = self.handler('session.new', [ version, addresses ] )
|
self.handler([('session.new', [ version, addresses ])] )
|
||||||
self.session_id, self.message = ast.literal_eval( out )
|
|
||||||
thread.start_new_thread(self.poll_thread, ())
|
thread.start_new_thread(self.poll_thread, ())
|
||||||
|
|
||||||
def poll_interval(self):
|
def poll_interval(self):
|
||||||
return 5
|
return 5
|
||||||
|
|
||||||
def retrieve_history(self, address):
|
def get_history(self, address):
|
||||||
out = self.handler('address.get_history', address )
|
self.handler([('address.get_history', [address] )])
|
||||||
return out
|
|
||||||
|
|
||||||
def get_history(self, addr):
|
|
||||||
data = self.retrieve_history(addr)
|
|
||||||
apply(self.history_callback, (addr, data) )
|
|
||||||
self.was_updated = True
|
|
||||||
|
|
||||||
def subscribe(self, addresses):
|
def subscribe(self, addresses):
|
||||||
for addr in addresses:
|
for addr in addresses:
|
||||||
status = self.handler('address.subscribe', [ self.session_id, addr ] )
|
self.handler([('address.subscribe', [ self.session_id, addr ] )])
|
||||||
apply(self.address_callback, (addr, status) )
|
|
||||||
|
|
||||||
def update_wallet(self):
|
def update_wallet(self):
|
||||||
while True:
|
while True:
|
||||||
changed_addresses = self.poll()
|
self.handler([('session.poll', self.session_id )])
|
||||||
if changed_addresses:
|
if self.is_up_to_date: break
|
||||||
self.is_up_to_date = False
|
|
||||||
else:
|
|
||||||
self.is_up_to_date = True
|
|
||||||
break
|
|
||||||
|
|
||||||
for addr, status in changed_addresses.items():
|
|
||||||
apply(self.address_callback, (addr, status))
|
|
||||||
|
|
||||||
#if is_new or wallet.remote_url:
|
#if is_new or wallet.remote_url:
|
||||||
# self.was_updated = True
|
# self.was_updated = True
|
||||||
|
@ -102,13 +162,6 @@ class PollingInterface(Interface):
|
||||||
#else:
|
#else:
|
||||||
# return False
|
# return False
|
||||||
|
|
||||||
def poll(self):
|
|
||||||
out = self.handler('session.poll', self.session_id )
|
|
||||||
blocks, changed_addr = ast.literal_eval( out )
|
|
||||||
if blocks == -1: raise BaseException("session not found")
|
|
||||||
self.blocks = int(blocks)
|
|
||||||
return changed_addr
|
|
||||||
|
|
||||||
def poll_thread(self):
|
def poll_thread(self):
|
||||||
while self.is_connected:
|
while self.is_connected:
|
||||||
try:
|
try:
|
||||||
|
@ -136,8 +189,7 @@ class PollingInterface(Interface):
|
||||||
for server in DEFAULT_SERVERS:
|
for server in DEFAULT_SERVERS:
|
||||||
try:
|
try:
|
||||||
self.peers_server = server
|
self.peers_server = server
|
||||||
out = self.handler('server.peers')
|
self.handler([('server.peers',[])])
|
||||||
self.servers = map( lambda x:x[1], out )
|
|
||||||
# print "Received server list from %s" % self.peers_server, out
|
# print "Received server list from %s" % self.peers_server, out
|
||||||
break
|
break
|
||||||
except socket.timeout:
|
except socket.timeout:
|
||||||
|
@ -150,9 +202,12 @@ class PollingInterface(Interface):
|
||||||
time.sleep(5*60)
|
time.sleep(5*60)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class NativeInterface(PollingInterface):
|
class NativeInterface(PollingInterface):
|
||||||
|
|
||||||
def handler(self, method, params = ''):
|
def handler(self, messages):
|
||||||
import time
|
import time
|
||||||
cmds = {'session.new':'new_session',
|
cmds = {'session.new':'new_session',
|
||||||
'server.peers':'peers',
|
'server.peers':'peers',
|
||||||
|
@ -161,45 +216,72 @@ class NativeInterface(PollingInterface):
|
||||||
'address.get_history':'h',
|
'address.get_history':'h',
|
||||||
'address.subscribe':'address.subscribe'
|
'address.subscribe':'address.subscribe'
|
||||||
}
|
}
|
||||||
cmd = cmds[method]
|
|
||||||
if type(params) != type(''): params = repr( params )
|
for m in messages:
|
||||||
t1 = time.time()
|
method, params = m
|
||||||
request = repr ( (cmd, params) ) + "#"
|
cmd = cmds[method]
|
||||||
s = socket.socket( socket.AF_INET, socket.SOCK_STREAM)
|
|
||||||
s.settimeout(DEFAULT_TIMEOUT)
|
if cmd=='h':
|
||||||
s.connect(( self.host if cmd!='peers' else self.peers_server, self.port) )
|
str_params = params[0]
|
||||||
s.send( request )
|
elif type(params) != type(''):
|
||||||
out = ''
|
str_params = repr( params )
|
||||||
while 1:
|
else:
|
||||||
msg = s.recv(1024)
|
str_params = params
|
||||||
if msg: out += msg
|
t1 = time.time()
|
||||||
else: break
|
request = repr ( (cmd, str_params) ) + "#"
|
||||||
s.close()
|
s = socket.socket( socket.AF_INET, socket.SOCK_STREAM)
|
||||||
self.rtime = time.time() - t1
|
s.settimeout(DEFAULT_TIMEOUT)
|
||||||
self.is_connected = True
|
s.connect(( self.host if cmd!='peers' else self.peers_server, self.port) )
|
||||||
if cmd in[ 'peers','h']:
|
s.send( request )
|
||||||
out = ast.literal_eval( out )
|
out = ''
|
||||||
return out
|
while 1:
|
||||||
|
msg = s.recv(1024)
|
||||||
|
if msg: out += msg
|
||||||
|
else: break
|
||||||
|
s.close()
|
||||||
|
self.rtime = time.time() - t1
|
||||||
|
self.is_connected = True
|
||||||
|
if cmd in[ 'peers','h']:
|
||||||
|
out = ast.literal_eval( out )
|
||||||
|
|
||||||
|
if out=='': out=None #fixme
|
||||||
|
|
||||||
|
self.handle_response(method, params, out)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
class HttpInterface(PollingInterface):
|
class HttpInterface(PollingInterface):
|
||||||
|
|
||||||
def handler(self, method, params = []):
|
def __init__(self, host, port, address_callback=None, history_callback=None, newblock_callback=None):
|
||||||
|
Interface.__init__(self, host, port, address_callback, history_callback, newblock_callback)
|
||||||
|
self.message_id = 0
|
||||||
|
self.messages = {}
|
||||||
|
|
||||||
|
def handler(self, messages):
|
||||||
import urllib2, json, time
|
import urllib2, json, time
|
||||||
if type(params) != type([]): params = [ params ]
|
|
||||||
t1 = time.time()
|
data = []
|
||||||
data = { 'method':method, 'id':'jsonrpc', 'params':params }
|
for m in messages:
|
||||||
|
method, params = m
|
||||||
|
if type(params) != type([]): params = [params]
|
||||||
|
t1 = time.time()
|
||||||
|
data.append( { 'method':method, 'id':self.message_id, 'params':params } )
|
||||||
|
self.messages[self.message_id] = (method, params)
|
||||||
|
self.message_id += 1
|
||||||
|
|
||||||
data_json = json.dumps(data)
|
data_json = json.dumps(data)
|
||||||
host = 'http://%s:%d'%( self.host if method!='server.peers' else self.peers_server, self.port )
|
host = 'http://%s:%d'%( self.host if method!='server.peers' else self.peers_server, self.port )
|
||||||
req = urllib2.Request(host, data_json, {'content-type': 'application/json'})
|
req = urllib2.Request(host, data_json, {'content-type': 'application/json'})
|
||||||
response_stream = urllib2.urlopen(req)
|
response_stream = urllib2.urlopen(req)
|
||||||
response = json.loads( response_stream.read() )
|
response = json.loads( response_stream.read() )
|
||||||
out = response.get('result')
|
|
||||||
if not out:
|
|
||||||
print response
|
|
||||||
self.rtime = time.time() - t1
|
self.rtime = time.time() - t1
|
||||||
self.is_connected = True
|
self.is_connected = True
|
||||||
return out
|
|
||||||
|
for item in response:
|
||||||
|
self.handle_json_response(item)
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
@ -214,13 +296,6 @@ class AsynchronousInterface(Interface):
|
||||||
self.message_id = 0
|
self.message_id = 0
|
||||||
self.messages = {}
|
self.messages = {}
|
||||||
|
|
||||||
self.tx_event = threading.Event()
|
|
||||||
self.addresses_waiting_for_status = []
|
|
||||||
self.addresses_waiting_for_history = []
|
|
||||||
# up to date
|
|
||||||
self.up_to_date_event = threading.Event()
|
|
||||||
self.up_to_date_event.clear()
|
|
||||||
|
|
||||||
def listen_thread(self):
|
def listen_thread(self):
|
||||||
try:
|
try:
|
||||||
self.is_connected = True
|
self.is_connected = True
|
||||||
|
@ -239,53 +314,13 @@ class AsynchronousInterface(Interface):
|
||||||
c = out[0:s]
|
c = out[0:s]
|
||||||
out = out[s+1:]
|
out = out[s+1:]
|
||||||
c = json.loads(c)
|
c = json.loads(c)
|
||||||
|
self.handle_json_response(c)
|
||||||
#print c
|
|
||||||
msg_id = c.get('id')
|
|
||||||
result = c.get('result')
|
|
||||||
error = c.get('error')
|
|
||||||
|
|
||||||
if msg_id is None:
|
|
||||||
print "error: message without ID"
|
|
||||||
continue
|
|
||||||
|
|
||||||
method, params = self.messages[msg_id]
|
|
||||||
|
|
||||||
if method == 'server.banner':
|
|
||||||
self.message = result
|
|
||||||
self.was_updated = True
|
|
||||||
|
|
||||||
elif method == 'server.peers':
|
|
||||||
self.servers = map( lambda x:x[1], result )
|
|
||||||
|
|
||||||
elif method == 'address.subscribe':
|
|
||||||
addr = params[0]
|
|
||||||
if addr in self.addresses_waiting_for_status:
|
|
||||||
self.addresses_waiting_for_status.remove(addr)
|
|
||||||
apply(self.address_callback,(addr, result))
|
|
||||||
|
|
||||||
elif method == 'address.get_history':
|
|
||||||
addr = params[0]
|
|
||||||
if addr in self.addresses_waiting_for_history:
|
|
||||||
self.addresses_waiting_for_history.remove(addr)
|
|
||||||
apply(self.history_callback, (addr, result))
|
|
||||||
self.was_updated = True
|
|
||||||
|
|
||||||
elif method == 'transaction.broadcast':
|
|
||||||
self.tx_result = result
|
|
||||||
self.tx_event.set()
|
|
||||||
|
|
||||||
elif method == 'numblocks.subscribe':
|
|
||||||
self.blocks = result
|
|
||||||
if self.newblock_callback: apply(self.newblock_callback,(result,))
|
|
||||||
else:
|
|
||||||
print "received message:", c
|
|
||||||
|
|
||||||
if self.addresses_waiting_for_status or self.addresses_waiting_for_history:
|
if self.addresses_waiting_for_status or self.addresses_waiting_for_history:
|
||||||
self.is_up_to_date = False
|
self.is_up_to_date = False
|
||||||
else:
|
else:
|
||||||
self.is_up_to_date = True
|
self.is_up_to_date = True
|
||||||
self.up_to_date_event.set()
|
self.up_to_date_event.set()
|
||||||
|
|
||||||
except:
|
except:
|
||||||
traceback.print_exc(file=sys.stdout)
|
traceback.print_exc(file=sys.stdout)
|
||||||
|
|
||||||
|
@ -336,6 +371,7 @@ class AsynchronousInterface(Interface):
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
def new_interface(wallet):
|
def new_interface(wallet):
|
||||||
if wallet.host:
|
if wallet.host:
|
||||||
host = wallet.host
|
host = wallet.host
|
||||||
|
|
|
@ -701,12 +701,12 @@ class Wallet:
|
||||||
|
|
||||||
def receive_status_callback(self, addr, status):
|
def receive_status_callback(self, addr, status):
|
||||||
if self.status.get(addr) != status:
|
if self.status.get(addr) != status:
|
||||||
#print "updating status for", addr
|
#print "updating status for", addr, repr(self.status.get(addr)), repr(status)
|
||||||
self.status[addr] = status
|
self.status[addr] = status
|
||||||
self.interface.get_history(addr)
|
self.interface.get_history(addr)
|
||||||
|
|
||||||
def receive_history_callback(self, addr, data):
|
def receive_history_callback(self, addr, data):
|
||||||
#print "updating history for", addr
|
#print "updating history for", addr, repr(data)
|
||||||
self.history[addr] = data
|
self.history[addr] = data
|
||||||
self.synchronize()
|
self.synchronize()
|
||||||
self.update_tx_history()
|
self.update_tx_history()
|
||||||
|
|
Loading…
Reference in New Issue