如何改变Pydantic的行为如何包括一个额外的深层领域

Question

我有两个后续课程。

class Library(BaseModel):
    package: str

class JobTaskSettings(TaskKey):
    description: Optional[str]
    depends_on: Optional[List[TaskKey]]
    on_cluster_id: str = Field(..., alias="existing_cluster_id")
    libraries: List[Library]

当我调用.dict doc时， 在jobtasksettings上，我想

{
  "libraries": [
    {
      "pypi": {
        "package": "requests"
      }
    },
    {
      "pypi": {
        "package": "bs4"
      }
    }
  ]
}

在每个库模型中都用“ PYPI”键包裹在另一个JSON对象的情况下，以下结果。 这可能吗？

Answer 1

您只需要添加另一个类和Modifi库类：

from pydantic import BaseModel
from typing import List


class Package(BaseModel):
    package: str


class Library(BaseModel):
    pypi: Package


class JobTaskSettings(BaseModel):
    libraries: List[Library]


package_1 = Package(package="requests")
package_2 = Package(package="bs4")
libraries = [Library(pypi=package_1), Library(pypi=package_2)]

my_instance = JobTaskSettings(libraries=libraries)

和输出：

>>> my_instance.json()
{
    "libraries": [
        {
            "pypi": {
                "package": "requests"
            }
        },
        {
            "pypi": {
                "package": "bs4"
            }
        }
    ]
}

这比更改.dict（）或.json（）Bechaviour更安全