Python安全的字典导航,正确的方式

时间:2013-08-22 16:03:38

标签: python dictionary idioms

TLDR摘要

我写了一个函数navigateDict,它在dict上进行安全导航,类似于dict.get()但是嵌套。它取代了像

这样的代码
if 1 in data and 'i' in data[1] and 'a' in data[1]['i']:
    print data[1]['i']['a']
else:
    print "Not found"

与大致相同的

found = navigateDict(data, 1, 'i', 'a')
if found is not None:
    print found
else:
    print "Not found"
  • 这个已经是标准库的一部分吗?
  • 是否有更惯用的方法来做同样的事情?
    • 任何需要多次输入任何路径组件键的响应可能都是无法解答的。

其他详细信息

实施如下:

# Allow fallback value other than None
def navigateDictEx(d, keys, fallback=None):
    for key in keys:
        if key in d:
            d = d[key]
        else:
            return fallback
    return d

def navigateDict(d, *keys):
    return navigateDictEx(d, keys)

请参阅摘要以了解示例用法。

Pythonic与否,此功能可以减少冗余是一个坏主意的地方的重复。例如,更改示例中的一个路径组件需要在原始示例中将最多三个不同的值修改为一个,但在修改的示例中仅修改一个。鉴于我经常犯错误,这是一个很大的胜利。

最后我问这个问题:标准库中有没有这样做,或者我需要在我的项目库中找到它的位置?

如果预计命中率将占据主导地位

brionius正确地指出,捕捉KeyError将起作用:

try:
    print data[1]['i']['a']
except KeyError:
    print "Not found"

这可能是我走的路;这很简洁,减少了重复。然而,它确实反映了一个假设,即会有更多的命中而不是未命中。如果有更好的方式来反对我也想知道,也是。

3 个答案:

答案 0 :(得分:6)

执行此操作的一种方法如下:

try:
    print data[1]['i']['a']
except KeyError:
    print "Not found!"

这符合鸭子打字的精神。它可能会也可能不会那么快,因为我认为处理异常会带来一定的开销,但它肯定是“安全的”。

答案 1 :(得分:3)

像这样的解决方案很酷

https://twitter.com/raymondh/status/343823801278140417

>>> from collections import defaultdict
>>> infinite_defaultdict = lambda: defaultdict(infinite_defaultdict)
>>> d = infinite_defaultdict() 
>>> d['x']['y']['z'] = 10
>>> if d['x']['y']['z']: print d['x']['y']['z'] #better reflects that misses are common

答案 2 :(得分:-1)

比赛结束多年,但对于任何绊脚石的人来说,似乎仍然没有一种本地的,流畅的方式来安全地导航Python字典。

输入RestResponse

" RestResponse旨在成为一个流畅的python对象,用于与RESTful JSON API进行交互"

该库包含一个NoneProp对象,允许安全地导航(和构建)JSON数据结构。

>>> import RestResponse
>>> data = RestResponse.parse({})
>>> data.property.is_none
None
>>> bool(data.property.is_none)
False
>>> isinstance(data.property.is_none, RestResponse.NoneProp)
True
>>> data.property.is_none = None
>>> isinstance(data.property.is_none, RestResponse.NoneProp)
False
>>> print data.pretty_print()
{
    "property": {
        "is_none": null
    }
}