Python装饰器使函数忘记它属于一个类

Question

我试图写一个装饰器来记录日志：

def logger(myFunc):
    def new(*args, **keyargs):
        print 'Entering %s.%s' % (myFunc.im_class.__name__, myFunc.__name__)
        return myFunc(*args, **keyargs)

    return new

class C(object):
    @logger
    def f():
        pass

C().f()

我想打印：

Entering C.f

但相反，我收到此错误消息：

AttributeError: 'function' object has no attribute 'im_class'

大概这与’logger’中’myFunc’的范围有关，但是我不知道是什么。

Answer 1

Claudiu的答案是正确的，但是您也可以通过从self参数中删除类名来作弊。在继承的情况下，这将产生误导性的日志语句，但会告诉您正在调用其方法的对象的类。例如：

from functools import wraps  # use this to preserve function signatures and docstrings
def logger(func):
    @wraps(func)
    def with_logging(*args, **kwargs):
        print "Entering %s.%s" % (args[0].__class__.__name__, func.__name__)
        return func(*args, **kwargs)
    return with_logging

class C(object):
    @logger
    def f(self):
        pass

C().f()

就像我说的那样，如果您从父类继承了一个函数，这将无法正常工作。在这种情况下，您可能会说

class B(C):
    pass

b = B()
b.f()

并获取消息Entering B.f，Entering C.f因为那是正确的类，所以您实际上希望在其中获取消息。另一方面，这可能是可以接受的，在这种情况下，我会建议这种方法胜过Claudiu的建议。