Python库中如何让代码自动选用用户扩展的子类而非基类

核心实现思路（优化你最初的注册字典思路） #

你原本的全局注册思路是这类场景的标准落地方案，仅需要修正原参考代码的几个错误：注册中心存储的是类本身（构造函数）而非预创建的实例、Python实例指向使用self而非this、封装安全的读写接口避免用户直接操作内部字典。

库端实现代码 #

# 库内部私有注册中心，禁止用户直接修改
_lib_class_registry = {}

def register_class(key: str, constructor) -> None:
    """对外暴露的注册接口，用户可通过该方法替换库的默认类实现"""
    if key not in _lib_class_registry:
        raise ValueError(f"无效的注册键：{key}，可注册的键包括：{list(_lib_class_registry.keys())}")
    _lib_class_registry[key] = constructor

def create_instance(key: str, *args, **kwargs):
    """库内部统一调用的实例创建接口，所有需要实例化可扩展类的位置都走该方法"""
    target_constructor = _lib_class_registry[key]
    return target_constructor(*args, **kwargs)


# 注册默认基类
class BaseClassA:
    pass
_lib_class_registry["ClassA"] = BaseClassA

class BaseClassD:
    def __init__(self):
        # 所有实例化ClassA的位置统一调用create_instance
        self.class_a_obj = create_instance("ClassA")
_lib_class_registry["ClassD"] = BaseClassD

class BaseClassC:
    def __init__(self):
        self.class_d_obj = create_instance("ClassD")
_lib_class_registry["ClassC"] = BaseClassC

class BaseClassB:
    def func0(self):
        self.class_c_obj = create_instance("ClassC")
_lib_class_registry["ClassB"] = BaseClassB

用户端使用代码 #

用户仅需要在调用库接口前完成一次自定义类注册即可，不需要修改任何库内部逻辑、不需要重写继承类的方法：

from your_lib import register_class, BaseClassA

# 用户自定义扩展类
class ExtendClassA(BaseClassA):
    # 自定义扩展逻辑
    def custom_func(self):
        print("扩展的ClassA逻辑")

# 注册替换默认的ClassA实现
register_class("ClassA", ExtendClassA)

# 后续正常调用库接口，自动使用用户扩展的类
from your_lib import BaseClassB
b = BaseClassB()
b.func0()
# 验证：深层嵌套的ClassA实例已经是用户扩展的版本
print(type(b.class_c_obj.class_d_obj.class_a_obj)) # 输出 <class '__main__.ExtendClassA'>

可选扩展能力 #

如果需要支持更灵活的实例化逻辑，可对注册中心做如下增强：

• 支持注册工厂函数：注册的constructor可以是任意可调用对象，不一定是类，用户可以自定义实例化逻辑、传入固定初始化参数
• 支持多实例隔离：如果需要避免全局注册的冲突，可以把注册中心封装为独立的Registry类，用户创建自定义的Registry实例后传入库的入口类即可

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