| from mako import exceptions |
| |
| try: |
| from beaker import cache |
| cache = cache.CacheManager() |
| except ImportError: |
| cache = None |
| |
| class Cache(object): |
| def __init__(self, id, starttime): |
| self.id = id |
| self.starttime = starttime |
| self.def_regions = {} |
| |
| def put(self, key, value, **kwargs): |
| defname = kwargs.pop('defname', None) |
| expiretime = kwargs.pop('expiretime', None) |
| createfunc = kwargs.pop('createfunc', None) |
| |
| self._get_cache(defname, **kwargs).put_value(key, starttime=self.starttime, expiretime=expiretime) |
| |
| def get(self, key, **kwargs): |
| defname = kwargs.pop('defname', None) |
| expiretime = kwargs.pop('expiretime', None) |
| createfunc = kwargs.pop('createfunc', None) |
| |
| return self._get_cache(defname, **kwargs).get_value(key, starttime=self.starttime, expiretime=expiretime, createfunc=createfunc) |
| |
| def invalidate(self, key, **kwargs): |
| defname = kwargs.pop('defname', None) |
| expiretime = kwargs.pop('expiretime', None) |
| createfunc = kwargs.pop('createfunc', None) |
| |
| self._get_cache(defname, **kwargs).remove_value(key, starttime=self.starttime, expiretime=expiretime) |
| |
| def invalidate_body(self): |
| self.invalidate('render_body', defname='render_body') |
| |
| def invalidate_def(self, name): |
| self.invalidate('render_%s' % name, defname='render_%s' % name) |
| |
| def invalidate_closure(self, name): |
| self.invalidate(name, defname=name) |
| |
| def _get_cache(self, defname, type=None, **kw): |
| if not cache: |
| raise exceptions.RuntimeException("the Beaker package is required to use cache functionality.") |
| if type == 'memcached': |
| type = 'ext:memcached' |
| if not type: |
| (type, kw) = self.def_regions.get(defname, ('memory', {})) |
| else: |
| self.def_regions[defname] = (type, kw) |
| return cache.get_cache(self.id, type=type, **kw) |
| |