python - python 中的进程 - 获取非终止进程的标准输出

Question

假设有一个程序，它通过不同的输入产生不同的输出，并且如果输入是特定值则终止。例如，可以用C++编写:

int main() {
    int i;
    while(true) {
        cin >> i;
        if (i) cout << i << endl;
        else break;
    }
    return 0;
}

在此程序中，如果您键入一个整数，它会将其打印在屏幕上。在您输入 0 之前，程序永远不会终止。

那么如何使用Python立即获取与stdin对应的stdout呢？也就是说，如果我向标准输入输入“1”，我将期望立即从标准输出获取“1”，而进程尚未终止。

<小时/>

结论:有两种方法可以实现这一点。

1. 使用subprocess.Popen，将子进程的stdout或stdin当作一个文件，写入stdin(需要'\n')，并且使用 readline

从标准输出读取

2. 使用 pexpect 库。

Answer 1

使用subprocess.Popen():

>>> import subprocess
>>> p = subprocess.Popen(['/your/cpp/program'], stdin=subprocess.PIPE, stdout=subprocess.PIPE)
>>> p.stdin.write('1\n')
>>> p.stdout.readline()
'1\n'
>>> p.stdin.write('10\n')
>>> p.stdout.readline()
'10\n'
>>> p.stdin.write('0\n')
>>> p.stdout.readline()
''
>>> p.wait()
0