3 回答
TA贡献1772条经验 获得超5个赞
您需要validate_assignment在模型配置中启用选项:
from typing import Optional
from pydantic import BaseModel, validator
class User(BaseModel):
name: Optional[str] = ''
password: Optional[str] = ''
class Config:
validate_assignment = True
@validator('name')
def set_name(cls, name):
return name or 'foo'
user = User(name=None, password='some_password', )
print("Name is ", user.name)
user.name = None
print("Name is ", user.name)
Name is foo
Name is foo
TA贡献1876条经验 获得超7个赞
这个问题问得很好,所以我想提供一个更广泛的例子,因为有很多方法可以动态分配值。
它仅在字段直接继承数据类时才有效,更具体地说,这样的东西是行不通的。
class User(BaseModel):
name: Optional[str] = ""
password: Optional[str] = ""
class Config:
validate_assignment = True
@validator("name")
def set_name(cls, name):
return name or "bar"
user_dict = {"password": "so_secret"}
user_one = User(**user_dict)
Out: name='' password='so_secret'
始终验证
出于性能原因,默认情况下,当未提供值时,不会为字段调用验证器。但是在这种情况下,当您需要设置动态默认值时,我们可以将其设置为True
class User(BaseModel):
name: Optional[str] = ""
@validator("name", pre=True, always=True)
def set_name(cls, name):
return name or "bar"
In: user_one = User(name=None)
In: user_two = User()
Out: name='bar'
Out: name='bar'
但是always有一个重要的问题,因为我们使用always=True,pydantic会尝试验证默认的None,这会导致错误。
将 Pre 设置为True它将在发生验证错误之前调用该字段,验证器 pre 的默认设置为False,在这种情况下,它们将在字段验证后调用。
使用配置
但这有一些缺点。
class User(BaseModel):
name: Optional[str] = ""
class Config:
validate_assignment = True
@validator("name")
def set_name(cls, name):
return name or "foo"
In: user = User(name=None)
Out: name='foo'
当您将其设置为 None 时,它会正确返回动态值,但在某些情况下,例如完全返回动态值None,它会失败。
In: user = User()
Out: name=''
同样,您需要进行设置才能使其发挥作用。
pre=True
always=True
使用default_factory
当您想要设置默认值(例如 UUID 或日期时间等)时,这非常有用。在这种情况下,您可能想要使用default_factory,但有一个很大的问题,您无法将Callable参数分配给 default_factory 。
class User(BaseModel):
created_at: datetime = Field(default_factory=datetime.now)
In: user = User()
Out: created_at=datetime.datetime(2020, 8, 29, 2, 40, 12, 780986)
TA贡献1779条经验 获得超6个赞
分配默认值的多种方法
方法#1:id具有默认值的必填字段
class User(BaseModel):
id: str = uuid.uuid4()
方法#2id具有默认值的可选字段
class User(BaseModel):
id: Optional[str] = uuid.uuid4()
方法#3:id具有默认值的必填字段
class User(BaseModel):
id: str = Field(default=uuid.uuid4())
方法#4:具有可id调用默认值的必填字段。这对于生成按需值(例如unique UUIDs或 )非常有用Timestamps。
class User(BaseModel):
id: str = Field(default_factory=uuid.uuid4) # uuid.uuid4 is not executed immediately
添加回答
举报