Python中如何关联Entity与Domain类型边界实现mypy检查

要解决Entity和Domain类型不匹配的问题，你需要抛弃Union的松散约束，转而用实体-领域的映射类型建立两者的强关联，让mypy在静态检查阶段就能识别错误配对。

1. 定义实体与领域的映射关系 #

先明确每个Entity对应的Domain类型，用映射结构把它们绑定起来：

from typing import TypeVar, Generic, Type, Dict

# 示例实体类
class UserModel:
    pass

class TermsOfUse:
    pass

# 对应领域类
class UserDomain:
    pass

class TermsOfUseDomain:
    pass

# 核心：建立实体到领域的严格一对一映射
EntityToDomain: Dict[Type, Type] = {
    UserModel: UserDomain,
    TermsOfUse: TermsOfUseDomain,
}

# 定义泛型变量，限制E只能是映射中的实体类型
E = TypeVar("E", bound=tuple(EntityToDomain.keys()))

2. 实现带自动类型推导的抽象仓库 #

利用泛型和映射，让仓库类自动根据Entity推导对应的Domain类型：

from abc import ABC, abstractmethod

class RepoPattern(Generic[E], ABC):
    def __init__(self, entity_type: Type[E]) -> None:
        self.entity_type = entity_type
        # 自动绑定对应Domain类型，无需手动传入
        self.domain_type = EntityToDomain[entity_type]

    @abstractmethod
    def get(self, entity_id: int) -> EntityToDomain[E]:
        # 返回值自动关联对应Domain类型
        pass

3. 验证类型检查效果 #

现在mypy会严格拦截错误的类型配对：

# 正确用法：传入UserModel，自动关联UserDomain（mypy无报错）
user_repo = RepoPattern(UserModel)
# get方法返回值会被推导为UserDomain类型

# 错误用法：强制绑定不匹配的类型（mypy直接抛出类型错误）
invalid_repo = RepoPattern[UserModel](TermsOfUse)

如果需要显式传入Domain做双重验证，可调整仓库类初始化逻辑：

class RepoPattern(Generic[E], ABC):
    def __init__(self, entity_type: Type[E], domain_type: Type[EntityToDomain[E]]) -> None:
        self.entity_type = entity_type
        self.domain_type = domain_type

# 正确调用
user_repo = RepoPattern(UserModel, UserDomain)
# 错误调用：mypy会报错
invalid_repo = RepoPattern(UserModel, TermsOfUseDomain)

核心优势 #

• 替代Union的松散约束，建立Entity与Domain的一对一绑定关系
• mypy在静态检查阶段就能拦截类型不匹配错误，无需运行时验证
• 新增实体时只需扩展EntityToDomain映射，无需修改仓库类核心逻辑

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