命令执行或者函数使用超时解决办法

 当我们在写执行一个命令或者一个函数的时候,处理的时间太长或者卡死在某个位置时,我们想要中断,就需要进行一定的处理。废话不多说,上代码,自己看。

        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

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章