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

Question

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

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

在这个程序中，如果你输入一个整数，它会在屏幕上打印出来。在您输入0 之前，程序永远不会终止。

那我怎样才能用 Python 立即获取一个标准输入对应的标准输出呢？也就是说，如果我给标准输入一个“1”，我会期望立即从标准输出中获取一个“1”，而这个过程还没有终止。

---

结论：有两种实现方式。

1. 使用subprocess.Popen，将子进程的stdout或stdin当作文件，写入stdin（需要'\n'），用readline从stdout读取

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