python 执行shell超时时间

最新推荐文章于 2024-07-30 07:08:11 发布

自由自在梦在何方

最新推荐文章于 2024-07-30 07:08:11 发布

阅读量4.3k

点赞数 1

分类专栏： python及shell

本文链接：https://blog.csdn.net/lifestxx/article/details/82898751

版权

python及shell 专栏收录该内容

84 篇文章 0 订阅

订阅专栏

import datetime
import subprocess
import time

def execute_command(cmdstring, timeout=None):

    if timeout:
        end_time = datetime.datetime.now() + datetime.timedelta(seconds=timeout)

    sub = subprocess.Popen(cmdstring, stdin=subprocess.PIPE, bufsize=4096, shell=True)

    while sub.poll() is None:
        time.sleep(0.1)
        if timeout:
            if end_time <= datetime.datetime.now():
                sub.kill()
                return "TIME_OUT"

    return str(sub.returncode)

参考网络实现。加上kill

如果需要输出结果，则应修改为：

def execute_command(cmdstring, timeout=None):

    if timeout:
        end_time = datetime.datetime.now() + datetime.timedelta(seconds=timeout)

    sub = subprocess.Popen(cmdstring, stdin=subprocess.PIPE, bufsize=4096, shell=True)

    while True:
        if sub.poll() is not None:
            break
        time.sleep(0.1)
        if timeout:
            if end_time <= datetime.datetime.now():
                sub.kill()
                return "TIME_OUT"

return str(sub.stdout.read())