如何在bash中使python脚本"可管理"？

Question

我写了一个脚本,我希望它可以在bash中管道化.就像是:

echo "1stArg" | myscript.py

可能吗？怎么样？

Answer 1

看到这个简单echo.py:

import sys

if __name__ == "__main__":
    for line in sys.stdin:
        sys.stderr.write("DEBUG: got line: " + line)
        sys.stdout.write(line)

运行:

ls | python echo.py 2>debug_output.txt | sort

输出:

echo.py
test.py
test.sh

debug_output.txt内容:

DEBUG: got line: echo.py
DEBUG: got line: test.py
DEBUG: got line: test.sh

Answer 2

我将使用一个grep示例来补充其他答案,该示例使用fileinput来实现UNIX工具的典型行为:1)如果没有特定文件,则从stdin读取数据; 2)许多文件可以作为参数发送; 3)-表示stdin.

import fileinput
import re
import sys

def grep(lines, regexp):
    return (line for line in lines if regexp.search(line))

def main(args):
    if len(args) < 1:
        print("Usage: grep.py PATTERN [FILE...]", file=sys.stderr)
        return 2 
    regexp = re.compile(args[0])
    input_lines = fileinput.input(args[1:])
    for output_line in grep(input_lines, regexp):
        sys.stdout.write(output_line)

if __name__ == '__main__':
    sys.exit(main(sys.argv[1:]))

例:

$ seq 1 20 | python grep.py "4"
4
14

Answer 3

在您的Python脚本中,您只需阅读stdin.

Answer 4

从 stdin 读取的所有内容都是“可管道化的”。管道只是将前一个程序的标准输出重定向到后者。