如何不重写Python dict类为其新增扩展方法实现链式安全取值？

下面提供3种不同侵入程度的实现方式，可根据场景选择：

1. 轻量包装类（最推荐，无第三方依赖，无全局污染） #

仅需要对最外层的event做一次包装即可实现链式调用，不会改变原生dict的行为，也不需要修改event的生成逻辑：

class DictWrapper:
    def __init__(self, data):
        # 非dict类型统一转成空dict，避免后续调用报错
        self._raw = data if isinstance(data, dict) else {}

    # 代理原生dict的所有内置方法，不影响原有逻辑
    def __getattr__(self, method_name):
        if hasattr(self._raw, method_name):
            return getattr(self._raw, method_name)
        raise AttributeError(f"'DictWrapper' object has no attribute '{method_name}'")
    
    def safe_get(self, key):
        # 返回的依然是包装类实例，支持连续链式调用
        return DictWrapper(self._raw.get(key))
    
    # 提取最终的原始值
    def get_value(self):
        return self._raw if self._raw else None

使用示例：

# 仅需最外层包装1次，无需修改后续所有取值逻辑
wrapped_event = DictWrapper(event)
result = wrapped_event.safe_get("request_context").safe_get("identity").safe_get("sourceIp").get_value()

2. 全局猴子补丁（直接给原生dict加方法，需要第三方库） #

如果一定要直接调用dict.safe_get的写法，可以用forbiddenfruit库突破Python对内置类型的修改限制，实现全局扩展：

1. 先安装依赖：

pip install forbiddenfruit

2. 扩展dict的方法：

from forbiddenfruit import curse

def safe_get(self, key):
    return self.get(key) if isinstance(self, dict) else None

# 给全局dict类新增safe_get方法
curse(dict, "safe_get", safe_get)

之后所有dict实例都可以直接用你期望的链式写法：

result = event.safe_get("request_context").safe_get("identity").safe_get("sourceIp")

注意：该方案会全局修改内置类型，可能和其他第三方库的扩展产生冲突，生产环境慎用。

3. 无侵入替代写法（不需要扩展dict也不需要包装） #

如果不想做任何类型修改，也可以用reduce简化嵌套调用，语法同样简洁：

from functools import reduce

def safe_chain_get(data, key):
    return data.get(key) if isinstance(data, dict) else None

# 按路径依次传入key即可，不需要嵌套
result = reduce(safe_chain_get, ["request_context", "identity", "sourceIp"], event)

内容的提问来源于stack exchange，提问作者Darren Oakey