如下所示:
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
|
# coding=utf-8 import signal import time def set_timeout(num, callback): def wrap(func): raise RuntimeError def to_do( * args, * * kwargs): try : signal.signal(signal.SIGALRM, handle) # 設置信號和回調函數 signal.alarm(num) # 設置 num 秒的鬧鐘 print 'start alarm signal.' r = func( * args, * * kwargs) print 'close alarm signal.' signal.alarm( 0 ) # 關閉鬧鐘 return r except RuntimeError as e: callback() return to_do return wrap if __name__ = = '__main__' : def after_timeout(): # 超時后的處理函數 print "do something after timeout." @set_timeout ( 2 , after_timeout) # 限時 2 秒 def connect(): # 要執行的函數 time.sleep( 1 ) # 函數執行時間,寫大于2的值,可測試超時 return 'connect success.' print connect() |
以上這篇python 限制函數執行時間,自己實現timeout的實例就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持服務器之家。
原文鏈接:https://blog.csdn.net/zelinhehe/article/details/77529844