使用python设置超时

@TraceFunc
def WorkInTime(Func, Timeout = 0, *args):
'''
If function cann't finished in time then timeout
'''

def handler(signum, frame):
raise AssertionError

try:
signal.signal(signal.SIGALRM, handler)
signal.alarm(Timeout)
ret = Func(*args)
signal.alarm(0)
return ret
except AssertionError:
logger.error("%s conn't finished in %d seconds, timeout!" % (Func.__name__, Timeout))
return -1