多文件下Python类无法注册到Class Factory的问题排查与解决

问题背景 #

尝试将不同文件中的类注册到Class Factory类，该工厂类包含一个名为registry的字典，用于将自定义名称映射到待注册类。当工厂类与待注册类在同一.py文件时一切正常，但将待注册类移至独立文件并导入工厂类使用装饰器注册时，registry字典始终为空，类未完成注册。

需要解决两个问题：

• 为何同一文件正常而拆分后失效？
• 如何让多文件方案生效？

问题原因 #

核心是两个常见误区：

1. registry是实例属性而非类属性：如果把registry定义在工厂类的__init__方法里作为实例变量，每个工厂实例都会拥有独立的字典。拆分文件后，待注册类文件中使用的工厂实例，和主程序中创建的工厂实例不是同一个，自然看不到已注册的类；而同一文件中用的是同一个实例，所以能正常工作。
2. 待注册类的模块未被实际导入：装饰器只有在模块被导入时才会执行，如果主程序只导入了工厂类，却没导入存放待注册类的文件，装饰器根本没机会运行，registry当然是空的。

解决办法 #

1. 改造工厂类为类属性共享模式 #

将registry设为类属性，注册方法用@classmethod修饰，确保所有工厂实例共用同一个注册字典：

# factory.py
class ClassFactory:
    # 类属性，所有实例共享该字典
    registry = {}

    @classmethod
    def register(cls, name):
        def decorator(target_cls):
            cls.registry[name] = target_cls
            return target_cls
        return decorator

    @classmethod
    def create(cls, name, *args, **kwargs):
        if name not in cls.registry:
            raise ValueError(f"未找到注册的类: {name}")
        return cls.registry[name](*args, **kwargs)

2. 确保待注册类模块被导入 #

假设待注册类放在my_classes.py中：

# my_classes.py
from factory import ClassFactory

@ClassFactory.register("class_a")
class ClassA:
    def __init__(self):
        print("ClassA 实例创建成功")

@ClassFactory.register("class_b")
class ClassB:
    def __init__(self):
        print("ClassB 实例创建成功")

主程序中必须显式导入该模块，触发装饰器执行：

# main.py
from factory import ClassFactory
# 必须导入此模块，否则装饰器不会运行
import my_classes

# 测试创建实例
obj_a = ClassFactory.create("class_a")
obj_b = ClassFactory.create("class_b")

3. 规避循环导入 #

如果待注册类需要引用工厂类，同时工厂类又要引用待注册类，会出现循环导入问题。解决方式：

• 工厂类单独放在一个独立模块，仅在待注册类文件中导入工厂类；
• 主程序中先导入工厂类，再导入待注册类的模块，避免导入顺序冲突。

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