通过管道将输入传送到脚本,然后从用户获取输入

2022-04-16 00:00:00 python bash stdin

问题描述

假设我想通过管道将输入传递给一个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。请注意,邮件列表线程是旧的,因此您可能需要修改您选择的解决方案。

相关文章