將Python標準輸入流更改爲Windows控制檯輸入

Question

我想運行一個python 3腳本，通過標準輸入或手動輸入數據。

E.g.假設我想用一行打印輸入的內容。該腳本，稱爲app.py會是這個樣子：

from sys import stdin 

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

然後，我可能會在以下兩種方式運行：

1.傳遞一個文件作爲標準輸入

python app.py < input.txt 

2.提示用戶輸入

python app.py 

我的問題是，在這兩種情況下，後閱讀stdin我想提示用戶一些額外的數據。繼之前提到的例子那就是：

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()。要強制它使用控制檯，如果標準輸入重定向，您可以暫時設置sys.stdin=None或call ReadConsoleW() yourself (cumbersome)。