装饰器类和装饰器函数之间的区别

Question

我想这就是它们的称呼，但我会举一些例子以防万一。

装饰类：

class decorator(object):
    def __init__(self, func):
        self.func = func

    def __call__(self, *args, **kwargs):
        print 'something'
        self.func(*args, **kwargs)

装饰器功能：

def decorator(func):
    def wrapper(*args, **kwargs):
        print 'something'
        return func(*args, **kwargs)
    return wrapper

使用一个或另一个只是口味的问题吗？有实际区别吗？

Answer 1

如果您可以编写函数来实现装饰器，则应首选它。但是并非所有装饰器都可以轻松地编写为一个函数-例如，当您要存储一些内部状态时。

class counted(object):
    """ counts how often a function is called """
    def __init__(self, func):
        self.func = func
        self.counter = 0

    def __call__(self, *args, **kwargs):
        self.counter += 1
        return self.func(*args, **kwargs)


@counted
def something():
    pass

something()
print something.counter

我见过人们（包括我自己）经过荒唐的努力，只用函数编写装饰器。我仍然不知道为什么，一个班级的开销通常可以忽略不计。