有没有办法让 python 脚本“知道”它是否被传送到？

Question

myscript.py

import sys
if something_to_read_from_stdin_aka_piped_to:
    cmd = sys.stdin.read()
    print(cmd)
else:
    print("standard behavior")

Bash 示例：

echo "test" | python myscript.py

如果没有通过管道传输，它将挂起。

知道脚本是否应该从标准输入读取的方法是什么；我希望除了脚本的命令行参数之外还有一些明显的东西。

Answer 1

是的，很有可能，我已经修改了你的例子来实现这一点。

import sys
import os
if not os.isatty(0):
    cmd = sys.stdin.read()
    print(cmd)
else:
    print("standard behavior")

我们知道 0 是程序输入的文件描述符。 isatty 是检查文件描述符是否引用终端的测试。 正如其他用户所指出的，上述方法将判断程序是否正在从终端读取。然而，上面的编程解决了问题的目的，但仍有改进的空间。我们可以有另一种解决方案，它只检测程序是否正在从管道读取。

import sys
import os
from stat import S_ISFIFO

if S_ISFIFO(os.fstat(0).st_mode):
    print("Reading from pipe")
    cmd = sys.stdin.read()
    print(cmd)
else:
    print("not reading from pipe")