类和函数的装饰器

Question

我的 Python 装饰器有问题。

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

    def __call__(self, *args, **kwargs):
        print('sth to log: %s : %s' % (self.function.__name__, args))
        return self.function(*args, **kwargs)


@decorator
def sum_test(a, b):
    print('sum: %s' % (a+b))

@decorator
class Class_test(object):
    def __init__(self):
        pass
    def sum_func(self, a, b):
        print('class sum: %s' % (a+b))
        return a+b

if __name__ == '__main__':
    sum_test(3, 4)
    func = Class_test()
    var1 = func.sum_func(1, 4)
    print(var1)

输出：

sth to log: sum_test : (3, 4)
sum: 7
sth to log: Class_test : ()
class sum: 5
5

装饰器正在为函数sum_test 工作。我可以记录该函数 sum_test 与变量 3 和 4 一起使用。

我在装饰课程时遇到了问题。我可以记录 Class_test 类的对象已创建，但我没有使用函数 sum_func 的信息。

为什么在类对象上运行sum_func时没有触发装饰器中的__call__()，而直接在函数sum_test上使用时触发了？

Answer 1

使用decorator 装饰类意味着Class_test 现在是decorator 对象，因此Class_test.function 是您为其编写主体的类。

请记住，@decorator 语法是

的语法糖

class Class_test(object):
    def __init__(self):
        pass
    def sum_func(self, a, b):
        print('class sum: %s' % (a+b))
        return a+b

Class_test = decorator(Class_test)

（除非Class_test 符号绝不代表您的班级）。

因此，当您调用 Class_test 时会调用此 decorator 对象，这正是您在日志中看到的情况：

sth to log: Class_test : ()

另一方面，func.sum_func 是一种常规绑定方法。它对decorator一无所知。

如果你打算在类中使用它，我更喜欢闭包风格的装饰器（一个返回函数的函数）。它返回的函数通常由class 机器注册一个方法。

from functools import wraps

def decorator(func):
    @wraps(func)
    def wrapper(*args, **kwargs):
        print('sth to log: %s : %s' % (func.__name__, args))
        return func(*args, **kwargs)
    return wrapper

class Class_test(object):
    @decorator
    def sum_func(self, a, b):
        print('class sum: %s' % (a+b))
        return a+b

如果您仍想使用基于类的方法，请阅读Python Class Based Decorator with parameters that can decorate a method or a function