使用 pydantic json_decoders 解析不同的时间格式无法按预期工作

Question

有人可以向我解释以下 pydantic 模型的行为吗？

from datetime import datetime
from pydantic import BaseModel

first_format = {'time': '2018-01-05T16:59:33+00:00',}
second_format = {'time': '2021-03-05T08:21:00.000Z',}

class TimeModel(BaseModel):
    time: datetime

    class Config:
        json_encoders = {
            datetime: lambda v: v.isoformat(),
        }
        json_decoders = {
            datetime: lambda v: datetime.fromisoformat(v),
        }

print(TimeModel.parse_obj(first_format))
print("first_format successfull")
print(TimeModel.parse_obj(second_format))
print("second_format successfull")

输出：

time=datetime.datetime(2018, 1, 5, 16, 59, 33, tzinfo=datetime.timezone.utc)
first_format successfull
time=datetime.datetime(2021, 3, 5, 8, 21, tzinfo=datetime.timezone.utc)
second_format successfull

预期行为：

像现在一样在

first_format

上工作，但在

second_format

上失败，因为

datetime.fromisoformat(second_format)

会因格式错误而抛出错误

我想做的事：

仅当时间格式为

first_format

时才解析时间，否则引发异常。

感谢您的帮助

编辑： 后来我意识到没有

json_decoders

这样的东西，所以请不要对此感到困惑

问题已解决

Answer 1

您需要在此处使用

pre

validator

，以便解析和验证传入的日期时间字符串：

from datetime import datetime
from pydantic import BaseModel, validator

first_format = {'time': '2018-01-05T16:59:33+00:00',}
second_format = {'time': '2021-03-05T08:21:00.000Z',}


class TimeModel(BaseModel):
    time: datetime

    class Config:
        json_encoders = {
            datetime: lambda v: v.isoformat(),
        }

    @validator('time', pre=True)
    def time_validate(cls, v):
        return datetime.fromisoformat(v)


print(TimeModel.parse_obj(first_format).json())
print("first_format successfull")
print(TimeModel.parse_obj(second_format))
print("second_format successfull")

更新：

如果您使用的是 pydantic v2，则需要使用

field_validator

，因为

validator

已被弃用。关键字参数现在是

mode='before'

。