46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
"Memcached cache backend"
|
|
|
|
from django.core.cache.backends.base import BaseCache, InvalidCacheBackendError
|
|
from django.utils.encoding import smart_unicode, smart_str
|
|
|
|
try:
|
|
import cmemcache as memcache
|
|
except ImportError:
|
|
try:
|
|
import memcache
|
|
except:
|
|
raise InvalidCacheBackendError("Memcached cache backend requires either the 'memcache' or 'cmemcache' library")
|
|
|
|
class CacheClass(BaseCache):
|
|
def __init__(self, server, params):
|
|
BaseCache.__init__(self, params)
|
|
self._cache = memcache.Client(server.split(';'))
|
|
|
|
def add(self, key, value, timeout=0):
|
|
return self._cache.add(key.encode('ascii', 'ignore'), value, timeout or self.default_timeout)
|
|
|
|
def get(self, key, default=None):
|
|
val = self._cache.get(smart_str(key))
|
|
if val is None:
|
|
return default
|
|
else:
|
|
if isinstance(val, basestring):
|
|
return smart_unicode(val)
|
|
else:
|
|
return val
|
|
|
|
def set(self, key, value, timeout=0):
|
|
if isinstance(value, unicode):
|
|
value = value.encode('utf-8')
|
|
self._cache.set(smart_str(key), value, timeout or self.default_timeout)
|
|
|
|
def delete(self, key):
|
|
self._cache.delete(smart_str(key))
|
|
|
|
def get_many(self, keys):
|
|
return self._cache.get_multi(map(smart_str,keys))
|
|
|
|
def close(self, **kwargs):
|
|
self._cache.disconnect_all()
|
|
|