Python脚本超时设置方法详解

使用signal模块可在Unix/Linux系统中通过SIGALRM信号实现超时控制，设置定时器并在超时后触发异常，任务执行完毕或超时后需关闭定时器；2. 跨平台场景推荐使用多线程或多进程配合threading或multiprocessing模块，通过守护线程或进程实现超时终止，确保脚本在规定时间内停止，防止资源浪费或程序阻塞，最终实现超时即中断的核心目标。

Python脚本的超时控制，说白了就是给你的代码设定一个“死线”。当它跑得太久，超出了你给的时间限制，就得被强制叫停。这事儿在实际开发里特别常见，比如处理外部请求、跑一些可能耗时很久的计算任务，或者简单点说，就是防止你的脚本“失控”。实现这功能，在Unix/Linux环境里，signal模块是个利器，简单直接；要是想跨平台，那多进程或多线程配合threading或multiprocessing模块，就是更稳妥的选择。核心目的都是一个：超时即终止，避免资源耗尽或服务卡死。

解决方案

要给Python脚本设置执行超时，我个人比较常用的有两大类方法，各有各的适用场景和脾气。

1. 利用signal模块（Unix/Linux专属，简单粗暴）

这方法在Unix-like系统上特别好使，因为signal模块能捕获系统信号。我们常用的就是SIGALRM信号，它能设定一个定时器，时间一到就发出信号。

import signal
import time

class TimeoutException(Exception):
    """自定义超时异常"""
    pass

def timeout_handler(signum, frame):
    """信号处理函数，当SIGALRM信号到达时抛出异常"""
    raise TimeoutException("脚本执行超时了！")

def long_running_task():
    """模拟一个可能耗时很长的任务"""
    print("开始执行一个可能很耗时的任务...")
    try:
        # 模拟一个长时间运行的任务
        # 实际应用中这里是你的业务逻辑
        time.sleep(5) # 假设任务需要5秒完成
        print("任务完成。")
    except TimeoutException:
        # 这里捕获到的是由signal handler抛出的异常
        print("任务被超时中断了！")
        # 可以在这里做一些清理工作，比如关闭文件、释放资源
    finally:
        # 确保定时器被清除，避免影响后续代码
        signal.alarm(0) # 取消之前的alarm定时器

if __name__ == "__main__":
    # 设置SIGALRM信号的处理函数
    signal.signal(signal.SIGALRM, timeout_handler)

    # 尝试一个会超时的任务
    print("\n--- 尝试一个会超时的任务 (设置3秒超时) ---")
    signal.alarm(3) # 3秒后发出SIGALRM信号
    try:
        long_running_task()
    except TimeoutException as e:
        # 主程序捕获到由long_running_task内部传递出来的超时异常
        print(f"主程序捕获到超时异常: {e}")