使用父字段

时间:2018-03-27 06:04:43

标签: flask marshmallow

很抱歉,如果之前有人询问,我实际上找不到解决方案或类似的问题(可能使用了错误的单词)。

我正在更新现有的Flask API,它使用marshmallow和peewee从我们无法控制的客户端(无法更改JSON数据格式)接收数据。

数据格式是这样的:

{
    "site_id": "0102931",
    "update_date": "2018/02/11-09:33:23",
    "updated_by": "chan1",
    "crc": "a82131cf232ff120aaf00001293f",
    "data": [{"num": 1,
              "id": "09213/12312/1",
              "chain": "chain2",
              "operator": "0000122",
              "op_name": "Fred",
              "oid": "12092109300293"
             },
             {"num": 2,
              "id": "09213/12312/2",
              "chain": "chain1",
              "operator": "0000021",
              "op_name": "Melissa",
              "oid": "8883390393"
             }]           
}

我们对主块中的任何内容都不感兴趣,但是site_id,在反序列化以创建模型并存储数据时,必须必须复制到列表中的每个对象中。

这是peeewee中的模型:

class production_item(db.Model):
   site_id = TextField(null=False)
   id_prod = TextField(null=False)
   num = SmallIntegerField(null=False)
   chain = TextField(null=False)
   operator = TextField(null=False)
   operator_name = TextField(null=True)
   order_id = TextField(null=False)

这是marshamallow架构:

class prodItemSchema(Schema):
    num=String(required=True)
    id=String(required=True)
    chain=String(required=True)
    operator=String(required=True)
    op_name=String(required=False, allow_none=True)
    oid=String(required=False, allow_none=True)

我找不到使用load()方法从主结构传递site-id以及为prodItemSchema预加载/后加载装饰器的方法,因此无法创建模型。另外,我希望marshmallow能够为我验证整个结构,而不是在资源和模式之间进行两部分,就像他们现在在代码中所做的那样。

但是在文档中找不到这样的方法,这可能吗?

1 个答案:

答案 0 :(得分:0)

在棉花糖中,可以通过在父方案上使用pre_dump装饰器来设置context,从而在序列化之前将父方案的值传递给子方案。设置上下文后,可以使用function field从父级获取值。

class Parent(Schema):
    id = fields.String(required=True)
    data = fields.Nested('Child', many=True)

    @pre_dump
    def set_context(self, parent, **kwargs):
        self.context['site_id'] = parent['id']
        return data

class Child(Schema):
    site_id = fields.Function(inherit_from_parent)

def inherit_from_parent(child, context):
    child['site_id'] = context['site_id']
    return child