python - 如何使用来自同一来源的数据嵌套棉花糖模式？

Question

我想将输出中的嵌套对象分组contact_type并contact_value放入contact以下模型：

class Account(Base):
    __tablename__ = 'accounts'
    id = sa.Column(sa.Integer, primary_key=True)
    contact_type = sa.Column(sa.String)
    contact_value = sa.Column(sa.String)

这个结果是预期的：

{
  "id": 1,
  "contact": {
    "type": "phone",
    "value": "1234567"
  }
}

实现这一点的最佳方法是什么？

Answer 1

我不知道这是否是最好的方法，但你可以这样做：

from marshmallow import fields

class AccountSchema(ma.Schema):
    id = ma.Int()
    contact = fields.Function(lambda x : {'type': x.contact_type, 'value': x.contact_value})

并使用：

>>> acc = Account(id=1, contact_type="phone", contact_value="1234567")
>>> acc_dict, errors = AccountSchema().dump(acc)
>>> print acc_dict    
{u'contact': {'type': 'phone', 'value': '1234567'}, u'id': 1}

或者您可以使用其他样式的 dict：

class AccountSchema(ma.Schema):
    id = ma.Int()
    contact = fields.Function(lambda x : {x.contact_type : x.contact_value})

这将导致：

{u'contact': {'phone': '1234567'}, u'id': 1}

查看自定义字段

Answer 2

ma = Marshmallow()

class AccountContactSchema(ma.Schema):
    type = ma.Str()
    value = ma.Str()

class AccountSchema(ma.Schema):
    id = ma.Int()
    contact = ma.Nested(AccountContactSchema)

account_schema = AccountSchema()