如何处理KeyError异常并定位引发异常的字典变量名？

直接通过KeyError异常对象本身无法获取触发异常的字典对应的变量名。

原因 #

Python里的变量只是指向对象的引用，字典对象本身不知道自己被哪个（哪些）变量名绑定，而KeyError异常仅会记录缺失的键值（存在args属性中），不会保存触发异常的字典对象的绑定变量信息。

替代解决方案 #

如果确实需要获取这个信息，可以通过以下几种方式实现：

1. 自定义异常，主动传递变量标识

自己定义异常类，在触发异常时手动传入字典的变量名或标识：

class DictKeyError(KeyError):
    def __init__(self, key, dict_name):
        super().__init__(key)
        self.dict_name = dict_name

data = {"my_key": "my_value"}

try:
    key = "unknown_key"
    if key not in data:
        raise DictKeyError(key, "data")
    print(data[key])
except DictKeyError as e:
    print("key name: ", e.args[0])
    print("dictionary name: ", e.dict_name)

2. 封装字典操作，记录变量名

写一个简单的包装函数来处理字典的键访问，同时传入变量名：

def get_dict_value(dict_obj, key, dict_name):
    try:
        return dict_obj[key]
    except KeyError as e:
        raise KeyError(f"Key {key} not found in dict '{dict_name}'") from e

data = {"my_key": "my_value"}

try:
    get_dict_value(data, "unknown_key", "data")
except KeyError as e:
    print(e)

3. 调试场景下用inspect模块（不推荐生产环境）

通过inspect模块查看调用栈的源代码，解析出对应的变量名，但这种方式依赖代码的语法结构，稳定性差，只适合调试：

import inspect

data = {"my_key": "my_value"}

try:
    data["unknown_key"]
except KeyError as e:
    # 获取当前栈帧的上一帧（即触发异常的代码帧）
    frame = inspect.currentframe().f_back
    # 解析源代码行
    line = inspect.getframeinfo(frame).code_context[0].strip()
    # 提取字典变量名（假设是简单的变量访问形式）
    dict_name = line.split("[")[0]
    print("key name: ", e.args[0])
    print("dictionary name: ", dict_name)

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