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