如何以故障安全的方式将内容从一个dict映射到另一个dict?

时间:2016-03-09 12:17:37

标签: python dictionary mapping try-except

我从api得到了一个词:

initial_dict = {
    "content": {
        "text":
    },
    "meta": {
        "title": "something",
        "created": "2016-03-04 15:30",
        "author": "Pete",
        "extra": {
            "a": 123,
            "b": 456
        }
    }
}

我需要将其映射到另一个词典:

new_dict = {
    "content_text": initial_dict['content']['text'],
    "meta_title": initial_dict['meta']['title'],
    "meta_extras": {
        "time_related": {
            initial_dict['meta']['created']
        },
        "by": initial_dict['meta']['author']
    }
}

问题是并非所有字段都始终在initial_dict中。我当然可以将new_dict的整个创建包装成try/except,但如果其中一个初始字段不存在则会失败。

除了为try/except添加的每个字段创建new_dict之外别无他法吗?实际上,dict比这更大(大约400个键/值对),所以这将变得非常快。

是不是有更好更多的pythonic方法呢?

1 个答案:

答案 0 :(得分:2)

如何使用dict.get?如果密钥不在字典中,则返回None而不是抛出错误。

new_dict = {
    "content_text": initial_dict['content'].get('text'),
    "meta_title": initial_dict['meta'].get('title'),
    "meta_extras": {
        "time_related": {
            initial_dict['meta'].get('created')
        },
        "by": initial_dict['meta'].get('author')
    }
}

如果这比一个级别更深,则可以按照评论中的建议some_dict.get('key1', {}).get('key2')进行。

将原始字典转换为defaultdict也是一个选项,它允许您继续使用[]表示法(比必须链接get方法更实用):

from collections import defaultdict
def to_defaultdict(d):
    return defaultdict(lambda: None, ((k, to_defaultdict(v) if isinstance(v, dict) else v)
                                      for k, v in d.items()))
initial_dict = to_defaultdict(initial_dict)

然后,您可以过滤掉None值:

def filter_dict(d):
    return dict((k, filter_dict(v) if isinstance(v, dict) else v)
                for k, v in d.items() if v is not None)
new_dict = filter_dict(new_dict)