1. 程式人生 > >Python設置函數調用超時

Python設置函數調用超時

異常 sleep error 函數 main 卡住 調用 frame html

http://blog.sina.com.cn/s/blog_63041bb80102uy5o.html

背景:

最近寫的Python代碼不知為何,總是執行到一半卡住不動,為了使程序能夠繼續運行,設置了函數調用超時機制。
代碼:
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

Python設置函數調用超時