调用超级初始化的python装饰器

Question

我试图做一个装饰器，它会自动调用父类的init。 它适用于单一继承，但是当我尝试链接效果时，出现堆栈溢出错误。 谁能解释（a）为什么会发生这种情况，以及（b）如何完成我想完成的工作？

def init(__init):
    def wrapper(self, *args, **kwargs):
        self.__class__.__bases__[0].__init__(self)
        __init(self)
    return wrapper

class Base:
    def __init__(self):
        print("base init")


class Parent(Base):
    @init
    def __init__(self):
        print("parent init")

class Child(Parent):
    @init
    def __init__(self):
        print("child init")

a = Parent() #works
#c = Child() #stack overflow

Answer 1

正确的方式

正确的方法是直接在代码中调用父初始化程序。 在Python 3中：

super().__init__()

在Python 2中：

super(Parent, self).__init__()

要么

super(Child, self).__init__()

错误

但是要回答您的紧迫问题，由于您如何获取父代，所以发生了无限递归：

self.__class__.__bases__[0]

无论您调用wrapper函数多少次， self都不会停止成为Child的实例。 您最终会无限期地调用Parent.__init__(self)因为父类永远不会改变。

错误的方法

您可能想要的是找到定义当前正在调用的__init__方法的类，并获取该方法的父级。 @AlexMartelli在他的答案中提供了执行此操作或Python的功能，我在此处逐字复制：

import inspect

def get_class_that_defined_method(meth):
    for cls in inspect.getmro(meth.im_class):
        if meth.__name__ in cls.__dict__: 
            return cls
    return None

如果您使用的是Python 3，请改用@Yoel的版本 ：

def get_class_that_defined_method(meth):
    if inspect.ismethod(meth):
        for cls in inspect.getmro(meth.__self__.__class__):
           if cls.__dict__.get(meth.__name__) is meth:
                return cls
        meth = meth.__func__  # fallback to __qualname__ parsing
    if inspect.isfunction(meth):
        cls = getattr(inspect.getmodule(meth),
                      meth.__qualname__.split('.<locals>', 1)[0].rsplit('.', 1)[0])
        if isinstance(cls, type):
            return cls
    return getattr(meth, '__objclass__', None)  # handle special descriptor objects

现在，您可以按以下方式重新定义init函数：

def init(__init):
    def wrapper(self, *args, **kwargs):
        get_class_that_defined_method(__init).__bases__[0].__init__(self)
        __init(self)
    return wrapper

再次。 请不要这样做。 从__bases__[0]周围的所有内容开始，可能有很多我未涉及的__bases__[0]情况会咬你。 只需使用super 。 无论如何，这是一个更好的想法。