Pydantic:检测字段值是否缺失或为空

时间:2021-02-16 17:38:12

标签: python pydantic

我希望允许用户使用 PUT 调用有选择地更新字段。在 pydantic 模型上,我将字段设为可选。在 FastAPI 处理程序中,如果模型属性为 None,则未给出该字段,我不更新它。

这种方法的问题在于客户端无法“清除”某些类型不需要的字段。

特别是,我希望客户端能够通过在 JSON 中发送空值来清除日期字段。如何检测客户端发送 null 或客户端根本不发送字段/值之间的差异?在任何一种情况下,模型属性都只是 None。

2 个答案:

答案 0 :(得分:3)

pydantic documentation 描述了可以与模型的 .dict() 方法一起使用的两个选项。

  • exclude_unset:是否应从返回的字典中排除创建模型时未明确设置的字段;默认为假。在 v1.0 之前, exclude_unset 被称为 skip_defaults;现在不推荐使用skip_defaults

  • exclude_defaults:是否应从返回的字典中排除等于其默认值(无论是否设置)的字段;默认错误

所以我创建了一个模型类:

from typing import Optional
from pydantic import BaseModel


class MyModel(BaseModel):
    foo: Optional[int] = None
    bar: Optional[int] = None


baz = MyModel(foo=None)
assert baz.dict(exclude_unset=True) == {"foo": None}

baz = MyModel(bar=None)
assert baz.dict(exclude_unset=True) == {"bar": None}

答案 1 :(得分:1)

您可以检查 obj.__fields_set__ 以查看该值是否缺失。

from typing import Optional
from pydantic import BaseModel

class Foo(BaseModel):
    first: Optional[int] = None
    second: Optional[int] = None

foo = Foo.parse_raw('{"first": null}')

assert foo.first is None and foo.second is None
assert foo.__fields_set__ == {"first"}