python - 如何让 Python 程序处理 here 文档？

Question

我已经编写了一个 Python 包装器 (pyprog) 来运行一个程序 (someprogram)，如下所示:

...do some setup stuff in Python...
print("run [y=yes]")
CHOICE=input()
...do some setup stuff in Python...
if CHOICE == "y":
  status=subprocess.call(["someprogram"])
  sys.exit(status)

用户想要使用 shell 脚本来运行程序并使用如下文档为其提供输入:

#!/bin/sh
pyprog > pyprog.log << EOF
y
file1
file2
EOF

有没有一种方法可以生成子进程，以便 here 文档可以工作(“y”被 Python input() 使用，而“file1”和“file2”继续作为某个程序的标准输入)？现在，Python input() 接受“y”，但其余部分消失了。

Answer 1

您需要将 sys.stdin 连接到调用的 stdin。

status=subprocess.call(["someprogram"], stdin=sys.stdin)