From 24e4f7ddcc44d49f297c1c3aa3a7ef6c4d375361 Mon Sep 17 00:00:00 2001
From: 2Shirt <2xShirt@gmail.com>
Date: Sun, 16 Jul 2023 18:07:08 -0700
Subject: [PATCH] Fix pickling ddrescue State()

---
 scripts/wk/clone/ddrescue.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/scripts/wk/clone/ddrescue.py b/scripts/wk/clone/ddrescue.py
index c2adb4ae..3bff9f10 100644
--- a/scripts/wk/clone/ddrescue.py
+++ b/scripts/wk/clone/ddrescue.py
@@ -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()