如何在 Marshmallow 中人为地嵌套模式?

ast*_*asr 6 python serialization marshmallow

在 Marshmallow 中,有没有办法将当前对象传递到字段Nested以产生人为嵌套的序列化?例如,考虑我正在序列化的这个对象:

example = Example(
    name="Foo",
    address="301 Elm Street",
    city="Kalamazoo",
    state="MI",
)
Run Code Online (Sandbox Code Playgroud)

我想为此生成 JSON,如下所示:

{
    "name": "Foo",
    "address": {
            "street": "301 Elm Street",
            "city": "Kalamazoo",
            "state": "MI"
    }
}
Run Code Online (Sandbox Code Playgroud)

本质上,这将是嵌套AddressSchema在 中ExampleSchema,如下所示:

class AddressSchema:
    street = fields.String(attribute="address")
    city = fields.String()
    state = fields.String()

class ExampleSchema:
    name = fields.String()
    address = fields.Nested(AddressSchema)
Run Code Online (Sandbox Code Playgroud)

...但这并不完全符合我的要求。我可以使用自定义函数,但如果可能的话我想使用内置方法。

ast*_*asr 4

我设法找到了一个解决方案,使我能够保留内省并仅使用内置字段;不过,这有点奇怪。我修改ExampleSchema为包含一个@pre_dump添加自引用属性的钩子,并将该字段指向该属性:

class ExampleSchema:
    name = fields.String()
    address = fields.Nested(
        AddressSchema, attribute="_marshmallow_self_reference"
    )

    @pre_dump
    def add_self_reference(self, data):
        setattr(data, "_marshmallow_self_reference", data)
Run Code Online (Sandbox Code Playgroud)