内置标准库之fileinput

在 Python 中使用 fileinput 模块优雅的读取文件！

此模块，实现了一个辅助类和一些函数用来快速编写访问标准输入或文件列表的循环此模块，实现了一个辅助类和一些函数用来快速访问标准输入或文件列表的循环。

• 其会迭代 sys.argv[1:] 中列出的所有文件的内容，如果列表为空则会使用 sys.stdin 替代。
• 文件都默认以文本模式打开，但可以通过调用 input() 或 FileInput 时指定 mode 形参来重载此行为。
• 如果 sys.stdin 被使用超过一次，则第二次之后的使用将不返回任何行，除非是被交互式的使用或被显式地重置，例如使用 sys.stdin.seek(0))重置。

内置标准库之fileinput

• [1] 函数概览

编号	模块名称	解释说明
1	fileinput.input()	返回能够用于 for 循环遍历的对象；创建一个 FileInput 类的实例
2	fileinput.filename()	返回当前被读取的文件的名称
3	fileinput.fileno()	返回以整数表示的当前文件“文件描述符”
4	fileinput.lineno()	返回当前已经读取的行的数量或者序号
5	fileinput.filelineno()	返回当前读取的行的行号
6	fileinput.isfirstline()	检查当前行是否是文件的第一行；是则返回 True，否则返回 False
7	fileinput.isstdin()	判断最后一行是否从 stdin 中读取；是则返回 True，否则返回 False
8	fileinput.nextfile()	关闭当前正在读取的文件，并开始读取下一个文件
9	fileinput.close()	关闭 FileInput 对象
10	fileinput.hook_compressed()	使用 gzip 和 bz2 模块透明地打开 gzip 和 bzip2 压缩的文件
11	fileinput.hook_encoded()	返回一个通过 open() 打开每个文件的钩子，使用给定的 encoding 和 errors 来读取文件

• [2] 典型用法

# 典型用法import fileinputfor line in fileinput.input():    process(line)# FileInput实例可以在with语句中被用作上下文管理器with fileinput.input(files=('spam.txt', 'eggs.txt')) as f:    for line in f:        process(line)

• 这里重点说下，fileinput.input() 这个函数的参数：

# 函数使用格式fileinput.input(files=None, inplace=False, backup='', *, mode='r', openhook=None)# 函数参数列表files: 多个文件的路径列表inplace: 用于指定是否将标准输出的结果写回到文件，此参数默认值为 Falsebackup: 用于指定备份文件的扩展名bufsize: 指定缓冲区的大小，默认为 0mode: 打开文件的格式，默认为 r（只读格式）openhook: 控制文件的打开方式，例如编码格式等

• 这个模块的实际用于在，可以接受标准输入，即可以通过管道符传入数据给 Python 程序使用。而且，fileinput 的强大之处在可以支持从多个文件同时读取内容。
• [1] 利用 fileinput 读取一个文件所有行

import fileinputfor line in fileinput.input():    filename = fileinput.filename()    lineno = fileinput.lineno()    print(f"{filename} | Line Number: {lineno} | {line}")

# 代码执行输出>>> python3 test.py data.txtdata.txt | Line Number: 1 |:  Pythondata.txt | Line Number: 2 |:  Javadata.txt | Line Number: 3 |:  C/C++data.txt | Line Number: 4 |:  Shell

• [2] 利用 fileinput 读取一个文件所有行

import fileinputdef process(line):    return line.strip() + ' line'for line in fileinput.input(inplace=True):    print process(line

# 代码执行输出>>> python3 test.py 1.txt 2.txtfirst linesecond linethird linefourth lin

• [3] 利用 fileinput 实现文件内容替换并将原文件作备份

import fileinputfor line in fileinput.input():    print(f"Tag: {line}")

# 代码执行输出>>> python3 test.py < data.txt > data_out.txt

• [4] 利用 fileinput 做正则替换

import fileinputimport refor line in fileinput.input():    line = re.sub(r'* [(.*)](#(.*))', r'<h2 id="\2">\1</h2>', line.rstrip())    print(line)

# 代码执行输出>>> python3 test.py input.txt<h2 id="author:Mark Lutz">Learning Python</h2>

• [5] 利用 fileinput 给每行日志前加上接收时的时间戳

from datetime import datetimeimport fileinputfor line in fileinput.input():    print(f'[{datetime.now()}] {line}', end='')

# 代码执行输出>>> tail -f xxx.log | python3 test.py[2021-05-09 10:41:13.514709] some good thing happend[2021-05-09 10:41:13.525803] some good thing happend[2021-05-09 10:41:13.630232] some bad thing happend

以上就是本次分享的所有内容，想要了解更多 python 知识欢迎前往公众号：Python 编程学习圈 ，发送 “J” 即可免费获取，每日干货分享