• 13.1 通过重定向/管道/文件接受输入
    • 问题
    • 解决方案
    • 讨论

    13.1 通过重定向/管道/文件接受输入

    问题

    你希望你的脚本接受任何用户认为最简单的输入方式。包括将命令行的输出通过管道传递给该脚本、重定向文件到该脚本,或在命令行中传递一个文件名或文件名列表给该脚本。

    解决方案

    Python内置的 fileinput 模块让这个变得简单。如果你有一个下面这样的脚本:

    1. #!/usr/bin/env python3
    2. import fileinput
    3.  
    4. with fileinput.input() as f_input:
    5. for line in f_input:
    6. print(line, end='')

    那么你就能以前面提到的所有方式来为此脚本提供输入。假设你将此脚本保存为 filein.py 并将其变为可执行文件,那么你可以像下面这样调用它,得到期望的输出:

    1. - ls | ./filein.py # Prints a directory listing to stdout.
    2. - ./filein.py /etc/passwd # Reads /etc/passwd to stdout.
    3. - ./filein.py < /etc/passwd # Reads /etc/passwd to stdout.

    讨论

    fileinput.input() 创建并返回一个 FileInput 类的实例。该实例除了拥有一些有用的帮助方法外,它还可被当做一个上下文管理器使用。因此,整合起来,如果我们要写一个打印多个文件输出的脚本,那么我们需要在输出中包含文件名和行号,如下所示:

    1. >>> import fileinput
    2. >>> with fileinput.input('/etc/passwd') as f:
    3. >>> for line in f:
    4. ... print(f.filename(), f.lineno(), line, end='')
    5. ...
    6. /etc/passwd 1 ##
    7. /etc/passwd 2 # User Database
    8. /etc/passwd 3 #
    9.  
    10. <other output omitted>

    通过将它作为一个上下文管理器使用,可以确保它不再使用时文件能自动关闭,而且我们在之后还演示了 FileInput 的一些有用的帮助方法来获取输出中的一些其他信息。

    原文:

    http://python3-cookbook.readthedocs.io/zh_CN/latest/c13/p01_accept_input_via_redirect_pips_or_input_files.html