pydantic root_validator：mypy严格类型检查

Question

我正在寻找一种“正确”的方法，以在Pydantic root_validator 装饰方法中进行严格的类型检查。

以下面的示例：在 validate_model 方法中，我希望能够使用MyPy严格的类型检查。由于 first_name 和 age 均已通过此方法进行验证和类型检查，因此我们可以假设 values ['first_name'] 和值['age'] 分别为'str'和'int'。

您可以看到＃类型：ignore “ hack”，我用来允许键入提示和停止mypy抱怨，但这很丑陋，如果我更改任何变量类型，可以创建错误。理想情况下， root_validator 装饰的方法将具有 value 参数与类 SimpleModel 本身相同，否则应该有一种简单的方法将 SimpleModel 转换为 typedDict ，因此我们可以在 validate_model 签名中使用它，例如： def validate_model（cls，values：simpleModelTypedDict） - ＆gt; 无论如何， SimpleModelTypedDict

，如果有人遇到了类似问题，请让我知道您如何解决它。

from pydantic import BaseModel, root_validator

class SimpleModel(BaseModel):
    first_name: str
    age: int

@root_validator(skip_on_failure=True)
def validate_model(cls, values: dict) -> dict:
    """checks there are no Bobs under 60 (sorry Bob!)"""
    first_name: str = values.get("first_name")  # type:ignore
    age: int = values.get("age")  # type:ignore
    if first_name.lower() == "bob" and age < 60:
        raise ValueError("No Bobs under 60")

    return values

Answer 1

first_name 和 age 的正确类型提示应分别为可选[str] 和可选[int] 。 .get（） dict 的方法可以返回 none ，而值 dict of valivator将在那里是与提供的名称相对应的字段，或者是无效的：先前对该字段的验证失败。您使用 skip_on_failure = true ，我相信它应该涵盖先前失败的验证的情况，但是 .get（）的返回值仍然可以是 none 如果由于某种原因，您尝试获取不存在的字段。

使用该类型提示，如果 first_name 和 age 变量是 none ，则应验证您是否在尝试执行任何其他验证之前。

完整示例：

from typing import Optional

from pydantic import BaseModel, root_validator


class SimpleModel(BaseModel):
    first_name: str
    age: int


@root_validator(skip_on_failure=True)
def validate_model(cls, values: dict) -> dict:
    """checks there are no Bobs under 60 (sorry Bob!)"""
    first_name: Optional[str] = values.get("first_name")
    age: Optional[int] = values.get("age")
    if first_name is not None and age is not None and first_name.lower(
    ) == "bob" and age < 60:
        raise ValueError("No Bobs under 60")

    return values