Python中的可选链接

在 JavaScript 中,如果我不确定链的每个元素是否存在/是否未定义,我可以做 foo?.bar ,如果 bar 上不存在 foo ,解释器会默默地短路它而不是抛出错误。

Python中有类似的东西吗?现在,我一直在这样做:

if foo and foo.bar and foo.bar.baz:
    # do something

我的直觉告诉我,这不是检查链中每个元素是否存在的最佳方法。有没有更优雅/Pythonic 的方式来做到这一点?

stack overflow Optional chaining in Python
原文答案

答案:

作者头像

大多数pythonic方式是:

try:
    # do something
    ...
except (NameError, AttributeError) as e:
    # do something else
    ...
作者头像

您可以使用 getattr

getattr(getattr(foo, 'bar', None), 'baz', None)
作者头像

如果是字典,您可以使用 get(keyname, value)

{'foo': {'bar': 'baz'}}.get('foo', {}).get('bar')
作者头像

您可以使用格洛姆。

from glom import glom

target = {'a': {'b': {'c': 'd'}}}
glom(target, 'a.b.c', default=None)  # returns 'd'

https://github.com/mahmoud/glom

作者头像

结合我在这里看到的一些东西。

from functools import reduce

def optional_chain(obj, keys):
    try:
        reduce(getattr, keys.split('.'), root)
    except AttributeError:
        return None

optional_chain(foo, 'bar.baz')

或者改为扩展 getattr ,以便您也可以将其用作 getattr 的替代品

from functools import reduce

def rgetattr(obj, attr, *args):
    def _getattr(obj, attr):
        return getattr(obj, attr, *args)
    return reduce(_getattr, attr.split('.'), obj)

使用 rgetattr 如果路径不存在,它仍然可以引发 AttributeError ,并且您可以指定自己的默认值而不是 None。

作者头像

我喜欢像 Kotlin 这样的现代语言,它允许这样做:

foo?.bar?.baz

最近我尝试在 python 中实现类似的东西很有趣: https://gist.github.com/karbachinsky/cc5164b77b09170edce7e67e57f1636c

不幸的是,问号在 python 中的属性名称中不是有效符号,因此我使用了 Unicode 中的类似标记:)

作者头像

将其他一些答案组合到一个函数中,可以为我们提供易于阅读的内容以及可以与对象和字典一起使用的内容。

def optional_chain(root, *keys):
    result = root
    for k in keys:
        if isinstance(result, dict):
            result = result.get(k, None)
        else:
            result = getattr(result, k, None)
        if result is None:
            break
    return result

使用此函数,您只需在第一个参数之后添加键/属性。

obj = {'a': {'b': {'c': {'d': 1}}}}
print(optional_chain(obj, 'a', 'b'), optional_chain(obj, 'a', 'z'))

给我们:

{'c': {'d': 1}} None