将 python 标准输入流更改为 Windows 控制台输入

Question

我想运行一个 python 3 脚本，通过标准输入或手动输入数据给它一个文件。

例如假设我想用一行打印输入的内容。名为app.py 的脚本如下所示：

from sys import stdin

print("Input says:"+stdin.readline())

然后，我可以通过以下两种方式运行它：

1.将文件作为标准输入传递

python app.py < input.txt

2。提示用户输入

python app.py

我的问题是，在这两种情况下，在阅读标准输入后，我想提示用户输入一些额外的数据。按照前面提到的示例，它将是：

from sys import stdin

print("Input says:"+stdin.readline())

print("Did you like that? (Yes/No)")
ans = input() # This line is the issue!
if( ans == "Yes"):
    print("You liked it!")    

上面注释的行非常适用于案例 2，但对于案例 1，它会抛出 EOFError: EOF when reading a line 错误，因为它试图从文件中读取。

我想知道在那条线之前我是否可以做类似的事情

sys.stdin = SOMETHING

其中 SOMETHING 代表 Windows 控制台输入。我认为如果我能做到这一点，那么这两种情况都会正常工作。

Answer 1

您可以认为这两种情况是相同的（忽略差异）。您的脚本只从标准输入读取两行。 stdin 可以从文件重定向，也可以附加到控制台，您的脚本在许多情况下都可以正常工作：

print("Read the first line from stdin", input())
answer = input("Did you like that? (Yes/No): ") # read 2nd line
if answer == "Yes":
    print("You liked it!")

见executable code example。

问：我想要的是从文件或控制台读取一些输入（取决于运行应用程序时使用的参数）。我希望它们始终从控制台读取其他一些行。

无论标准输入是否从文件重定向，要直接从控制台读取，您可以使用msvcrt.getwch()。请参阅getpass.win_getpass() 中的示例用法。

如果您在接受 Unicode 输入时遇到问题；安装win_unicode_console package。您可以为您的 python 安装或特定用户或整个脚本或临时使用win_unicode_console.enable()/.disable() 全局启用它。要强制它使用控制台，如果 stdin 被重定向或call ReadConsoleW() yourself (cumbersome)，您可以临时设置sys.stdin=None。