pydantic：当类字段由父类键入时，不正确

Question

以下代码尝试序列化c的实例，然后再次对c进行后序列化。因为data的类型信息在dict（）中丢失了，因此data实例被解释为a 的实例/代码>而不是b（我的期望）。

我想要c_again和c相互匹配。我该怎么做？

from typing import Tuple, List
from pydantic import BaseModel

class A(BaseModel):
    a: int

class B(A):
    b: float

class C(BaseModel):
    data: A

b = B(a=1, b=0.2)
c = C(data=b)
c_again = C(**c.dict())
print(c, ", ", c_again)
assert c == c_again

输出

data=B(a=1, b=0.2) ,  data=A(a=1)
Traceback (most recent call last):
  File "/home/h-ishida/tmp/tmp.py", line 18, in <module>
    assert c == c_again
AssertionError

Answer 1

在 @bc291的评论上进行构建：

您需要的起点是键入data作为几种可能类型的union，在您的示例中数据：联合[A，B]。但这还不够：如果您只这样做，Pydantic将尝试将您的数据与它可以完成的第一个类型匹配，因此它仍然可以实例化c_again.data as a（ a = 1），忽略b键/属性。您可以首先在联合中使用类型b来解决此问题，例如union [b，a]，它既行之有效，又不优雅，也不会强大，尤其是在您的实际用例涉及您所说的许多子类。

因此，您还需要将config添加到a，如果存在b，它将使Pydantic拒绝实例化，因此它将跳过它并实例化Union的类，该类改为匹配所有密钥/属性（在这种情况下，b）。请注意，旨意继承这种行为的sublasses。

from typing import Union
from pydantic import BaseModel


class A(BaseModel):
    a: int

    class Config:
        extra = 'forbid'


class B(A):
    b: float


class C(BaseModel):
    data: Union[A, B]


b = B(a=1, b=0.2)
c = C(data=b)
c_again = C(**c.dict())
print(c, ", ", c_again)
assert c == c_again

输出：

data=B(a=1, b=0.2) ,  data=B(a=1, b=0.2)
# no assertion error