72 lines
2.0 KiB
Python
72 lines
2.0 KiB
Python
import time, electrum, Queue
|
|
from electrum import Interface, SimpleConfig
|
|
from electrum.network import filter_protocol, parse_servers
|
|
|
|
# electrum.util.set_verbosity(1)
|
|
|
|
def get_peers():
|
|
# 1. start interface and wait for connection
|
|
interface = electrum.Interface('ecdsa.net:110:s')
|
|
q = Queue.Queue()
|
|
interface.start(q)
|
|
i, r = q.get()
|
|
if not interface.is_connected:
|
|
raise BaseException("not connected")
|
|
# 2. get list of peers
|
|
interface.send_request({'id':0, 'method':'server.peers.subscribe','params':[]})
|
|
i, r = q.get(timeout=10000)
|
|
peers = parse_servers(r.get('result'))
|
|
peers = filter_protocol(peers,'s')
|
|
i.stop()
|
|
return peers
|
|
|
|
def send_request(peers, request):
|
|
print "Contacting %d servers"%len(peers)
|
|
# start interfaces
|
|
q2 = Queue.Queue()
|
|
config = SimpleConfig()
|
|
interfaces = map ( lambda server: Interface(server, config), peers )
|
|
reached_servers = []
|
|
for i in interfaces:
|
|
i.start(q2)
|
|
t0 = time.time()
|
|
while peers:
|
|
try:
|
|
i, r = q2.get(timeout=1)
|
|
except:
|
|
if time.time() - t0 > 10:
|
|
print "timeout"
|
|
break
|
|
else:
|
|
continue
|
|
if i.server in peers:
|
|
peers.remove(i.server)
|
|
if i.is_connected:
|
|
reached_servers.append(i)
|
|
else:
|
|
print "Connection failed:", i.server
|
|
|
|
print "%d servers could be reached"%len(reached_servers)
|
|
|
|
results_queue = Queue.Queue()
|
|
for i in reached_servers:
|
|
i.send_request(request, results_queue)
|
|
results = {}
|
|
t0 = time.time()
|
|
while reached_servers:
|
|
try:
|
|
i, r = results_queue.get(timeout=1)
|
|
except:
|
|
if time.time() - t0 > 10:
|
|
break
|
|
else:
|
|
continue
|
|
results[i.server] = r.get('result')
|
|
reached_servers.remove(i)
|
|
i.stop()
|
|
|
|
for i in reached_servers:
|
|
print i.server, "did not answer"
|
|
print "%d answers"%len(results)
|
|
return results
|