Fixed pickling of RealDictRow objects

1 files changed, 7 insertions, 0 deletions

diff --git a/lib/extras.py b/lib/extras.py
index 22ae3f6..0c4df56 100644
--- a/lib/extras.py
+++ b/lib/extras.py
@@ -245,6 +245,13 @@ class RealDictRow(dict):
             name = self._column_mapping[name]
         return dict.__setitem__(self, name, value)
 
+    def __getstate__(self):
+        return (self.copy(), self._column_mapping[:])
+
+    def __setstate__(self, data):
+        self.update(data[0])
+        self._column_mapping = data[1]
+
 
 class NamedTupleConnection(_connection):
     """A connection that uses `NamedTupleCursor` automatically."""