refactor: cleanup graph

replication_client.py

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

replication_graph.py

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