如何识别输入源?

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)

这篇关于如何识别输入源?的文章就介绍到这了,希望我们推荐的答案对大家有所帮助,也希望大家多多支持编程学习网!

本文标题为:如何识别输入源?

基础教程推荐