3 changed files with 81 additions and 136 deletions
@ -0,0 +1,71 @@ |
|||
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 |
Loading…
Reference in new issue