如何在 Python 中显示重定向的标准输入？

Question

我正在开始一个 Python 项目，其中需要标准输入重定向，使用类似于以下的代码：

import sys
import StringIO

s = StringIO.StringIO("Hello")
sys.stdin = s
a = raw_input("Type something: ")
sys.stdin = sys.__stdin__
print("You typed in: "+a)

问题是，代码运行后，显示如下：

输入一些东西：你输入：你好

有没有办法修改我的代码以显示以下内容？

输入一些东西：你好

你输入：你好

我一直在寻找高低，但还没有找到答案。如果有人有想法，我将不胜感激。 谢谢！

Answer 1

这样做。

class MyRawInputFakeOutObject( object ):
    def __init__( self, the_fake_out_input_text ):
        self.input= the_fake_out_input_text
    def __call__( self, prompt ):
        print( prompt )
        return self.input

raw_input= MyRawInputFakeOutObject( "Hello" )

import some_existing_module

some_existing_module.the_existing_main()

现在现有模块正在使用您的raw_input，而不是内置的raw_input。你可以做任何事情来提供假输入和假输出。

Answer 2

编辑：在阅读了其他答案和 cmets 之后，我想我找到了一种真正重定向标准输入的好方法。请注意，我假设您知道最终用户的 raw_inputs 的输入需要是什么。

用户代码（命名为 some_module.py）

print "running some module with 5 raw_input requests"
for x in range(5):
    value = raw_input("This is someone else's code asking its (" + str(x) + ") raw_input: ")
    print 'stdin value: ' + value

你的测试脚本（随便命名）

    import sys
    class MY_STD_IN( object ):
        def __init__(self, response_list):
            self.std_in_list = response_list
            self.std_in_length = len(response_list)
            self.index = 0

        def readline(self):
            value = self.std_in_list[self.index]      
            print value
            if self.index < self.std_in_length -1:
                self.index += 1
            else:
                self.index = 0

            return value

    predetermined_stdin_responses = ['Value 1\r', 'Value 2\r', 'Value 3\r']
    sys.stdin = MY_STD_IN( predetermined_stdin_responses )

    import some_module

运行脚本产量

运行带有 5 个 raw_input 请求的模块
这是别人的代码询问其 (0) raw_input: Value 1
标准输入值：值 1
这是别人的代码询问其 (1) raw_input: Value 2
标准输入值：值 2
这是别人的代码询问其 (2) raw_input: Value 3
标准输入值：值 3
这是别人的代码询问其 (3) raw_input: Value 1
标准输入值：值 1
这是别人的代码询问其 (4) raw_input: Value 2
标准输入值：值 2

原答案

不确定您是否正在寻找这样一个字面意思的答案，但在这里

import sys
import StringIO

s = StringIO.StringIO("Hello")
sys.stdin = s
a = raw_input("Type something: ")
sys.stdin = sys.__stdin__
print(a+"\nYou typed in: "+a)

产量：

输入一些东西：你好

你输入：你好

Answer 3

我不确定你为什么需要这样做，但你总是可以这样做：

a = raw_input("Type something: ")
if sys.stdin is not sys.__stdin__:
    print(a)
print("You typed in: "+a)

再一次，根据需要将raw_input 换成您自己的实现可能更有意义。

编辑：好的，根据您的评论，您似乎想要做一些猴子补丁。像这样的：

old_raw_input = raw_input

def new_raw_input(prompt):
    result = old_raw_input(prompt)
    if sys.stdin is not sys.__stdin__:
        print result
    return result

raw_input = new_raw_input

当然，这可能会使重定向标准输入毫无意义。