通过管道将输入传送到脚本,然后从用户获取输入
问题描述
假设我想通过管道将输入传递给一个Python程序,然后在命令行上从用户那里获得输入。
echo http://example.com/image.jpg | python solve_captcha.py
,solve_captcha.py
的内容为:
import sys
image_url = sys.stdin.readline()
# Download and open the captcha...
captcha = raw_input("Solve this captcha:")
# do some processing...
以上操作将触发EOFError: EOF when reading a line
错误。
我还尝试添加sys.stdin.close()
行,提示ValueError: I/O operation on closed file
。
您能否将信息通过管道传输到stdin
,然后从用户那里获得输入?
注意:这是一个简化的示例-请不要这样回答:"为什么要在第一种情况下这样做?"这真的很令人沮丧。我只想知道您是否可以将信息通过管道传输到stdin
,然后提示用户输入。
解决方案
此问题没有通用解决方案。最佳资源似乎是this mailing list thread。
基本上,通过管道进入程序会将程序的stdin
连接到该管道,而不是连接到终端。
邮件列表线程有两个相对simple solutionsfor*nix:
打开/dev/tty替换sys.stdin:
sys.stdin = open('/dev/tty')
a = raw_input('Prompt: ')
运行脚本时将stdin重定向到另一个文件句柄,并从中读取:
sys.stdin = os.fdopen(3)
a = raw_input('Prompt: ')
$ (echo -n test | ./x.py) 3<&0
以及suggestion to use curses。请注意,邮件列表线程是旧的,因此您可能需要修改您选择的解决方案。
相关文章