python - 基于实例变量有条件地禁用缓存装饰器

Question

我想缓存一个类的某些方法——但read_only前提是在类的实例上设置了一个标志。

所以，在下面的对象中，我希望该get()方法是可缓存的，但前提self.readonly是它是 True。

我不能有条件地使用装饰器，因为它是在定义类时设置的，而不是在实例化时设置的。

from beaker.cache import cache_regions, cache_region
cache_regions.update({
    'long_term':{
        'expire':86400,
        'type':'dbm',
        'data_dir':'/tmp',
        'key_length': 32,
    }
})

class Foo(object):
  def __init__(self, read_only=True):
    self.read_only = read_only

  @cache_region('long_term')
  def get(self, arg):
    return arg + 1

Answer 1

您可以使用装饰器调用正确的（缓存与否）函数来检查所需的属性：

def conditional(decorator):

    def conditional_decorator(fn):
        dec = decorator(fn)
        def wrapper(self, *args, **kw):
             if self.read_only:
                 return dec(self, *args, **kw)
             return fn(self, *args, **kw)
        return wrapper

    return conditional_decorator

像这样使用：

@conditional(cache_region('long_term'))
def get(self, arg):
    return arg + 1