本文介绍了如何识别输入源？的处理方法，对大家解决问题具有一定的参考价值，需要的朋友们下面随着小编来一起学习吧！
问题描述
我正在寻找一种方法来确定stdin输入是否通过管道来自另一个应用程序。
假设我有一个程序，要么接受来自管道标准输入的输入数据(当您从另一个应用程序-grep、Tail等...管道输入数据时)，要么使用默认数据文件。我不希望用户在出现提示时手动填写数据，因为没有标准输入管道。
我的简单代码示例如下所示：
from sys import stdin
for line in stdin:
print line
当我使用以下命令运行脚本时：
echo "data" | python example.py
我得到
data
脚本结束。
如果我以以下方式运行该脚本，
python example.py
它提示用户填写输入内容，然后等待。
因此，我正在寻找类似以下内容的内容，以避免在没有通过管道传输数据时出现提示。
from sys import stdin
if is_stdin_piped():
for line in stdin:
print line
else:
print "default"
这样的事情有可能发生吗？谢谢
推荐答案
如果使用输入重定向，则标准输入不会像通常那样连接到终端。您可以使用isatty函数检查文件描述符是否连接到终端：
import os
def is_stdin_piped():
return not os.isatty(0)
为更安全起见，请使用sys.stdin.fileno()而不是0。
更新：若要检查标准输入是否从文件(而不是其他程序，如IDE或外壳管道)重定向，可以使用fstat：
import os, stat
def is_stdin_a_file():
status = os.fstat(0)
return stat.S_ISREG(status.st_mode)
这篇关于如何识别输入源？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持编程学习网！
本站部分内容来源互联网,如果有图片或者内容侵犯您的权益请联系我们删除！