Python类装饰器和超出的最大递归深度

Question

我尝试定义类装饰器。 我在装饰类中__init__方法的问题。 如果__init__方法调用super ，则会引发RuntimeError 超出的最大递归深度 。

代码示例：

def decorate(cls):
    class NewClass(cls): pass
    return NewClass

@decorate
class Foo(object):
    def __init__(self, *args, **kwargs):
        super(Foo, self).__init__(*args, **kwargs)

我究竟做错了什么？

谢谢，Michał

编辑1

感谢Mike Boers的回答，我意识到正确的问题是我应该怎样做才能让超级（Foo，self）指向适当的阶级。

我还有两个限制。 我想调用Foo.__init__方法，我不能改变Foo类的定义。

编辑2

我已经解决了这个问题。 我修改了装饰函数体。 我不回新课。 而不是我包装原始类的方法。

Answer 1

你需要覆盖NewClass.__init__以防止递归，因为NewClass.__init__是Foo.__init__并且它一直在调用自己。

def decorate(cls):
    class NewClass(cls):
        def __init__(self):
            pass
    return NewClass

---

新想法：

如何不进行子类化？ 也许猴子补丁是你的朋友？

def decorate(cls):
    old_do_something = cls.do_something
    def new_do_something(self):
        print "decorated",
        old_do_something(self)

    cls.do_something = new_do_something
    return cls

@decorate
class Foo(object):
    def __init__(self, *args, **kwargs):
        super(Foo, self).__init__(*args, **kwargs)

    def do_something(self):
        print "Foo"

f = Foo()
f.do_something()

Answer 2

请记住，装饰器只是语法糖：

>>> Foo = decorate(Foo)

所以在这种情况下，名称Foo实际上是指NewClass类。 在Foo.__init__方法中，你实际上要求NewClass的超__init__ ，即Foo.__init__ （这是当前正在运行的）。

因此，你的Foo.__init__继续接收它自己的__init__来调用，你最终会进行无限递归。