You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
142 lines
4.1 KiB
142 lines
4.1 KiB
import interface |
|
from blockchain import Blockchain |
|
import threading, time, Queue, os, sys, shutil |
|
from util import user_dir, appdata_dir, print_error |
|
from bitcoin import * |
|
|
|
|
|
class Network(threading.Thread): |
|
|
|
def __init__(self, config): |
|
threading.Thread.__init__(self) |
|
self.daemon = True |
|
self.config = config |
|
self.lock = threading.Lock() |
|
self.blockchain = Blockchain(config) |
|
self.interfaces = {} |
|
self.queue = Queue.Queue() |
|
self.default_server = self.config.get('server') |
|
self.servers_list = interface.filter_protocol(interface.DEFAULT_SERVERS,'s') |
|
self.callbacks = {} |
|
|
|
|
|
def register_callback(self, event, callback): |
|
with self.lock: |
|
if not self.callbacks.get(event): |
|
self.callbacks[event] = [] |
|
self.callbacks[event].append(callback) |
|
|
|
|
|
def trigger_callback(self, event): |
|
with self.lock: |
|
callbacks = self.callbacks.get(event,[])[:] |
|
if callbacks: |
|
[callback() for callback in callbacks] |
|
|
|
|
|
def start_interfaces(self): |
|
|
|
for server in self.servers_list: |
|
self.interfaces[server] = interface.Interface({'server':server}) |
|
|
|
for i in self.interfaces.values(): |
|
i.network = self # fixme |
|
i.start(self.queue) |
|
|
|
if self.default_server: |
|
self.interface = interface.Interface({'server':self.default_server}) |
|
self.interface.network = self # fixme |
|
self.interface.start(self.queue) |
|
else: |
|
self.interface = self.interfaces[0] |
|
|
|
|
|
def start(self, wait=False): |
|
|
|
self.start_interfaces() |
|
threading.Thread.__init__(self) |
|
if wait: |
|
self.interface.connect_event.wait() |
|
return self.interface.is_connected |
|
|
|
|
|
|
|
def run(self): |
|
self.blockchain.start() |
|
|
|
with self.lock: |
|
self.running = True |
|
|
|
while self.is_running(): |
|
i = self.queue.get() |
|
|
|
if i.is_connected: |
|
i.register_channel('verifier', self.blockchain.queue) |
|
i.register_channel('get_header') |
|
i.send([ ('blockchain.headers.subscribe',[])], 'verifier') |
|
if i == self.interface: |
|
i.send([('server.banner',[])]) |
|
i.send([('server.peers.subscribe',[])]) |
|
else: |
|
self.interfaces.pop(i.server) |
|
if i == self.interface: |
|
if self.default_server is None: |
|
print_msg("Using random server...") |
|
server = random.choice( self.servers_list ) |
|
self.interface = interface.Interface({'server':self.default_server}) |
|
else: |
|
#i.trigger_callback('disconnected') |
|
pass |
|
|
|
def on_peers(self, resut): |
|
pass |
|
|
|
def on_banner(self, result): |
|
pass |
|
|
|
def stop(self): |
|
with self.lock: self.running = False |
|
|
|
def is_running(self): |
|
with self.lock: return self.running |
|
|
|
|
|
def resend_subscriptions(self): |
|
for channel, messages in self.subscriptions.items(): |
|
if messages: |
|
self.send(messages, channel) |
|
|
|
|
|
def auto_cycle(self): |
|
if not self.is_connected and self.config.get('auto_cycle'): |
|
print_msg("Using random server...") |
|
servers = filter_protocol(DEFAULT_SERVERS, 's') |
|
while servers: |
|
server = random.choice( servers ) |
|
servers.remove(server) |
|
print server |
|
self.config.set_key('server', server, False) |
|
self.init_with_server(self.config) |
|
if self.is_connected: break |
|
|
|
if not self.is_connected: |
|
print 'no server available' |
|
self.connect_event.set() # to finish start |
|
self.server = 'ecdsa.org:50001:t' |
|
self.proxy = None |
|
return |
|
|
|
|
|
|
|
|
|
if __name__ == "__main__": |
|
import simple_config |
|
config = simple_config.SimpleConfig({'verbose':True}) |
|
network = Network(config) |
|
network.start() |
|
|
|
while 1: |
|
time.sleep(1) |
|
|
|
|
|
|
|
|