你希望你的脚本接受任何用户认为最简单的输入方式。包括将命令行的输出通过管道传递给该脚本、 重定向文件到该脚本,或在命令行中传递一个文件名或文件名列表给该脚本。
Python内置的 fileinput
模块让这个变得简单。如果你有一个下面这样的脚本:
#!/usr/bin/env python3
import fileinput
with fileinput.input() as f_input:
for line in f_input:
print(line, end='')
那么你就能以前面提到的所有方式来为此脚本提供输入。假设你将此脚本保存为 filein.py
并将其变为可执行文件,
那么你可以像下面这样调用它,得到期望的输出:
$ ls | ./filein.py # Prints a directory listing to stdout.
$ ./filein.py /etc/passwd # Reads /etc/passwd to stdout.
$ ./filein.py < /etc/passwd # Reads /etc/passwd to stdout.
fileinput.input()
创建并返回一个 FileInput
类的实例。
该实例除了拥有一些有用的帮助方法外,它还可被当做一个上下文管理器使用。
因此,整合起来,如果我们要写一个打印多个文件输出的脚本,那么我们需要在输出中包含文件名和行号,如下所示:
>>> import fileinput
>>> with fileinput.input('/etc/passwd') as f:
>>> for line in f:
... print(f.filename(), f.lineno(), line, end='')
...
/etc/passwd 1 ##
/etc/passwd 2 # User Database
/etc/passwd 3 #
<other output omitted>
通过将它作为一个上下文管理器使用,可以确保它不再使用时文件能自动关闭,
而且我们在之类还演示了 FileInput
的一些有用的帮助方法来获取输出中的一些其他信息。