如何让mypy认可Pydantic自定义Timestamp类型接收字符串参数？

针对你遇到的Mypy类型检查错误，这里提供两种实用解决方案：

方法一：启用Pydantic的Mypy插件（最简单） #

Pydantic官方提供了Mypy插件，能自动识别字段的运行时转换逻辑，无需修改代码。

在项目的mypy.ini或pyproject.toml中添加配置：

• 若用mypy.ini：

[mypy]
plugins = pydantic.mypy

• 若用pyproject.toml：

[tool.mypy]
plugins = ["pydantic.mypy"]

配置完成后，Mypy会认可字符串字面量可直接赋值给Timestamp类型字段，同时第29行的Timestamp("xxx")调用也不会报错。

方法二：使用Annotated结合验证器（无需插件） #

如果你不想依赖Mypy插件，可以通过Annotated显式声明字段的类型规则，让Mypy正确识别兼容关系：

修改后的代码示例：

from __future__ import annotations

from typing import Annotated
from pydantic import BaseModel, PlainValidator


class Timestamp(str):
    @classmethod
    def validate(cls, value: str) -> Timestamp:
        # 这里写入你的时间格式验证逻辑
        return cls(value)


class Foo(BaseModel):
    title: str
    bar: Annotated[str, PlainValidator(Timestamp.validate)]


def _main() -> None:
    f1 = Foo(
        title="F1",
        bar="2023-04-05T06:07:08Z",  # Mypy不再报错
    )
    f2 = Foo(
        title="F2",
        bar=Timestamp("2023-04-05T06:07:08Z"),  # Mypy正常通过
    )

Annotated[str, PlainValidator(Timestamp.validate)]告诉Mypy：字段接收str类型输入，同时Pydantic会通过Timestamp.validate将其转换为Timestamp实例，因此字符串字面量和Timestamp实例都能合法赋值。

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