Python：装饰器特定参数（与包装函数无关）？

Question

我正在寻找一个缓存装饰器，给定 function 将 function 的结果缓存到装饰中指定的位置。 像这样的东西：

@cacheable('/path/to/cache/file')
def my_function(a, b, c):
    return 'something'

装饰器的参数与它包装的 function 的参数完全分开。 我已经看过很多例子，但我不太明白如何做到这一点 - 是否有可能为装饰器提供一个与包装的 function 无关且未传递的参数？

Answer 1

这个想法是你的装饰器是一个 function 返回一个装饰器。

首先编写你的装饰器，就好像你知道你的参数是一个全局变量一样。 让我们这样说：

-

def decorator(f):
  def decorated(*args,**kwargs):
      cache = Cache(cachepath)
      if cache.iscached(*args,**kwargs):
          ...
      else:
          res = f(*args,**kwargs)
          cache.store((*args,**kwargs), res)
          return res
  return decorated

然后编写一个function ，它将缓存路径作为参数并返回您的装饰器。

-

def cache(filepath)
    def decorator(f):
      def decorated(*args,**kwargs):
          cache = Cache(cachepath)
          if cache.iscached(*args,**kwargs):
              ...
          else:
              res = f(*args,**kwargs)
              cache.store((*args,**kwargs), res)
              return res
      return decorated
    return decorator

Answer 2

是的。 如您所知，装饰器是 function。 当写成表格时：

def mydecorator(func):
   def wrapper(*args, **kwargs):
       return func(*args, **kwargs)
   return wrapper

@mydecorator
def foo(a, b, c):
    pass

传递给mydecorator的参数是 function foo本身。

当装饰器接受一个参数时，调用@mydecorator('/path/to')实际上将首先调用带有'/path/to' 的mydecorator function。 然后调用mydecorator(path)的结果来接收 function foo 。 您实际上是在定义一个动态包装器 function。

简而言之，您需要另一层装饰器功能。

这是这个有点愚蠢的例子：

def addint(val):
    def decorator(func):
        def wrapped(*args, **kwargs):
            result = func(*args, **kwargs)
            return result + val
        return wrapped # returns the decorated function "add_together"
     return decorator # returns the definition of the decorator "addint"
                      # specifically built to return an extra 5 to the sum

@addint(5)
def add_together(a, b):
    return a + b

print add_together(1, 2)
# prints 8, not 3

Answer 3

保罗的回答很好，我会移动缓存 object 所以它不需要每次都构建，并设计你的缓存，以便在缓存未命中时引发 KeyError：

def cache(filepath):
    def decorator(f):
        f._cache = Cache(cachepath)
        def decorated(*args,**kwargs):
            try:
                key = (args, kwargs)
                res = f._cache.get(key)
            except KeyError:
                res = f(*args, **kwargs)
                f._cache.put(key, res)
            return res
        return decorated
    return decorator