进程终止后,Popen对象的返回码为None

时间:2016-06-21 10:32:29

标签: python subprocess

我正在使用Popen运行流程。我需要等待进程终止。我检查过程是否已通过returncode终止。当returncodeNone不同时,流程必须已终止。问题是当print_outputFalse时,returncode始终为None,即使进程已完成运行(已终止)。但是print_outputTrue时并非如此。我使用以下代码运行该过程:

def run(command, print_output=True):
    # code mostly from: http://sharats.me/the-ever-useful-and-neat-subprocess-module.html
    from subprocess import Popen, PIPE
    from threading import Thread
    from queue import Queue, Empty
    from time import sleep

    io_q = Queue()

    def stream_watcher(identifier, stream):
        for line in stream:
            io_q.put((identifier, line))

        if not stream.closed:
            stream.close()

    with Popen(command, stdout=PIPE, stderr=PIPE, universal_newlines=True) as proc:
        if print_output:

            Thread(target=stream_watcher, name='stdout-watcher', args=('STDOUT', proc.stdout)).start()
            Thread(target=stream_watcher, name='stderr-watcher', args=('STDERR', proc.stderr)).start()

            def printer():
                while True:
                    try:
                        # Block for 1 second.
                        item = io_q.get(True, 1)
                    except Empty:
                        # No output in either streams for a second. Are we done?
                        if proc.poll() is not None:
                            break
                    else:
                        identifier, line = item
                        print(identifier + ':', line, end='')

            Thread(target=printer, name='printer').start()

        while proc.returncode is None:
            sleep(2)
            proc.poll()

        if not proc.returncode == 0:
            raise RuntimeError(
                'The process call "{}" returned with code {}. The return code is not 0, thus an error '
                'occurred.'.format(list(command), proc.returncode))

        return proc.stdout, proc.stderr

可能导致此问题的任何线索?

编辑:发现了一些非常奇怪的东西。我正在运行以下代码:

run(my_command, True)
print('--------done--------')
run(my_command, False)
print('--------done--------')

'--------done--------'即使run(my_command, False)被执行也从未打印过。

2 个答案:

答案 0 :(得分:7)

TL; DR

realloc

之后添加popen.wait()

解释部分(排序)

Python过快,子进程已结束但无法读取返回码

(我真的不知道为什么这样做。欢迎解释)

为什么我使用它:

Shell命令执行并获取两者返回代码和输出(stdout)

subprocess.Popen()

另外:请阅读popen page

上的def exec_cmd(cmd): pop = subprocess.Popen(shlex.split(cmd), stderr=subprocess.STDOUT, stdout=subprocess.PIPE) pop.wait() return [pop.returncode, pop.communicate()[0]] 警告

答案 1 :(得分:2)

我不确定它为什么不起作用,但我认为它与不关闭流有关。以下代码有效:

def run(command, print_output=True):
    from subprocess import Popen, PIPE, STDOUT
    from io import StringIO

    popen = Popen(command, stdout=PIPE, stderr=STDOUT, universal_newlines=True)
    out = StringIO()
    for line in popen.stdout:
        if print_output:
            print(line, end='')
        else:
            out.write(line)

    popen.stdout.close()
    return_code = popen.wait()

    if not return_code == 0:
        raise RuntimeError(
            'The process call "{}" returned with code {}. The return code is not 0, thus an error '
            'occurred.'.format(list(command), return_code))

    stdout_string = out.getvalue()
    out.close()

    return stdout_string