如何基于字符串字面量从联合类型创建dataclass实例？

核心方案：工厂模式 + 事件注册表 #

完全可以通过一行代码创建对应事件实例，同时保留mypy的强类型检查能力。核心思路是用事件注册表替代冗余的if判断，结合工厂函数封装实例创建逻辑。

具体实现代码 #

from dataclasses import dataclass
from typing import Union, Literal, TypeVar, Generic, Type, overload

# 定义通用类型变量
T = TypeVar('T')

# 基础事件类，抽象通用结构
@dataclass
class BaseEvent(Generic[T]):
    name: str
    value: T

# 定义RegisterInfo类型
@dataclass
class RegisterInfo:
    address: int
    data: bytes

# 温度事件类，绑定固定事件名和value类型
@dataclass
class TemperatureEvent(BaseEvent[int]):
    name: Literal["temperature"] = "temperature"

# 寄存器事件类，绑定固定事件名和value类型
@dataclass
class RegisterEvent(BaseEvent[RegisterInfo]):
    name: Literal["register"] = "register"

# 事件注册表：映射事件名到对应类
EVENT_REGISTRY: dict[str, Type[Union[TemperatureEvent, RegisterEvent]]] = {
    "temperature": TemperatureEvent,
    "register": RegisterEvent
}

# 重载函数签名，让mypy能精确推断返回类型
@overload
def create_event(name: Literal["temperature"], value: int) -> TemperatureEvent: ...

@overload
def create_event(name: Literal["register"], value: RegisterInfo) -> RegisterEvent: ...

@overload
def create_event(name: str, value: Union[int, RegisterInfo]) -> Union[TemperatureEvent, RegisterEvent]: ...

# 工厂函数：一键创建事件实例
def create_event(name: str, value: Union[int, RegisterInfo]) -> Union[TemperatureEvent, RegisterEvent]:
    event_cls = EVENT_REGISTRY.get(name)
    if not event_cls:
        raise ValueError(f"未知事件类型: {name}")
    return event_cls(value=value)

# 使用示例
temp_event = create_event("temperature", 25)
register_event = create_event("register", RegisterInfo(0x10, b"\x01"))

# mypy会报错：int没有upper()方法，符合强类型检查需求
# temp_event.value.upper()

优化：自动注册事件类 #

如果事件类型较多，可以用装饰器自动维护注册表，避免手动更新：

def register_event(event_name: str):
    def decorator(cls: Type[BaseEvent[T]]) -> Type[BaseEvent[T]]:
        EVENT_REGISTRY[event_name] = cls
        return cls
    return decorator

# 用装饰器注册事件类
@register_event("temperature")
@dataclass
class TemperatureEvent(BaseEvent[int]):
    name: Literal["temperature"] = "temperature"

@register_event("register")
@dataclass
class RegisterEvent(BaseEvent[RegisterInfo]):
    name: Literal["register"] = "register"

架构合理性分析 #

1. 当前方案的问题：原方案中手动编写大量if判断属于冗余代码，违反了DRY原则，且新增事件时需要修改多处逻辑，不符合开闭原则。
2. 优化后的优势：
   • 保留了dataclass联合类型的强类型检查能力，mypy能精准识别类型错误；
   • 消除冗余判断，新增事件仅需添加对应类和注册（或装饰器），无需修改工厂函数；
   • 实例创建逻辑封装统一，调用方只需一行代码完成实例化。

内容的提问来源于stack exchange，提问作者Quint van Dijk