如何使用 pydantic 创建具有 uuid 的子项

Question

我尝试创建 Foo 的子级，每个子级都应该有自己的 uuid。在真实的代码中，不会创建 Foo 的实例，只会创建它的子实例。子项稍后将保存在数据库中，uuid 用于从数据库中检索正确的对象。

在第一个代码片段中，我尝试使用 init 方法，这会导致 AttributeError。我还尝试使用类方法，这会导致丢失子对象中的所有字段。

如果我的第二个片段每个孩子都会获得相同的 uuid，这对我来说很有意义，因为它作为默认值传递。

我可以将 uuid 创建放入子级中，尽管在使用继承时这感觉不对。

有没有更好的方法为每个孩子创建一个uuid？

# foo_init_.py
class Foo(BaseModel):
    def __init__(self):
          self.id_ = uuid4()
# >>> AttributeError: __fields_set__

# foo_classmethod.py
class Foo(BaseModel):
    @classmethod
    def __init__(cls):
          cls.id_ = uuid4()
# >>> Bar loses id_ fields

from uuid import uuid4, UUID
from pydantic import BaseModel


class Foo(BaseModel):
    id_: UUID = uuid4()


class Bar(Foo):
    pass


class Spam(Foo):
    pass


if __name__ == '__main__':
    b1 = Bar()
    print(b1.id_)  # >>> 73860f46-5606-4912-95d3-4abaa6e1fd2c
    b2 = Bar()
    print(b2.id_)  # >>> 73860f46-5606-4912-95d3-4abaa6e1fd2c
    s1 = Spam()
    print(s1.id_)  # >>> 73860f46-5606-4912-95d3-4abaa6e1fd2c

Answer 1

您可以使用default_factory参数：

class Foo(BaseModel):
    id_: UUID = Field(default_factory=uuid4)