通过管道将文本传输到 Python 脚本或提示符

Question

我正在尝试用 python 编写一个非常简单的电子邮件脚本。它基本上是一个穷人的杂种狗。工作中，我们会从周围的服务器发送大量数据，直接从服务器发送会方便很多。

我坚持的部分是处理消息。我希望用户能够执行以下操作：

$ cat message.txt | emailer.py [email protected]
$ tail -n 2000 /var/log/messages | emailer.py [email protected]

这两件事都很简单。我只需 sys.stdin.read() 即可获取我的数据。

我遇到的问题是，我还想支持使用以下用法输入消息的提示：

emailer.py --attach-file /var/log/messages [email protected]

Enter Your message. Use ^D when finished.
>>   Steve,
>>   See the attached system log. See all those NFS errors around 2300 UTC today.
>>
>>   ^D

我遇到的问题是，如果我尝试 sys.stdin.read()，并且没有数据，然后我的程序会阻塞，直到 stdin 获取数据，但我无法打印提示符。 我可以采取安全的方法并使用 raw_input("Enter Your message. Use ^D when finish.") 而不是 stdin.read()，但我总是打印提示。

有没有办法查看用户是否将文本通过管道传输到 python 中而不使用会阻塞的方法？

Answer 1

您可以使用 sys.stdin.isatty 来检查脚本是否正在交互运行。例子：

if sys.stdin.isatty():
    message = raw_input('Enter your message ')
else:
    message = sys.stdin.read()