如何在不重复代码的情况下覆盖多个类（Python）

Question

我有一个父类 A 有几个继承自它的类 (B,C,...)。

我有一些函数is_class_A 在A 中实现，我想以同样的方式在B 和C 中覆盖它。

让我们说：

class A:
    def is_class_A():
        print("We are in class A")

class B(A):
    def is_class_A():
        print("Nope")

class C(A):
    def is_class_A():
        print("Nope")

请注意，在 A 类和其他类中实现的这两个函数都很长，而且更复杂。

我想到了2个解决方案，以避免B，C实现中的重复：

在课堂上使用C:

def is_class_A():
    return B.is_class_A()

我不确定这将如何发挥作用，因为C 的对象不是B 的对象。而且我没有在这里传递任何self，所以我看不出这是如何工作的，也许类似的东西可以？

下一个解决方案是使用另一个遗产：

class C(B)

这并不总是有效，因为并非总是可行（如果它们具有不同的属性）并且可能不符合设计目的。

到目前为止，我最好的尝试是使用另一个类：

class BC(A):
    def is_class_A():
        print("Nope")

class B(BC):
    pass

class C(BC):
    pass

你怎么看？还有一些想法？也许是一些不涉及程序设计的更技术性的东西？

谢谢。

Answer 1

一种选择是在全局范围内定义一次替代方法，然后直接进行类属性分配。

class A:
    def is_class_A(self):
        print("We are in class A")


def alt_is_class_A(self):
    print("Nope")


class B(A):
    is_class_A = alt_is_class_A


class C(A):
    is_class_A = alt_is_class_A


class D(A):
    pass  # No override

分配也可以由装饰器处理：

def mod_it(cls):
    cls.is_class_A = alt_is_class_A


@mod_it
class B(A):
    pass


@mod_it
class C(A):
    pass


# No override
class D(A):
    pass

或通过A.__init_subclass__:

class A:
    def is_class_A(self):
        print("We are in class A")

    def __init_subclass__(cls, is_class_A=None):
        super().__init_subclass__()
        if is_class_A is not None:
            cls.is_class_A = is_class_A


def alt_is_class_A(self):
    print("Nope")


class B(A, is_class_A=alt_is_class_A):
    pass


class C(A, is_class_A=alt_is_class_A):
    pass


# No override 
class D(A):
    pass