如何覆写导入模块中类函数的异常处理逻辑？

听起来你遇到的问题挺闹心的——第三方模块硬卡着重试无效的WebSocket地址，完全不给你输出'hello world'的机会。别慌，我们可以通过几种方式接管这个异常处理流程，把“死磕同一无效地址”改成“自动尝试新地址”的逻辑。

核心思路 #

先拆解下问题：第三方模块的类方法里，捕获了WebSocket连接异常后，会进入循环反复重试同一个无效地址。我们的目标就是拦截这个异常处理环节，替换成自己的多地址尝试逻辑。

下面给你两种实用的实现方案：

方案1：继承原类并重写目标方法（推荐，安全易维护） #

这种方法不会影响原类的其他使用场景，是长期维护的最优解。

假设第三方模块叫third_party_ws，出问题的类是WSClient，调用的指定方法是send_message（参数n是WebSocket地址）。我们可以创建一个子类，重写这个方法：

from third_party_ws import WSClient

class CustomWSClient(WSClient):
    def send_message(self, n, message):
        # 这里放你的备选地址列表，可以从配置文件读取或动态生成
        backup_addresses = [
            "wss://ws-backup-1.example.com",
            "wss://ws-backup-2.example.com",
            "wss://ws-backup-3.example.com"
        ]
        
        # 先尝试用户传入的地址，失败则遍历备选地址
        for addr in [n] + backup_addresses:
            try:
                # 调用父类的原始方法，用当前地址替换原参数n
                return super().send_message(addr, message)
            except ConnectionError as e:
                # 注意：要和原模块抛出的异常类型匹配！比如可能是WebSocket专属异常
                print(f"地址 {addr} 连接失败: {str(e)}，尝试下一个...")
        
        # 所有地址都失败时，抛出自定义异常或返回默认值
        raise RuntimeError("所有WebSocket地址均连接失败")

之后用这个子类替代原类实例化调用：

client = CustomWSClient()
# 哪怕传入无效地址，也会自动尝试备选地址
client.send_message("wss://invalid-dead-ws.example.com", "hello world")

方案2：Monkey-Patch 原类方法（快速临时修复） #

如果不想写子类，或者需要快速修改全局的原类行为，可以用猴子补丁直接替换原方法：

from third_party_ws import WSClient

# 先保存原方法，避免丢失原始逻辑
original_send_message = WSClient.send_message

def custom_send_message(self, n, message):
    backup_addresses = [
        "wss://ws-backup-1.example.com",
        "wss://ws-backup-2.example.com"
    ]
    
    for addr in [n] + backup_addresses:
        try:
            return original_send_message(self, addr, message)
        except ConnectionError:
            print(f"切换到新地址重试: {addr}")
    
    raise RuntimeError("无可用的有效WebSocket地址")

# 替换原类的方法
WSClient.send_message = custom_send_message

这样之后，所有地方调用WSClient.send_message都会自动执行你的自定义逻辑，不用修改原有代码的实例化部分。

关键注意事项 #

• 异常类型要匹配：一定要确认原模块抛出的异常具体是什么类型（比如可能是WebSocketException、ConnectionRefusedError等），捕获错了就拦截不到重试逻辑。可以先打印原模块的异常信息，或者查看它的源码/文档。
• 兼容原方法的参数和返回值：重写或替换方法时，要保证参数、返回值和原方法完全一致，避免影响其他依赖这个方法的业务逻辑。
• 私有方法谨慎修改：如果目标方法是私有方法（比如_send_message），优先用继承方案，尽量不要直接修改私有方法，避免原模块版本更新后出现兼容问题。

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