Fix pickling ddrescue State()

This commit is contained in:
2Shirt 2023-07-16 18:07:08 -07:00
parent bddf47816f
commit 24e4f7ddcc
Signed by: 2Shirt
GPG key ID: 152FAC923B0E132C

View file

@ -171,6 +171,12 @@ class BlockPair():
# Set initial status
self.set_initial_status()
def __getstate__(self):
"""Override to allow pickling ddrescue.State() objects."""
bp_state = self.__dict__.copy()
del bp_state['view_proc']
return bp_state
def get_error_size(self) -> int:
"""Get error size in bytes, returns int."""
return self.size - self.get_rescued_size()