Python subprocess 多次使用 stdin.write 和 stdout.read

Question

Python subprocess 多次使用 stdin.write 和 stdout.read

5

感谢你抽出时间来回答问题。我正在尝试使用Python 3.4，并编写了两个简单的Python程序。其中一个叫做test.py，可以接收用户输入并输出一些内容。

while True:
    print("enter something...")
    x = input()
    print(x)
    time.sleep(1)

为了向此程序发送输入，我有另一个使用子进程的程序：

from subprocess import Popen, PIPE

cat = Popen('python test.py', shell=True, stdin=PIPE, stdout=PIPE)
cat.stdin.write("hello, world!\n")
cat.stdin.flush()
print(cat.stdout.readline())

cat.stdin.write("and another line\n")
cat.stdin.flush()
print(cat.stdout.readline())

然而，当我运行上面的程序时，出现了一个错误：

enter something...

hello, world!
Traceback (most recent call last):
  File "/opt/test.py", line 9, in <module>
    x = input()
EOFError: EOF when reading a line
Exception ignored in: <_io.TextIOWrapper name='<stdout>' mode='w' encoding='UTF-8'>
BrokenPipeError: [Errno 32] Broken pipe

如果我将test.py替换为类似于“cat”的标准Linux命令，则一切正常。

有没有办法发送多个stdin写入并读取多个stdout？

- waka-waka-waka

2

我认为您在Popen命令中漏掉了python，正确的写法是Popen(['python', 'test.py'], shell=True, stdin=PIPE, stdout=PIPE)。 - Ramon Moraes

笔误，如果你看到第一个打印输出正常，那么程序实际上是可以运行的。 - waka-waka-waka

如果您使用Python 3，则 cat.stdin.write("hello, world!\n") 将引发错误（需要使用 universal_newlines=True 来启用文本模式），因此您的实际代码可能有所不同或者您没有使用Python 3。 - jfs

与此无关：您应该import模块并调用其函数（必要时使用multiprocessing）而不是将其作为子进程运行。 - jfs

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- jfs · Accepted Answer

一般来说，您应该使用pexpect进行交互式程序（基于对话框的交互）。

您的具体问题可能是由于Python版本不匹配引起的（您认为代码是使用Python 3执行的，但实际上可能是使用Python 2执行的）。第二个问题（EOFError）是预期的：要么在子脚本中捕获它，要么为子进程提供退出信号（在下面的代码示例中，我使用空行）。

以下是一个Python 3代码，在Python 2上会出现错误提示：

#!/usr/bin/env python3
import sys
from subprocess import Popen, PIPE

with Popen([sys.executable, '-u', 'test.py'], stdin=PIPE, stdout=PIPE,
           universal_newlines=True, bufsize=1) as cat:
    for input_string in ["hello, world!", "and another line", ""]:
        print(input_string, file=cat.stdin, flush=True)
        print(cat.stdout.readline(), end='')

注意：

sys.exectable 是当前Python可执行文件（本例中为Python 3）
universal_newlines=True 启用文本模式（否则，cat.stdin 和 cat.stdout 将使用字节而不是字符串进行操作）
-u 使子进程的输出行缓冲（否则，直到子进程刷新其内部stdout缓冲区，您将看不到任何输出）
with语句关闭管道并等待子进程退出。

这是相应的test.py代码：

#!/usr/bin/env python3
import time

while True:
    x = input("enter something...")
    if not x: # exit if the input is empty
        break
    print(x)
    time.sleep(1)

输出

enter something...hello, world!
enter something...and another line
enter something...

注意："输入一些内容..."后面没有换行符。

它能够工作，但是很脆弱，请阅读Q: 为什么不使用管道（popen()）？和改用pexpect代替。

如果输入是有限的且不依赖于输出，则可以一次性传递所有输入：

#!/usr/bin/env python3
import sys
from subprocess import check_output

output = check_output([sys.executable, 'test.py'],
                      input="\n".join(["hello, world!", "and another line"]),
                      universal_newlines=True)
print(output, end='')

这个版本要求子程序正确处理文件结束符（EOF）：

#!/usr/bin/env python3
import time

while True:
    try:
        x = input("enter something...")
    except EOFError:
        break # no more input

    print(x)
    time.sleep(1)

输出结果与上面显示的相同。