python2.7偷看stdin

我想调用sys.stdin.readlines()而不从stdin中删除任何东西.我在Linux上使用Python2.7.

例如,我想要的是:

x = sys.stdin.readlines()
y = sys.stdin.readlines()

然后x和y具有相同的字符串.如果我能从stdin读取并将内容放回去,那将是可以接受的.

背景:

我有一个模块,EITHER将一个文件作为输入和一个-optional参数或“一些输入通过管道进入模块”和一个-optional参数

mymodule.py myfile -option
or
echo "some input" | mymodule.py -option

我有这个工作,它工作正常.我正在检查sys.stdin.isatty()以确定是否有任何输入管道输入.如果存在,如果命令行中有多个参数,模块将抛出错误(可能有一个-optional参数,但如果有stdin则没有指定文件)

我遇到问题的原因是因为我需要在命令行上传递单元测试,但也要在eclipse中传递.在命令行上一切正常,但看起来eclipse的PyUnit插件也使用了sys.stdin.

如果我调用sys.stdin.readlines(),那么eclipse会放弃运行单元测试.另外,即使我没有指定任何参数,eclipse也会将内容推送到sys.stdin中,这使得很难确定是否存在有效参数.

在我看来,以某种方式获取sys.stdin.readlines()而不更改内容将是一个解决方案,但我不知道如何做到这一点.解决这个问题的任何答案都会令人满意.

可能重复:Peek into stream of Popen pipeline in Python

谢谢!

编辑:没有运气的东西像……

foo = sys.stdin.readlines()
sys.stdin.write(foo)

编辑:删除了在tearDown中恢复stdin并将其放入测试函数中,但没有效果

最佳答案
从管道读取时,数据不再在管道中.你可以使用轮询机制来查看是否有任何数据要读取,但是你不能在不删除内容的情况下查看内容,至少在Linux上是这样.

如果提供了一个文件,我会忽略stdin(即首先检查文件参数),如果没有提供文件作为参数,则验证stdin.

转载注明原文:python2.7偷看stdin - 代码日志