我可以在python中装饰一个`__getattr__`方法吗？

Question

是否可以在python中装饰__getattr__之类的方法？

像这样

@myWrapper
def __getattr__(self,attrName):
    '''
    myCode Goes here
    '''

当我尝试使用类似上面的东西时，我遇到了一些错误：

感谢威廉指出..

我发现了我在代码中犯的错误。

我的意图是做这样的事情

class A:
    def __init__(self):
        self.a=someObject()
    @mywrapper
    def __getattr__(self,argName):
        # call self.a 's  method. If that doesnt have
        # then raise an exception

所以我的方法可能有一些参数，我也需要传递它..

所以我写了这样的东西来测试：

class A:
    def __init__(self):
        self.a=someObject()
    @mywrapper
    def __getattr__(self,argName):
        print('getAttr called for {}'.format(argName))
        def mymethod(*args,**kwargs):
            import ipdb;ipdb.set_trace()
        return mymethod

我的包装代码如下所示：

def mywrapper(method):
    def methodWrapper(obj,name):
        print("In decorator for {}".format(method));
        method(obj,name)
    return methodWrapper

但是当我像 obj.c(1) 一样执行时，我得到以下异常

TypeError: 'NoneType' object is not callable

事实上，装饰器和__getattr__ 中的打印语句都在执行，但我看到了异常..

如何解决？

Answer 1

是的，这是可能的。 __getattr__ 实际上只是另一个函数，尽管 Python 解释器会将其称为它的一种后备机制。 __init__、__getattr__ 等固有没有什么特别之处。

然而，按照惯例，带有前后双下划线的名称是 “魔法”对象或存在于用户控制的命名空间中的属性。例如。 __init__、__import__ 或 __file__。永远不要发明这样的名字；仅按文档说明使用。”。但这是一个风格指南（您最好遵循）。

但是，由于某种原因，您的 @myWrapper 不尊重（数量）属性。

例如，如果您要编写一个包装器：

def myWrapper(func):
    def g(self,attrName):
        return func(self,'foo_%s'%attrName)
    return g

那么就有没有问题了。

如果我们现在定义一个类Foo：

class Foo:

    @myWrapper
    def __getattr__(self,attrName):
        print(attrName)

然后我们看到，装饰器工作正常：

>>> foo = Foo()
>>> foo.bla
foo_bla

如您所见，__getattr__ 打印出<b>foo_</b>bla，这意味着myWrapper 在属性前添加了foo_。

至于您的（已编辑）问题，我猜它之所以说“TypeError: 'NoneType' object is not callable”是因为装饰器应该返回一个（新）函数。

可能你的装饰器看起来像：

def mywrapper(func):
    def g(self,attrName):
        return func(self,attrName)

所以你可能忘了打电话给return g。

编辑：根据您更新的问题，问题是您不返回内部__getattr__ 调用的结果：

def mywrapper(method):
    def methodWrapper(obj,name):
        print("In decorator for {}".format(method));
        return method(obj,name) -- with a return statement
    return methodWrapper

这很重要，因为您 __getattr__ 返回一个函数。请注意，在mywrapper 的上下文中，method 是原始 __getattr__，methodWrapper 是新的__getattr__。所以你应该传递原始__getattr__ 的结果。现在你不用它做任何事情并返回（默认）None。