Pydantic：如何为领域选择类型的选择？

Question

我有一个FastApi应用程序，需要创建一个汽车类，其中属性 wheel 和 speed 可以采用INT 或 str类型。怎么做？此代码不起作用，因为 wheel 和 speed 将只有一个整数类型（不是第二次打印中的str）：

from pydantic import BaseModel


class Car(BaseModel):
    wheel: int | str
    speed: int | str


bmw = Car(wheel=4, speed=250)
mercedes = Car(wheel='4', speed='200')

print(type(bmw.wheel), type(bmw.speed))
print(type(mercedes.wheel), type(mercedes.speed))

结果是：

<class 'int'> <class 'int'>
<class 'int'> <class 'int'>

Answer 1

因此，我个人会亲自使用pydantic.strictint和pydantic.stricstr在此处（实际上，我几乎在任何地方都使用这些，尤其是stricts strictstr，因为实际上任何对象都可以胁迫到字符串）：

import pydantic


class Car(pydantic.BaseModel):
    wheel: pydantic.StrictInt | pydantic.StrictStr
    speed: pydantic.StrictInt | pydantic.StrictStr


bmw = Car(wheel=4, speed=250)
mercedes = Car(wheel='4', speed='200')

print(type(bmw.wheel), type(bmw.speed))
print(type(mercedes.wheel), type(mercedes.speed))

此打印：

<class 'int'> <class 'int'>
<class 'str'> <class 'str'>

Answer 2

类型规范中有顺序。使用int | str，该值将被视为int，否则str。逆转顺序str | int将导致值将其视为str，否则int。反向订单的问题在于，几乎所有内容都可以视为str，因此bmw值将被施加到str。示例：

from pydantic import BaseModel


class Car(BaseModel):
    wheel: str | int
    speed: str | int


bmw = Car(wheel=4, speed=250)
mercedes = Car(wheel='4', speed='200')

print(type(bmw.wheel), type(bmw.speed))
print(type(mercedes.wheel), type(mercedes.speed))

<class 'str'> <class 'str'>
<class 'str'> <class 'str'>

这里的关键是您需要选择哪种类型优先。

Answer 3

您正在寻找的是键入的联合选项。一个示例在下面

from typing import Union
from pydantic import BaseModel

class Car(BaseModel):
    wheel: Union[str,int]
    speed: Union[str,int]

，而不是简单的str或int，您可以在pydantic中为这些类型编写自己的类，并根据需要添加更多属性。