~jonas-drange/ubuntu-start-page/1252899-mobile-friendly

from mako import exceptions

 

try:

    import beaker.container as container

    import beaker.exceptions

    clsmap = {

        'memory':container.MemoryContainer,

        'dbm':container.DBMContainer,

        'file':container.FileContainer,

    }

    try:

        import beaker.ext.memcached as memcached

        # XXX HACK: Python 2.3 under some circumstances will import this module

        #           even though there's no memcached. This ensures its really

        #           there before adding it.

        if hasattr(memcached, 'MemcachedContainer'):

            clsmap['memcached'] = memcached.MemcachedContainer

    except beaker.exceptions.BeakerException:

        pass

except ImportError:

    container = None

    clsmap = {}

 

class Cache(object):

    def __init__(self, id, starttime, **kwargs):

        self.id = id

        self.starttime = starttime

        if container is not None:

            self.context = container.ContainerContext()

        self._containers = {}

        self.kwargs = kwargs

    def put(self, key, value, type='memory', **kwargs):

        self._get_container(key, type, **kwargs).set_value(value)

    def get(self, key, type='memory', **kwargs):

        return self._get_container(key, type, **kwargs).get_value()

    def _get_container(self, key, type, **kwargs):

        try:

            return self._containers[key]

        except KeyError:

            if container is None:

                raise exceptions.RuntimeException("the Beaker package is required to use cache functionality.")

            kw = self.kwargs.copy()

            kw.update(kwargs)

            return self._containers.setdefault(key, clsmap[type](key, self.context, self.id, starttime=self.starttime, **kw))