如何识别输入源？ [英] Python - stdin - how to recognize the source of the input?

问题描述

我正在寻找一种方法来确定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)

这篇关于如何识别输入源？的文章就介绍到这了，希望我们推荐的答案对大家有所帮助，也希望大家多多支持IT屋！