python - 使用 stdout 时的交互式用户输入

Question

在我使用的脚本中sys.stdout.write()将处理后的数据输出到 stdout，稍后我在 CLI 上使用它将 stdout 重定向到文件:

python.exe script.py > file.out

我无法写入 python 脚本内的文件，因为无法知道重定向文件

我的问题是我还使用 raw_input() ，因为我需要用户在处理开始之前传递特定的数字，但是当我重定向标准输出时，提示不会显示 - 即脚本等待用户输入但不显示任何内容

有人可以告诉我如何处理这个问题吗？

TIA

Answer 1

看看这是否适合您:

#!/usr/bin/python
import sys
import os
# Disable buffering for stdout
sys.stdout = os.fdopen(sys.stdout.fileno(), 'w', 0)
x = raw_input(">")
print x

运行如下:

python ./test.py | tee ./file.out

现在，您将在控制台上看到输出，并且它也将被重定向到 file.out。