在没有 NoneType 错误的情况下访问嵌套字典的 pythonic 方法是什么

Question

我有一个深度嵌套的字典（从 json 解码，从 instagram api 解码）。 我最初的代码是这样的：

caption = post['caption']['text']

但是如果 'caption' 键或 'text' 键不存在，则会引发 NoneType 或 KeyError 错误。

所以我想出了这个：

caption = post.get('caption', {}).get("text")

哪个有效，但我不确定它的风格。例如，如果我将此技术应用于我试图检索的更深层次的嵌套属性之一，它看起来很丑：

image_url = post.get('images',{}).get('standard_resolution',{}).get('url')

有没有更好、更 Pythonic 的方式来写这个？我的目标是检索数据（如果存在），但不阻止执行（如果不存在）。

谢谢！

Answer 1

Python 3.4 和更新版本包含一个 contextlib 上下文管理器suppress，它正是用于这种事情。当您提前知道它们可能会发生并且您的代码可以处理它时，抑制特定错误。

from contextlib import suppress

sample = {'foo': 'bar'}

with suppress(KeyError):
    print(sample['baz'])

将阻止KeyError 被提升。

因此，为了获取深度嵌套的字典值，您可以像这样使用suppress。

value = None
with suppress(KeyError):
    value = data['deeply']['nested']['dictionary']['key']

Answer 2

我会创建一个自定义 dict 子类，然后解决这个问题：

class SafeDict(dict):
    def __getitem__(self,k):
        if k in self:
            return dict.__getitem__(self,k)
        return None


a = SafeDict({'a':'a'})
print a['a']
>> a
print a['b']
>> None

您可以执行自定义 init 来处理嵌套字典作为 SafeDict 的另一个实例（这将允许您传递它们），或者您可以使用测试（或 try/except 块）来处理防止 KeyErrors

另外，你可以把它变成一个对象类，重载 __getattr__ ，然后用点符号处理事情。我更喜欢这种方法（我第一次看到这个是在 Pylons 框架中）

class AttributeSafeObject(object):

    def __init__(self,**kwargs):
        for key in kwargs:
            setattr(self,key,kwargs[key])

    def __getattr__(self, name):
        try:
            return object.__getattribute__(self,name)
        except AttributeError:
            return None

post = AttributeSafeObject({'a':'a'})
print post.a
>> a
print post.title
>> None

Answer 3

最 Pythonic 的方式就是捕捉 KeyError:

try:
    caption = post['caption']['text']
except KeyError:
    caption = None

这对 Python 程序员来说简单、明显且立即可以理解。

Answer 4

你对这样的事情感觉如何

if 'caption' in post:
    caption = post['caption']['text']

但它也开始崩溃

if 'images' in post and 'standard_resolution' in post['images']:
    image_url = post['images']['standard_resolution']['url']

所以我认为最 Pythonic 的方式就是 ask for forgiveness and not permission

try:
    image_url = post['images']['standard_resolution']['url']
except KeyError:
    image_url = None