比方说,我希望有一个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)