如何使dotdict实例继承构造参数类型以解决mypy属性报错？

解决方案 #

1. 修复Self类型标注报错 #

root_validator的values参数是包含字段键值对的字典，并非模型实例，直接标注为Self不符合实际类型，会触发mypy错误。解决方式：

• 为模型字段定义对应的TypedDict，精准描述values结构，同时获得代码补全
• 替换Self为这个TypedDict类型

示例代码：

from pydantic import BaseModel, root_validator
from typing import TypedDict

# 定义与模型A字段匹配的TypedDict
class AValues(TypedDict):
    x: int
    y: str

class A(BaseModel):
    x: int
    y: str

    @root_validator(pre=True)
    def validate_all(cls, values: AValues) -> AValues:
        # 后续处理逻辑
        return values

2. 让DotDict继承传入参数的类型，消除属性访问报错 #

通过泛型+TypedDict组合，让DotDict实例拥有传入字典的类型信息，mypy就能识别属性：

from typing import Generic, TypeVar, Dict, Any

# 定义绑定字典类型的泛型变量
T = TypeVar('T', bound=Dict[str, Any])

class DotDict(Generic[T], dict):
    def __init__(self, data: T):
        super().__init__(data)
        # 将字典自身绑定到__dict__，实现点访问
        self.__dict__ = self

    # 为mypy提供属性访问的类型提示
    def __getattr__(self, item: str) -> Any:
        return self[item]

在验证器中使用：

@root_validator(pre=True)
def validate_all(cls, values: AValues) -> AValues:
    d = DotDict[AValues](values)
    print(d.x)  # mypy不再报错，且有代码补全
    return values

替代方案：用Pydantic BaseModel临时包装（更简洁） #

如果不想自定义DotDict，可以直接用当前模型临时转换values（注意pre=True阶段可能存在字段缺失，仅用于代码补全和类型检查）：

@root_validator(pre=True)
def validate_all(cls, values: AValues) -> AValues:
    # 临时转成模型实例，获得点访问和类型提示
    temp_instance = cls(**values)
    print(temp_instance.x)
    return values

注意事项 #

• Python 3.10及以下版本中，Self需从typing_extensions导入，但不建议在root_validator中使用Self，因为values不是实例
• Pydantic v2中root_validator已被model_validator(mode='before')替代，类型处理逻辑一致，只需替换装饰器

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