Python中如何实现多次捕获socket库的timeout超时异常？

现有代码的问题 #

你当前的写法无法捕获多次超时的核心原因是：同一个try代码块下，相同类型的异常只会匹配到第一个对应的except分支。当你在第一个except分支里再次调用recvfrom时，如果再次抛出超时异常，此时已经脱离了上层try的捕获范围，自然不会触发第二个socket.timeout的异常处理逻辑。

可行实现方案 #

用循环控制重试次数，将接收数据的逻辑放在循环内部的try块中即可，示例代码如下：

import socket

# 可自定义配置
MAX_RETRY_TIMES = 3
SOCKET_TIMEOUT = 2
RECV_BUFFER_SIZE = 2096

# 初始化socket（这里根据你自己的实际初始化逻辑调整即可）
clientSocket = socket.socket(socket.AF_INET, socket.SOCK_DGRAM)
clientSocket.settimeout(SOCKET_TIMEOUT)

received_data = None
for retry_idx in range(MAX_RETRY_TIMES):
    try:
        received_data, addr = clientSocket.recvfrom(RECV_BUFFER_SIZE)
        # 成功收到数据直接跳出重试循环
        break
    except socket.timeout:
        print(f"Timeout occurred! Attempt {retry_idx + 1} of {MAX_RETRY_TIMES}...")
        # 最后一次重试依然失败的处理逻辑
        if retry_idx == MAX_RETRY_TIMES - 1:
            print("All retry attempts failed, no response received.")
            # 可根据业务需求选择抛出异常或者返回空值
            raise TimeoutError("Socket exceeded maximum retry times")

# 后续处理收到的数据的逻辑
if received_data:
    print(f"Successfully received data: {received_data}")

实现要点 #

• 重试次数、超时时间、缓冲大小都统一用常量配置，后续修改不需要调整业务逻辑代码
• 每次接收数据都在独立的try块内，只要未超过最大重试次数，每次超时都可以被正常捕获
• 重试耗尽后可以根据业务需要选择抛出异常终止流程，或者返回空值走降级逻辑

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