Python-Stdin-如何识别输入的来源？

2024-04-09 22:50:47

我正在寻找一种方法,通过管道确定标准输入是否来自另一个应用程序.

假设我有一个程序,它从管道标准输入接收输入数据(从其他应用程序通过管道输入-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.

更新：要检查stdin是否从文件(而不是其他程序,例如IDE或Shell管道)重定向,可以使用fstat：

import os, stat

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

码农公寓