Python中如何为继承现有类的子类编写对应工厂函数？

直接修改__class__确实属于非正规的hack写法，不推荐在生产环境长期使用，存在不少潜在风险：

• 若后续B新增了自身的实例属性初始化逻辑，或者原包中A的内部结构调整，直接修改__class__会导致实例属性缺失、方法绑定异常等隐性BUG
• 若A或B使用了__slots__定义允许的属性，修改__class__会直接抛出类型错误
• 该写法依赖CPython的实现细节，在PyPy等其他Python解释器上可能出现不可预期的问题
• 代码可读性差，后续维护者很难第一时间理解强行修改实例类型的逻辑

针对你的需求，有多个更规范的实现方案，可以根据场景选择： #

方案1：修改原工厂函数（最优，适用可修改原some_package代码的场景） #

给make_A新增可选的cls参数，默认值为A，从根源支持自定义返回实例类型：

# some_package 中的修改
def make_A(*args, cls=A, **kwargs):
    out = cls(*args, **kwargs)
    # 原有初始化逻辑保留
    return out

你需要B实例时直接调用即可，不需要单独编写make_B：

b_inst = make_A(*args, cls=B)

方案2：组合代理模式（最稳妥，适用无法修改原包、无强类型校验要求的场景） #

不需要B继承A，而是在B内部持有A的实例，通过__getattr__自动代理A的所有方法和属性，仅自定义扩展逻辑：

class B:
    def __init__(self, a_inst):
        self._a = a_inst

    # 自动代理所有A的属性和方法
    def __getattr__(self, name):
        return getattr(self._a, name)

    def extra_method(self):
        # 你的扩展逻辑
        pass

def make_B(*args, **kwargs):
    return B(make_A(*args, **kwargs))

该方案完全不耦合A的内部实现，只要make_A返回的实例接口不变，原包无论怎么升级都不会影响你的扩展逻辑。 #

方案3：属性复制（适用必须要求B是A的子类的场景） #

如果有类型校验要求B必须是A的子类，可以手动复制A实例的所有属性到新的B实例中，避免直接修改类属性：

def make_B(*args, **kwargs):
    a_inst = make_A(*args, **kwargs)
    # 仅创建B实例，不触发__init__逻辑避免属性覆盖
    b_inst = B.__new__(B)
    # 复制A实例的所有属性
    b_inst.__dict__.update(a_inst.__dict__)
    # 如果A使用了__slots__，需要额外遍历slots复制属性
    for slot in getattr(A, '__slots__', ()):
        if hasattr(a_inst, slot):
            setattr(b_inst, slot, getattr(a_inst, slot))
    return b_inst

该方案比直接修改__class__兼容性更高，风险更低。

只有在你完全确认A和B的结构完全一致、不会用到__slots__、且仅做临时调试的场景下，才可以考虑使用修改__class__的写法，且必须添加详细注释说明该逻辑的作用和边界。

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