Python: специфический аргумент декоратора (не связанный с упакованной функцией)? - PullRequest
Новая
       11

Python: специфический аргумент декоратора (не связанный с упакованной функцией)?

6 голосов
/ 19 марта 2009

Я собираюсь создать кеширующий декоратор, в котором данная функция кэширует результат функции в месте, указанном в декорации. Примерно так: 

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

Аргумент декоратора полностью отделен от аргумента функции, которую он переносит. Я рассмотрел довольно много примеров, но я не совсем понимаю, как это сделать - возможно ли иметь аргумент для декоратора, который не связан и не передан в упакованную функцию?

Ответы [ 3 ]

9 голосов
/ 19 марта 2009

Идея состоит в том, что ваш декоратор - это функция, возвращающая декоратор.

FIRST Напишите ваш декоратор так, как если бы вы знали, что ваш аргумент является глобальной переменной. Давайте скажем что-то вроде:

- 

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

THEN Напишите функцию, которая принимает cachepath в качестве аргумента и возвращает ваш декоратор.

- 

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
5 голосов
/ 19 марта 2009

Да, это так. Как известно, декоратор - это функция. Когда написано в форме: 

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

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

аргумент, переданный mydecorator, является самой функцией foo.

Когда декоратор принимает аргумент, вызов @mydecorator('/path/to') на самом деле собирается сначала вызвать функцию mydecorator с помощью '/ path / to'. Затем будет вызван результат вызова mydecorator(path) для получения функции foo. Вы эффективно определяете динамическую функцию-обертку.

В двух словах, вам нужен еще один слой функций декоратора.

Вот этот слегка глупый пример: 

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
3 голосов
/ 19 марта 2009

Хороший ответ Пола, я бы переместил объект кэша, чтобы его не нужно было каждый раз создавать, и спроектировал бы ваш кэш так, чтобы он вызывал 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
...