Классы декораторов в Python

Я хочу построить классы для использования в качестве декораторов со следующими принципами:

  • Должна быть возможность складывать несколько таких декораторов класса сверху 1 функции.
  • Результирующий указатель имени функции должен быть неотличим от одной и той же функции без декоратора, за исключением, возможно, для того, какой тип/класс он есть.
  • Заказ декораторов не должен быть уместным, если фактически не будет санкционировано декораторами. То есть. независимые декораторы могут применяться в любом порядке.

Это для проекта Django, и конкретный случай, над которым я сейчас работаю, нуждается в 2 декораторах и должен выглядеть как обычная функция python:

@AccessCheck
@AutoTemplate
def view(request, item_id) {}

@AutoTemplate изменяет функцию таким образом, что вместо возврата HttpResponse он просто возвращает словарь для использования в контексте. Используется RequestContext, и имя шаблона выводится из имени метода и модуля.

@AccessCheck добавляет дополнительные проверки для пользователя на основе item_id.

Я предполагаю, что это просто, чтобы получить конструктор правильно и скопировать соответствующие атрибуты, но какие атрибуты это?

Следующий декоратор не будет работать, как я описываю:

class NullDecl (object):
    def __init__ (self, func):
        self.func = func
    def __call__ (self, * args):
        return self.func (*args)

Как показано в следующем коде:

@NullDecl
@NullDecl
def decorated():
    pass

def pure():
    pass

# results in set(['func_closure', 'func_dict', '__get__', 'func_name',
# 'func_defaults', '__name__', 'func_code', 'func_doc', 'func_globals'])
print set(dir(pure)) - set(dir(decorated));

Кроме того, попробуйте добавить "print func. name" в конструкторе NullDecl, и он будет работать для первого декоратора, но не второго - поскольку имя будет отсутствовать.

Уточненный eduffy ответил немного, и, похоже, он работает очень хорошо:

class NullDecl (object):
    def __init__ (self, func):
        self.func = func
        for n in set(dir(func)) - set(dir(self)):
            setattr(self, n, getattr(func, n))

    def __call__ (self, * args):
        return self.func (*args)
    def __repr__(self):
        return self.func

Ответы

Ответ 1

Класс декоратора do-nothing будет выглядеть следующим образом:

class NullDecl (object):
   def __init__ (self, func):
      self.func = func
      for name in set(dir(func)) - set(dir(self)):
        setattr(self, name, getattr(func, name))

   def __call__ (self, *args):
      return self.func (*args)

И затем вы можете применить его обычно:

@NullDecl
def myFunc (x,y,z):
   return (x+y)/z

Ответ 2

Модуль декоратор помогает вам создавать декораторы, сохраняющие подпись.

И PythonDecoratorLibrary могут служить полезными примерами для декораторов.

Ответ 3

Чтобы создать декоратор, который обертывает функции в материи, которые делают их неотличимыми от исходной функции, используйте functools.wraps.

Пример:


def mydecorator(func):
    @functools.wraps(func):
    def _mydecorator(*args, **kwargs):
        do_something()
        try:
            return func(*args, **kwargs)
        finally:
            clean_up()
    return _mydecorator

# ... and with parameters
def mydecorator(param1, param2):
    def _mydecorator(func):
        @functools.wraps(func)
        def __mydecorator(*args, **kwargs):
            do_something(param1, param2)
            try:
                return func(*args, **kwargs)
            finally:
                clean_up()
        return __mydecorator
    return _mydecorator

(мои личные предпочтения - создавать декораторы, используя функции, а не классы)

Порядок декораторов выглядит следующим образом:


@d1
@d2
def func():
    pass

# is equivalent to
def func():
    pass

func = d1(d2(func))