python - 棉花糖压平嵌套对象

Question

我的对象看起来有点像:

source = {
  id: 123,
  objects: [
    {
      random_data: aaaa,
      actual_data: {
        useful: 111,
        ...
      }
    }, 
    {
      random_data: bbbb,
      actual_data: {
        useful: 222,
        ...
      }
    }
  ]
}

虽然我可以轻松地按原样对其进行序列化，但将其转换为以下格式会更有用:

dest = {
  id: 123,
  objects: [
    {
      useful: 111,
      ...
    },
    {
      useful: 222,
      ...
    }
  ]
}

鉴于我已经有了最内层对象的架构(上面源代码中的 actual_data - 它比我的示例显示的要复杂得多)，展平嵌套对象的最佳方法是什么我只从 actual_data 获取对象而没有额外的深度？我本以为会是这样的

objects = Nested(UsefulSchema, many=True, load_from"<something>.useful_data")

但我一直无法在不对数据进行预处理的情况下找到解决方案。

注意我已经大大简化了我的例子，以给出我的问题的最简单形式。在我的实际代码中，源数据是来自 sqlalchemy 的大量类似对象。

Answer 1

对于 marshmallow 版本 >= 3 (Python 3)

您可以通过使用 Pluck 字段允许父级从其子级获取数据来实现这一点。

class ActualDataSchema(ModelSchema):
    useful = fields.String()
    also_useful = fields.String()

class ObjectSchema(ModelSchema):
    useful = fields.Pluck(ActualDataSchema, 'useful')
    also_useful = fields.Pluck(ActualDataSchema, 'also_useful')

class DestSchema(ModelSchema):
    id = fields.Integer()
    objects = fields.Nested(ObjectSchema, many=True)

对于 marshmallow 版本 < 3 (Python 2)

您必须改用 Method 或 Function 字段。

class ObjectSchema(ModelSchema):
    useful = fields.Function(lambda obj: obj.actual_data.useful)
    also_useful = fields.Function(lambda obj: obj.actual_data.also_useful)

class DestSchema(ModelSchema):
    id = fields.Integer()
    objects = fields.Nested(ObjectSchema, many=True)

更多信息

• 关于这个确切问题的另一个例子，它更加充实并且涉及 SQLAlchemy 和 Flask，请参见 my other answer here .
• documentation for fields.Pluck()
• documentation for fields.Function() and fields.Method()