运行Python脚本怎样设置执行的超时时间 运行Python脚本的超时控制实用技巧

signal

threading

multiprocessing

signal

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}")