Python继承场景下如何强制实例化子类而非父类？

现有代码 #

one.py #

class A:
    def __init__(self):
        pass

class B:
    def __init__(self):
        pass

    def assign_A(self):
        self.A = A()

two.py #

import one

class A(one.A):
    def bar(self):
        pass

class B(one.B):
    def foo(self):
        pass

b = B() 
b.assign_A()
b.A.bar()  # 抛出AttributeError: 'A' object has no attribute 'bar'

问题原因 #

b调用父类one.B的assign_A()方法时，实例化的是one.A而非two.A——因为one模块对two模块无依赖，这种行为符合逻辑，但不符合实际需求。

已想到的解决方案（均不满意） #

• 将A.bar()移至one模块：不符合业务逻辑，bar()概念上属于two模块
• 在two模块中重写one.B.assign_A()方法：存在代码重复
• 编写函数将b.A转换为two.A实例（复制属性）：实现繁琐

提问 #

当前代码结构是否设计不当？针对该问题最优雅的解决方案是什么？

最优解决方案 #

方案：让B类依赖A的类引用，而非硬编码实例化 #

修改one.py中的B类，通过类属性指定默认使用的A类型，子类只需替换该属性即可实现自定义，无需重写方法：

修改后的one.py

class A:
    def __init__(self):
        pass

class B:
    # 定义默认使用的A类
    a_class = A

    def __init__(self):
        pass

    def assign_A(self):
        # 使用类属性a_class实例化，而非直接硬编码A
        self.A = self.a_class()

修改后的two.py

import one

class A(one.A):
    def bar(self):
        pass

class B(one.B):
    # 替换a_class为当前模块的A子类
    a_class = A

    def foo(self):
        pass

b = B() 
b.assign_A()
b.A.bar()  # 正常调用，无错误

方案优势 #

1. 完全避免代码重复，无需重写assign_A方法
2. 符合依赖倒置原则，B不再硬依赖具体的A类，而是依赖抽象的类类型
3. 扩展性强，后续新增其他A的子类时，只需在对应的B子类中修改a_class即可

替代方案：动态传递A类（适合灵活切换场景） #

如果需要同一B类实例使用不同的A子类，可以在初始化时传入A类的引用：

one.py修改

class B:
    def __init__(self, a_class=None):
        # 优先使用传入的A类，无传入则用默认A
        self.a_class = a_class or A

    def assign_A(self):
        self.A = self.a_class()

two.py使用

b = B(a_class=A)
b.assign_A()
b.A.bar()

这种方式更灵活，适合需要动态切换A类型的业务场景。

内容的提问来源于stack exchange，提问作者freitreppe