Python 装饰器作为类成员

Question

我想为类函数创建一个装饰器，它将在函数的开头和结尾处生成日志消息，其中包含特定于运行该函数的类实例的信息。我尝试通过将装饰器创建为类成员来实现此目的，但这不起作用，因为装饰器期望函数作为第一个参数，但作为类方法，它需要是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)

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, logginglogging.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_logclass 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