当我们在写执行一个命令或者一个函数的时候,处理的时间太长或者卡死在某个位置时,我们想要中断,就需要进行一定的处理。废话不多说,上代码,自己看。
def set_command_exec_timeout(cmd_string):
'''
设置命令执行超时
:param cmd_string: 执行的命令
:return:
'''
class TimeoutException(Exception):pass
@contextmanager
def time_limit(seconds):
def signal_handler(signum, frame):
raise TimeoutException, "Timed out!"
signal.signal(signal.SIGALRM, signal_handler)
signal.alarm(seconds)
try:
yield
finally:
signal.alarm(0)
def function():
p = subprocess.Popen(cmd_string, shell=True, stdout=subprocess.PIPE, stderr=subprocess.STDOUT)
cmd_out_info = p.stdout.readlines()
p.terminate()
return cmd_out_info
try:
with time_limit(1):
return function()
except TimeoutException as e:
print "Time out!"
return
参考链接:https://www.cnblogs.com/buxizhizhoum/p/7113355.html