-

cool-RR · cool-RR · commit 3e48e347ba7d · 2013-09-02T03:09:38.000+03:00
diff --git a/python_toolbox/pickle_tools.py b/python_toolbox/pickle_tools.py
@@ -4,6 +4,7 @@
 '''Defines various tools for pickling and unpickling.'''
 
 
+import zlib
 import re
 import cPickle as pickle_module
 import pickle # Importing just to get dispatch table, not pickling with it.
@@ -186,4 +187,8 @@ def persistent_load(self, id_string):
             raise pickle_module.UnpicklingError('Invalid persistent id')
  
  
-    
+def pickle_and_compress(thing):
+    return zlib.compress(pickle_module.dumps(thing, protocol=2))
+
+def decompress_and_unpickle(thing):
+    return pickle_module.loads(zlib.decompress(thing))
diff --git a/test_python_toolbox/test_pickle_tools/test_compressing.py b/test_python_toolbox/test_pickle_tools/test_compressing.py
@@ -0,0 +1,29 @@
+# Copyright 2009-2013 Ram Rachum.
+# This program is distributed under the MIT license.
+
+# We're importing `pickle_module` from `pickle_tools`, so we get the exact same
+# pickle module it's using. (Giving it the freedom to change between `cPickle`
+# and `pickle`.)
+from python_toolbox.pickle_tools import pickle_module
+
+import nose
+
+from python_toolbox import import_tools
+
+from python_toolbox import pickle_tools
+    
+
+my_messy_object = (
+    'Whatever',
+    {1: 2,}, 
+    set([3, 4]), 
+    frozenset([3, 4]),
+    ((((((((((((())))))))))))),
+    u'unicode_too',
+    (((((3, 4, 5j)))))
+)
+
+def test():
+    pickled_and_compressed = pickle_tools.pickle_and_compress(my_messy_object)
+    assert pickle_tools.decompress_and_unpickle(pickled_and_compressed) == \
+                                                                my_messy_object
