From 6df6e6adfeaf553625a75a91b064d06a44b954e3 Mon Sep 17 00:00:00 2001
From: Daniele Varrazzo <daniele.varrazzo@gmail.com>
Date: Tue, 11 Dec 2012 01:10:01 +0000
Subject: Fixed pickling of DictRow objects too

---
 lib/extras.py | 9 ++++++++-
 1 file changed, 8 insertions(+), 1 deletion(-)

(limited to 'lib/extras.py')

diff --git a/lib/extras.py b/lib/extras.py
index 0c4df56..1924908 100644
--- a/lib/extras.py
+++ b/lib/extras.py
@@ -184,7 +184,14 @@ class DictRow(list):
     def __contains__(self, x):
         return x in self._index
 
-    # grop the crusty Py2 methods
+    def __getstate__(self):
+        return self[:], self._index.copy()
+
+    def __setstate__(self, data):
+        self[:] = data[0]
+        self._index = data[1]
+
+    # drop the crusty Py2 methods
     if sys.version_info[0] > 2:
         items = iteritems; del iteritems
         keys = iterkeys; del iterkeys
-- 
cgit v1.2.1