refactor: cleanup graph

This commit is contained in:
Swann
2019-07-23 10:38:14 +02:00
parent 9a7a42d03a
commit 3d04765543
2 changed files with 6 additions and 5 deletions

View File

@ -102,6 +102,9 @@ class ClientNetService(threading.Thread):
self.state = STATE_INITIAL self.state = STATE_INITIAL
def connect(self,address='127.0.0.1', port=5560): def connect(self,address='127.0.0.1', port=5560):
"""
Network socket setup
"""
if self.state == STATE_INITIAL: if self.state == STATE_INITIAL:
logger.debug("connecting on {}:{}".format(address,port)) logger.debug("connecting on {}:{}".format(address,port))
self.snapshot = self.context.socket(zmq.DEALER) self.snapshot = self.context.socket(zmq.DEALER)

View File

@ -10,13 +10,13 @@ class ReplicationGraph(collections.MutableMapping):
self.update(dict(*args, **kwargs)) # use the free update to set keys self.update(dict(*args, **kwargs)) # use the free update to set keys
def __getitem__(self, key): def __getitem__(self, key):
return self.store[self.__keytransform__(key)] return self.store[key]
def __setitem__(self, key, value): def __setitem__(self, key, value):
self.store[self.__keytransform__(key)] = value self.store[key] = value
def __delitem__(self, key): def __delitem__(self, key):
del self.store[self.__keytransform__(key)] del self.store[key]
def __iter__(self): def __iter__(self):
return iter(self.store) return iter(self.store)
@ -24,7 +24,5 @@ class ReplicationGraph(collections.MutableMapping):
def __len__(self): def __len__(self):
return len(self.store) return len(self.store)
def __keytransform__(self, key):
return key