如何实现请求5分钟未响应时自动取消?Python requests函数优化
解决requests请求超时未自动取消的问题
你的问题在生产环境里太常见了——没设置超时的HTTP请求简直是性能杀手!要实现5分钟(300秒)无响应自动取消,只需要给requests.head()加上超时控制,同时优化下异常处理逻辑,让代码更健壮。
修改后的代码
import requests from requests.exceptions import RequestException def get_url_type(data): url_list = data.get('url', []) url_types = {} for idx, url in enumerate(url_list): print(idx) try: # 设置总超时为300秒(5分钟),也可以拆分为(连接超时, 读取超时)比如(10, 300) response = requests.head(url, timeout=300) url_types[idx] = response.headers.get('content-type', 'Not Available') except RequestException: # 捕获所有requests相关异常(超时、连接失败、DNS错误等) url_types[idx] = 'Not Available' return url_types
关键优化点说明
- 超时参数设置:
timeout=300直接指定了请求的总超时时长,只要5分钟内没拿到响应,就会触发超时异常并进入捕获逻辑。如果需要更精细控制,也可以写成元组形式timeout=(10, 300),表示连接阶段最多等10秒,读取响应阶段最多等300秒。 - 循环逻辑优化:用
enumerate替代手动索引遍历,代码更简洁;同时用data.get('url', [])做兜底,避免data里没有url键时直接抛出KeyError。 - 精准异常捕获:用
RequestException捕获所有requests库的请求类异常(超时、连接失败、DNS解析失败等),比原代码的裸except更安全,不会误捕其他无关异常。 - content-type兜底:如果响应头里没有
content-type字段,直接返回Not Available,避免返回None影响后续逻辑。
这样调整后,你的函数就能在请求超时5分钟时自动终止,并返回预设的占位值,生产环境的效率和稳定性都会提升不少。
内容的提问来源于stack exchange,提问作者haimen
相关产品推荐
相关产品推荐

