Python 将 self 传递给装饰器

Question

我试图将self对象传递给我的装饰器以访问其内容，但出现以下错误：

def log_decorator():
    def log_real_decorator(f):
        @wraps(f)
        def wrapper(self,*args, **kw):      
            print "I am the decorator, I know that self is", self, "and I can do whatever I want with it!"
            print "I also got other args:", args, kw           
            f(*args, **kw)

        return wrapper        
    return log_real_decorator    

class Foo(object):
    @log_decorator

    def meth(self):
        print "I am the method, my self is", self


f = Foo()        
f.meth()

错误：-

TypeError: log_decorator() takes no arguments (1 given)

Answer 1

这是你如何做到的：

def log_real_decorator(f):
    @wraps(f)
    def wrapper(self, *args, **kw):      
        print "I am the decorator, I know that self is", self, "and I can do whatever I want with it!"
        print "I also got other args:", args, kw           
        f(self, *args, **kw)
        # ^ pass on self here

    return wrapper

简单地传递self 。

或者

如果你想创建一个可以同时用于类和方法的通用装饰器，你可以简单地这样做：

def log_real_decorator(f):
    @wraps(f)
    def wrapper(*args, **kw):
        # Do something here
        f(*args, **kw)

    return wrapper

此外，当您需要将一些参数传递给装饰器时，将使用您创建的装饰器（例如@log_real_decorator(some_param=1) 。在您的情况下，您不需要它，而是您应该做的是创建一个装饰器有两个嵌套函数，就像我在上面所做的那样，并将其称为@log_real_decorator 。