代码之家 › 专栏 › 技术社区 › PascalVKooten

python脚本有没有办法“知道”它是否被管道传输到?

pipe python

PascalVKooten · 技术社区 · 9 年前

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

如果没有管道连接到,它将挂起。

知道脚本是否应该从stdin读取的方法是什么;我希望除了脚本的命令行参数之外,还有其他明显的东西。

1 回复 | 直到 9 年前

AlokThakur 9 年前

是的,这是非常有可能的,我已经修改了您的示例来实现这一点。

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")

推荐文章

twisted · 将apache2 CustomLog管道传输到以其他用户身份运行的程序

3 年前

jjmerelo · 使用Proc::Async从绑定管道读取

7 年前

Vingtoft · 使用管道的Python多进程无阻塞内部通信

7 年前

marc · grep pipe sed命令

7 年前

madsantos · Bash-在不停止tail的情况下对每个grep通信运行命令-n0-f

7 年前

loretoparisi · 使用xargs将输入文件参数重定向到bash命令

7 年前

Wolfat · 无法从C中的管道执行读取?

7 年前

E.S. · Java-重定向进程输入/输出(管道)暂停[重复]

7 年前

AbhiNickz · 如何捕获通过管道传输到Rust程序的流程的输出?

7 年前

Arka Pal · 关闭管道中未使用的端部

7 年前