Python设置函数调用超时timout

2020-12-19 06:54:34 编辑 0 阅读量:724

在写python代码时，常常想有没有办法加个timeout参数，避免函数执行时间过久，目前在linux下找到一种比较方便的方式，代码如下：
```
    import time
    import signal

def test(i):
        time.sleep(i%4)
        print "%d within time"%(i)
        return i

if __name__ == '__main__':
        def handler(signum, frame):
            raise AssertionError

i = 0
        for i in range(1,10):
            try:
                signal.signal(signal.SIGALRM, handler)
                signal.alarm(3)
                test(i)
                i = i + 1
                signal.alarm(0)
            except AssertionError:
                print "%d timeout"%(i)
```

说明：
        1、调用test函数超时监控，使用sleep模拟函数执行超时
        2、引入signal模块，设置handler捕获超时信息，返回断言错误
        3、alarm(3)，设置3秒闹钟，函数调用超时3秒则直接返回
        4、捕获异常，打印超时信息

程序执行结果：
```
    1 within time
    2 within time
    3 timeout
    4 within time
    5 within time
    6 within time
    7 timeout
    8 within time
    9 within time
```

修改一下，做成一个带参数的装饰器

```
import time
import signal
from exceptions import RuntimeError

def setTimeout(runtime):
    def handle(signum, frame):
        raise RuntimeError("Out of runtime!")

def decorator(func):
        @functools.wraps(func)
        def wrapper(*args, **kw):
            signal.signal(signal.SIGALRM, handle)
            signal.alarm(runtime)  # 开启闹钟信号
            try:
                result = func(*args, **kw)
            finally:
                signal.alarm(0)  # 关闭闹钟信号
            return result
        return wrapper
    return decorator

if __name__ == "__main__":
    @setTimeout(5)
    def test():
        print "func start"
        time.sleep(3)
        print "func continue"
        time.sleep(6)
        # 不会打印这一段，说明函数被终止执行
        print "func end"
        return 'test OK'

@setTimeout(3)
    def test2():
        print 'test2 start'
        time.sleep(1)
        print "test2 end"

try:
        print test()
    except RuntimeError as e:
        print "test func abort: ", e
    test2()
    time.sleep(10)  
```
运行结果
```
func start
func continue
test func abort:  Out of runtime!
test2 start
test2 end
```
从结果可以看出，test函数没有执行到`print "func end"`这一步，说明是直接中断了函数的执行，也就是说不会出现超时时间结束后，函数突然返回值的情况

ps：这种方式只适用与linux系统，

还有一种做法，据说可以通用，采用杀死多线程的方式，但是我还没测试
[Python tips: 超时装饰器, @timeout decorator](http://www.cnblogs.com/fengmk2/archive/2008/08/30/python_tips_timeout_decorator.html)

Python设置函数调用超时timout

附件: