Python 如果所有字段都为 None，则将 Pydantic 模型返回为 None

发布于08月28日

假设我有一个带有可选字段的PUDINIC模型:

class MyModel(BaseModel):
    field_1: str | None

我通过使用parse_obj()传递一个dict来实例化

现在，如果没有设置任何字段成员，我希望PYDINIC对象为None.

示例:

data_a = {
    'field_1': 'value_1'
}
obj_a = MyModel.parse_obj(data_a)
print(type(obj_a))  # <class 'MyModel'>
  

data_b = {
    'foo': 'bar'
}
obj_b = MyModel.parse_obj(data_b)
print(type(obj_b))  # I would like this to be NoneType !

当然，我知道我可以在进行任何实例化之前判断输入数据中是否存在这些字段，但我希望避免这样做，并以一种更通用的方式来实现(想象一下，有许多不同的模型具有不同的字段).

推荐答案

你可以使用all()和一个列表理解:

if all(val is None for val in dict(obj_b).values()):
    obj_b = None

或者，如果没有任何字段故意设置为None，您可以判断是否有任何字段已设置:

if not obj_b.__fields_set__:
    obj_b = None

这两个文件都可以压缩:

# First
obj_b = None if all(val is None for val in dict(obj_b).values()) else obj_b
# Second
obj_b = obj_b if obj_b.__fields_set__ else None

下面是一个自动执行此操作的基类:

class NoneCheckModel(BaseModel):
    """Model with None checking"""
    @classmethod
    def parse_obj(*args, **kwargs):
        result = super().parse_obj(*args, **kwargs)
        return None if all(val is None for val in dict(result).values()) else result