今天在项目中使用 BaseModel 遇到了一个问题,model 的某个字段的值一直赋值不上,在经过一番研究之后,我将其简化,发出来和大伙们讨论
from pydantic import BaseModel
class A(BaseModel):
data: dict | list | BaseModel
class B(BaseModel):
data: BaseModel | dict | list
a1 = A(data={"a": 1, "b": 2})
a2 = A(data=a1)
a3 = A(data=["1", "2"])
b1 = B(data={"a": 1, "b": 2})
b2 = B(data=b1)
b3 = B(data=["1", "2"])
print(a1) # "data={'a': 1, 'b': 2}" 正常
print(a2) # "data=A(data={'a': 1, 'b': 2})" 正常
print(a3) # "data=['1', '2']" 正常
print(b1) # "data=BaseModel()" 不正常
print(b2) # "data=B(data=BaseModel())" 正常
print(b3) # "data=['1', '2']" 正常
这里可以看到对于 b1 的实例化,data 的值并没有成功赋值给 b1.data
按理来说,不管我做不做类型注解,这里都不应该影响我正常赋值和实例化
环境:
python3.10.8
pydantic==2.6.3
当我把 pydantic 更新到最新的 2.10.2 时,实例化 b1 会报错,但是 b3 依然可以执行
AttributeError: 'BaseModel' object has no attribute '__private_attributes__'
我有尝试去 pydantic 的 issue 搜索过,但相关 issue 太多了,没找到相似的。
感觉像是 pydantic 的 bug ?还是因为有什么特性?
1
Lychee0 1 天前
```python
class B(BaseModel): data: Union[Self, Dict, List] ``` 先这么写倒是可以,mypy & ruff 扫了下能过 我发现不标 Self 时 b2.data 还会变成 List ,好奇怪 |
2
Lychee0 1 天前
自动转 List 应该是特性?(这个不应该抛错误吗
|
3
mkroen OP @Lychee0 #1
python3.10 还没有 typing.Self ,这里我写 BaseModel 意思是,data 可以是其他继承 BaseModel 的 class ,而不一定仅是 B 这个 class |
5
sunfkny 21 小时 48 分钟前
缺少用于显示的 __private_attributes__ 和 __pydantic_computed_fields__,空继承一下或者补上属性就好了,正常继承,metaclass 会设置这两个属性的
class BaseModel(BaseModel): pass BaseModel.__private_attributes__ = {} BaseModel.__pydantic_computed_fields__ = {} |
6
chaunceywe 9 小时 17 分钟前
这种最好用 union+discriminator,没 type pydantic 也不知道到底按哪个类型解析
|