Question

在一个旨在从shell运行的简单Python脚本中，我是否可以可靠地确定sys.stdin是否已从实际文件重定向而不是从另一个进程传输？

我想根据stdin是来自数据文件还是来自另一个进程通过管道流来改变运行时行为。

正如预期的那样， isatty() 在两种情况下均返回False。这是一个快速的isatty()测试：

# test.py
import os
import sys
print sys.stdin.isatty()
print os.isatty(sys.stdin.fileno())

测试：

python test.py < file.txt

产生

False
False

和

ls -al | python test.py

产生

False
False

是否有这样做的pythonic方法？

Unix / Linux特定版本很好，但知道是否可以以便携方式执行此操作会很好。

编辑请注意评论者：为什么我关心？好吧，在我的情况下，我想处理从另一个进程传输时不规则间隔的时间戳数据;当我从文件中播放预先录制的数据时，我想使用固定或可变延迟重放它。

我同意使用更干净的方法可能是有利的（我可以想到几个，包括在播放流中插入延迟的中间脚本），但我最终好奇。

Answer 1

您正在寻找stat宏：

import os, stat

mode = os.fstat(0).st_mode
if stat.S_ISFIFO(mode):
     print "stdin is piped"
elif stat.S_ISREG(mode):
     print "stdin is redirected"
else:
     print "stdin is terminal"

如何确定sys.stdin是从文件重定向还是从另一个进程管道传输？

1 个答案: