Question

我正在寻找一种方法来确定stdin输入是否来自另一个应用程序是否通过管道传输。

假设我有一个程序，接受来自管道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"

这样的事情可能吗？感谢

Answer 1

如果使用输入重定向，则标准输入不会像通常那样连接到终端。您可以使用isatty功能检查文件描述符是否连接到终端：

import os

def is_stdin_piped():
    return not os.isatty(0)

为了安全起见，请使用sys.stdin.fileno()代替0。

更新：要检查stdin是否从文件（而不是其他程序，例如IDE或shell管道）重定向，您可以使用fstat：

import os, stat

def is_stdin_a_file():
    status = os.fstat(0)
    return stat.S_ISREG(status.st_mode)