feat: unregister key

feat: _repr_
This commit is contained in:
Swann
2019-07-23 11:34:43 +02:00
parent 3d04765543
commit 78f051a231
4 changed files with 81 additions and 6 deletions

View File

@ -31,6 +31,8 @@ class ReplicatedDataFactory(object):
# Default registered types
self.register_type(str,RepCommand)
self.register_type(RepDeleteCommand, RepDeleteCommand)
def register_type(self, dtype, implementation):
"""
@ -180,7 +182,12 @@ class ReplicatedDatablock(object):
"""
raise NotImplementedError
def __repr__(self):
return "{uuid} - Owner: {owner} - ETA: {state} ".format(
uuid=self.uuid,
owner=self.owner,
state=self.state
)
class RepCommand(ReplicatedDatablock):
def serialize(self,data):
@ -192,6 +199,21 @@ class RepCommand(ReplicatedDatablock):
def load(self,target):
target = self.pointer
class RepDeleteCommand(ReplicatedDatablock):
def serialize(self,data):
return pickle.dumps(data)
def deserialize(self,data):
return pickle.loads(data)
def store(self,rep_store):
assert(self.buffer)
if rep_store and self.buffer in rep_store.keys():
del rep_store[self.buffer]
# class RepObject(ReplicatedDatablock):
# def deserialize(self):
# try: