Python 装饰器作为类成员

Question

我想为类函数创建一个装饰器，它将在函数的开头和结尾处生成日志消息，其中包含特定于运行该函数的类实例的信息。我尝试通过将装饰器创建为类成员来实现此目的，但这不起作用，因为装饰器期望函数作为第一个参数，但作为类方法，它需要是self. 这是我到目前为止所写的不起作用的内容：import functools, logginglogging.basicConfig(format='%(asctime)s - %(levelname)s - %(funcName)s:%(message)s',                             datefmt='%Y-%m-%d %I:%M:%S %p',                            level=logging.DEBUG)class Foo:    def __init__(self, bar):        self.bar = bar            def log(func):        @functools.wraps(func)        def wrapper_log(*args, **kwargs):            logging.info(f"Started (instance {self.bar})")            func(*args, **kwargs)            logging.info(f"Finished (instance {self.bar}")            return func(*args, **kwargs)        return wrapper_log            @log    def test(self, a, b, c):        pass    foo = Foo("bar")foo.test(1, "b", [3, 4])我无法将其移出类并通过装饰类函数来将类实例作为参数传递，@log(self)因为self它不存在于函数之外。我怎样才能做到这一点？

Answer 1

您可以在类外部定义装饰器，它会正常工作，但您需要self在包装器签名中显式引用。

这样做的原因是因为函数内的所有定义仅在调用该函数时才会计算。当函数test被调用时，它已经绑定到实例并且self将存在于其命名空间中。

import functools, logging

logging.basicConfig(format='%(asctime)s - %(levelname)s - %(funcName)s:%(message)s', 

                            datefmt='%Y-%m-%d %I:%M:%S %p',

                            level=logging.DEBUG)

def log(func):

    @functools.wraps(func)

    def wrapper_log(self, *args, **kwargs):

        logging.info(f"Started (instance {self.bar})")

        func(self, *args, **kwargs)

        logging.info(f"Finished (instance {self.bar}")

        return func(self, *args, **kwargs)

    return wrapper_log

class Foo:

    def __init__(self, bar):

        self.bar = bar

        

    @log

    def test(self, a, b, c):

        pass

    

foo = Foo("bar")

foo.test(1, "b", [3, 4])

这将输出

2020-10-05 11:31:20 PM - INFO - wrapper_log:Started (instance bar)

2020-10-05 11:31:20 PM - INFO - wrapper_log:Finished (instance bar