具有相同名称的模型的金字塔列表

Posted 2021-05-03

比方说，我希望有一个User模型，其中也包含必须是用户列表的“ friends”字段：

class User(BaseModel):
    id: int
    name: str
    friends: List[User]

但是不可能。有没有办法实现这种行为？

答案

是，您需要使用update_forward_refs，请参阅文档中的self-referencing models。

from typing import List

from devtools import debug

from pydantic import BaseModel


class User(BaseModel):
    id: int
    name: str
    friends: List['User']


User.update_forward_refs()

u = User(id=123, name='hello', friends=[dict(id=321, name='goodbye', friends=[])])

debug(u)

输出：

test.py:18 <module>
    u: User(
        id=123,
        name='hello',
        friends=[
            User(
                id=321,
                name='goodbye',
                friends=[],
            ),
        ],
    ) (User)