如何在无法修改Lock实现时从redis-py的LockError中获取锁名？

应用大量使用redis-py官方包中的redis.lock.Lock，既有自研代码直接使用，也有第三方依赖包调用。偶尔会出现LockError（如应用异常重启时）。捕获该异常后，发现其仅包含错误信息，缺少执行清理任务所需的锁名称。在无法修改Lock实现或继承它（锁可能由第三方代码实例化）的情况下，如何为LockError添加信息，或在except块中获取锁名称？

示例代码：

try:
  do_many_things_with_many_locks()
except LockError as le:
  # 如何获取失败的锁名称？

方法1：猴子补丁改造Lock异常抛出逻辑 #

直接修改redis.lock.Lock中抛出LockError的核心方法，让异常携带锁名称。Lock的acquire、release等方法抛出异常时，能直接访问当前锁的name属性，通过补丁替换这些方法，给LockError实例添加锁名称属性或修改错误信息。

from redis.lock import Lock, LockError
import functools

# 保存原方法引用
original_acquire = Lock.acquire
original_release = Lock.release

def patched_acquire(self, *args, **kwargs):
    try:
        return original_acquire(self, *args, **kwargs)
    except LockError as e:
        # 为异常添加锁名称属性
        e.lock_name = self.name
        # 可选：修改错误信息，直观显示锁名称
        e.args = (f"{str(e)} (锁名称: {self.name})",)
        raise

def patched_release(self, *args, **kwargs):
    try:
        return original_release(self, *args, **kwargs)
    except LockError as e:
        e.lock_name = self.name
        e.args = (f"{str(e)} (锁名称: {self.name})",)
        raise

# 替换原方法
Lock.acquire = patched_acquire
Lock.release = patched_release

后续捕获异常时，可直接通过le.lock_name获取锁名称，或从错误信息中读取。

方法2：包装第三方代码调用，关联锁信息 #

如果第三方代码的调用范围可控，可在调用外层做包装：

• 若第三方代码返回或暴露Lock实例，直接记录锁名称，捕获异常时关联使用；
• 若能通过参数传递锁名称给第三方，提前记录名称，异常时匹配使用。

示例代码：

def wrap_third_party_operation():
    # 获取第三方创建的锁实例
    lock = third_party_get_lock_instance()
    try:
        third_party_execute_with_lock(lock)
    except LockError as e:
        print(f"锁 {lock.name} 执行出错: {e}")
        # 执行清理任务
        clean_up_lock_resource(lock.name)
        raise

方法3：通过栈帧追踪提取Lock实例 #

利用Python的traceback模块，在捕获LockError时回溯调用栈，找到抛出异常的Lock方法所在栈帧，提取self（即Lock实例）以获取锁名称。

import traceback
from redis.lock import Lock, LockError

try:
    do_many_things_with_many_locks()
except LockError as le:
    lock_name = None
    # 回溯栈帧，定位redis.lock模块的调用
    tb = le.__traceback__
    while tb:
        frame = tb.tb_frame
        if frame.f_code.co_filename.endswith("redis/lock.py"):
            # 从栈帧局部变量中获取Lock实例
            if 'self' in frame.f_locals and isinstance(frame.f_locals['self'], Lock):
                lock_name = frame.f_locals['self'].name
                break
        tb = tb.tb_next
    
    if lock_name:
        print(f"出错的锁名称: {lock_name}")
        clean_up_lock_resource(lock_name)
    else:
        print(f"无法定位锁名称，错误信息: {le}")

这种方法依赖redis-py的代码结构，若后续官方包路径或方法实现变更，需调整栈帧判断逻辑，但无需修改原类代码。

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