Ticket #10184: querydict-pickle.diff

django/http/__init__.py

@@ -190 +190 @@
             dict.__setitem__(result, copy.deepcopy(key, memo), copy.deepcopy(value, memo))
         return result
 
+    def __getstate__(self):
+        obj_dict = super(QueryDict, self).__getstate__()
+        obj_dict['_mutable'] = self._mutable
+        return obj_dict
+
+    def __setstate__(self, obj_dict):
+        data = obj_dict.pop('_data', [])
+        for k, v in data.items():
+            self.setlist(k, v)
+        self.__dict__.update(obj_dict)
+        
     def setlist(self, key, list_):
         self._assert_mutable()
         key = str_to_unicode(key, self.encoding)

django/utils/datastructures.py

@@ -222 +222 @@
             dict.__setitem__(result, copy.deepcopy(key, memo),
                              copy.deepcopy(value, memo))
         return result
-
+    
+    def __getstate__(self):
+        obj_dict = self.__dict__.copy()
+        obj_dict['_data'] = dict([(k, self.getlist(k)) for k in self])
+        return obj_dict
+    
+    def __setstate__(self, obj_dict):
+        data = obj_dict.pop('_data', [])
+        for k, v in data.items():
+            self.setlist(k, v)
+        self.__dict__.update(obj_dict)
+        
     def get(self, key, default=None):
         """
         Returns the last data value for the passed key. If key doesn't exist

tests/regressiontests/httpwrappers/tests.py

@@ -400 +400 @@
 >>> q1 = pickle.loads(pickle.dumps(q, 2))
 >>> q == q1
 True
+>>> q = QueryDict('a=b&c=d&a=1') 
+>>> q1 = pickle.loads(pickle.dumps(q, 2))
+>>> q == q1 
+True
 
 ######################################
 # HttpResponse with Unicode headers  #