replace serializer

Author: opapy

django_elastipymemcache/memcached.py

@@ -5,6 +5,11 @@
 import socket
 from functools import wraps
 
+try:
+    import cPickle as pickle
+except ImportError:
+    import pickle
+
 from django.core.cache import InvalidCacheBackendError
 from django.core.cache.backends.memcached import BaseMemcachedCache
 
@@ -15,6 +20,19 @@
 logger = logging.getLogger(__name__)
 
 
+def serialize_pickle(key, value):
+    if isinstance(value, str):
+        return value, 1
+    return pickle.dumps(value), 2
+
+
+def deserialize_pickle(key, value, flags):
+    if flags == 1:
+        return value
+    if flags == 2:
+        return pickle.loads(value)
+
+
 def invalidate_cache_after_error(f):
     """
     catch any exception and invalidate internal cache with list of nodes
@@ -83,6 +101,8 @@ def _cache(self):
         if getattr(self, '_client', None) is None:
 
             options = self._options
+            options['serializer'] = serialize_pickle
+            options['deserializer'] = deserialize_pickle
             options.setdefault('ignore_exc', True)
             options.pop('cluster_timeout', None)