python - 如何使用 Pydantic 将此数据结构转换为 JSON？

Question

我有一个数据结构，由带有字符串键的字典组成，每个键的值都是 Pydantic 模型。

例如，字典可能如下所示:

{
 "hello": MyPydanticModel(name="hello"),
 "there": MyPydanticModel(name="there")
}

我知道我可以调用 .json() 将 Pydantic 模型转换为 JSON，但是将字典转换为 JSON 的最直接方法是什么。我需要使用 py2json 或其他库吗？

提前非常感谢。

Answer 1

您还可以使用映射 root type 为数据结构创建模型并使用它的 json 方法。就像这样:

from typing import Dict
from pydantic import BaseModel


class MyPydanticModel(BaseModel):
    name: str


class MyPydanticModelOut(BaseModel):
    __root__: Dict[str, MyPydanticModel]


obj = {
    'hello': MyPydanticModel(name='a'),
    'there': MyPydanticModel(name='b')
}

print(MyPydanticModelOut.parse_obj(obj).json())  # {"hello": {"name": "a"}, "there": {"name": "b"}}