# coding=utf-8

import signal

import time

def set_timeout(num, callback):

def wrap(func):

def handle(signum, frame): # 收到信號(hào) SIGALRM 后的回調(diào)函數(shù)，第一個(gè)參數(shù)是信號(hào)的數(shù)字，第二個(gè)參數(shù)是the interrupted stack frame.

raise RuntimeError

def to_do(*args, **kwargs):

try:

signal.signal(signal.SIGALRM, handle) # 設(shè)置信號(hào)和回調(diào)函數(shù)

signal.alarm(num) # 設(shè)置 num 秒的鬧鐘

print 'start alarm signal.'

r = func(*args, **kwargs)

print 'close alarm signal.'

signal.alarm(0) # 關(guān)閉鬧鐘

return r

except RuntimeError as e:

callback()

return to_do

return wrap

if __name__ == '__main__':

def after_timeout(): # 超時(shí)后的處理函數(shù)

print "do something after timeout."

@set_timeout(2, after_timeout) # 限時(shí) 2 秒

def connect(): # 要執(zhí)行的函數(shù)

time.sleep(1) # 函數(shù)執(zhí)行時(shí)間，寫大于2的值，可測(cè)試超時(shí)

return 'connect success.'

print connect()

本文參與騰訊云自媒體分享計(jì)劃，歡迎正在閱讀的你也加入，一起分享。

總結(jié)

以上是生活随笔為你收集整理的python如何控制运行时间_python 限制函数执行时间，自己实现的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。