Question

我正在尝试使用Marshmallow从字典加载嵌套键，并且无法弄清楚如何执行此操作。

例如，我有一个字段＆＃39;价格＆＃39;它映射到传入字典的值字典[＆＃39;价格＆＃39;] [＆＃39;值＆＃39;]。我找不到Marshmallow支持的内置方式，所以我尝试使用pre_load处理器来压缩字典。但是，我无法让它工作。我的示例单元测试失败，出现错误＆＃34;不是有效字符串＆＃34;对于所有嵌套键。

有什么想法吗？

import unittest


class ResultSchema(Schema):
    title = fields.String()
    description = fields.String()
    brand = fields.String(load_from='brand__name')
    price = fields.String(load_from='price__value')
    url = fields.String(load_from='url__value')

    @pre_load
    def flatten_fields(self, data):
        keys = [('brand', 'name'), ('price', 'value'), ('url', 'value')]

        for outer, inner in keys:
            try:
                data[outer + '__' + inner] = data[outer][inner]
            except (KeyError, TypeError):
                pass

        return data

class SchemaTests(unittest.TestCase):

    def setUp(self):
        self.resultSchema = ResultSchema()
        self.expected = {
            'title': 'fake title',
            'description': 'fake description',
            'brand': {'name': 'fake brand name'},
            'price': {'value': '$82.99', 'integral': 8299},
            'url': {'value': 'fake url'},
            'images': [
                {'value': 'small url', 'xsize': 60, 'ysize': 60},
                {'value': 'small-medium url', 'xsize': 100, 'ysize': 100},
                {'value': 'medium-large url', 'xsize': 160, 'ysize': 160},
                {'value': 'large url', 'xsize': 400, 'ysize': 400}
            ]
        }

    def test_schema_load(self):
        loaded, err = self.resultSchema.load(self.expected)
        if err:
            self.fail(err)

Answer 1

当字段名称为is not found on the input时，分配有load_from参数的密钥名称将启动，如果原始密钥被删除，您的方法将起作用：

data[outer + '__' + inner] = data[outer][inner]
del data[outer]

但是，为什么不直接更改嵌套字段的值，那么load_from=...就变得不必要了：

data[outer] = data[outer][inner]

从嵌套密钥加载到marshmallow

1 个答案: