python调用函数设置超时机制

小程序：扫一扫查出行
【扫一扫了解最新限行尾号】
复制小程序

有时候需要给函数设置超时机制，以防止它卡住我们的程序，这里可以用python的signal模块，signal模块可以实现程序内部的信号处理。

# coding:utf8
import time
import signal# 自定义超时异常
class TimeoutError(Exception):
def __init__(self, msg):
super(TimeoutError, self).__init__()
self.msg = msgdef time_out(interval, callback):
def decorator(func):
def handler(signum, frame):
raise TimeoutError("run func timeout") def wrapper(*args, **kwargs):
try:
signal.signal(signal.SIGALRM, handler)
signal.alarm(interval) # interval秒后向进程发送SIGALRM信号
result = func(*args, **kwargs)
signal.alarm(0) # 函数在规定时间执行完后关闭alarm闹钟
return result
except TimeoutError, e:
callback(e)
return wrapper
return decoratordef timeout_callback(e):
print(e.msg)@time_out(2, timeout_callback)
def task1():
print("task1 start")
time.sleep(3)
print("task1 end")@time_out(2, timeout_callback)
def task2():
print("task2 start")
time.sleep(1)
print("task2 end")if __name__ == "__main__":
task1()
task2()