Python:使用来自另一个命令的输入

Question

我想知道如何通过 python 脚本管理来自另一个命令的输入。

例子:

$ 猫我的文件.txt | my_python_script.py

我的脚本如何管理来自 cat 命令的输入流？我如何从这个管道命令中获取输入？

...非常感谢。

Answer 1

实现此目的的一种简单且通用的方法是使用 fileinput 模块。

import fileinput

for line in fileinput.input()
    # do things with line

这样您既可以在管道中使用脚本(正如您现在需要的那样)，也可以将一个或多个文件作为参数提供给脚本(想想 my_python_script.py input.txt input2.txt).