Python动态获取类实例及返回类型的实现咨询

1. 实现合理性与优化方案 #

当前的工厂模式实现基本可行，但存在几个可以优化的点：

• 调用冗余：现在create返回的是类对象，需要用户手动加()实例化，不如在工厂方法内部直接完成实例化，简化调用逻辑。
• 缺乏异常处理：如果传入class_mapper中不存在的type值，会直接抛出KeyError，建议增加参数校验或异常捕获，提升代码健壮性。
• 扩展性不足：新增Person子类时，必须手动修改PersonFactory的class_mapper，可以用装饰器实现子类的自动注册，降低耦合。

优化后的示例代码 #

基础优化版（直接返回实例+异常处理）

from typing import Union, TypeVar

P = TypeVar('P', bound='Person')

class Person:
    pass

class Engineer(Person):
    pass

class Doctor(Person):
    pass

class PersonFactory:
    class_mapper = {'engineer': Engineer, 'doctor': Doctor}

    @classmethod
    def create(cls, type: str) -> Union[Engineer, Doctor, None]:
        try:
            return cls.class_mapper[type]()
        except KeyError:
            print(f"不支持的人物类型：{type}")
            return None

# 调用更简洁
doctor_instance = PersonFactory.create('doctor')
engineer_instance = PersonFactory.create('engineer')

装饰器注册版（高扩展性）

from typing import Union, TypeVar

P = TypeVar('P', bound='Person')

class Person:
    pass

class PersonFactory:
    class_mapper = {}

    @classmethod
    def register(cls, type_name: str):
        def decorator(subclass: type[P]) -> type[P]:
            cls.class_mapper[type_name] = subclass
            return subclass
        return decorator

    @classmethod
    def create(cls, type: str) -> Union[Person, None]:
        try:
            return cls.class_mapper[type]()
        except KeyError:
            print(f"不支持的人物类型：{type}")
            return None

# 新增子类时直接用装饰器注册，无需修改工厂类
@PersonFactory.register('engineer')
class Engineer(Person):
    pass

@PersonFactory.register('doctor')
class Doctor(Person):
    pass

doctor_instance = PersonFactory.create('doctor')
engineer_instance = PersonFactory.create('engineer')

2. 动态返回类型的实现 #

可以通过**Literal限定输入类型结合类型联合**来实现动态返回类型，让类型检查工具（如mypy）能识别返回的具体子类：

类型标注优化版 #

from typing import Union, Literal, TypeVar

P = TypeVar('P', bound='Person')
PersonType = Literal['engineer', 'doctor']

class Person:
    pass

class Engineer(Person):
    pass

class Doctor(Person):
    pass

class PersonFactory:
    class_mapper = {'engineer': Engineer, 'doctor': Doctor}

    @classmethod
    def create(cls, type: PersonType) -> Union[Engineer, Doctor]:
        return cls.class_mapper[type]()

# 此时类型检查工具能识别doctor_instance是Doctor类型，engineer_instance是Engineer类型
doctor_instance = PersonFactory.create('doctor')
engineer_instance = PersonFactory.create('engineer')

如果需要更精确的动态类型映射，可以结合typing.TypeGuard做类型断言，但上述方案已经能满足大部分场景的类型提示需求。

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