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

2019-11-21 01:55:17 阅读：307 来源： 互联网

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

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

标签：stdin,python
来源： https://codeday.me/bug/20191121/2048570.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9

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