一 简介
fileinput 是python 提供的一个可以快速遍历,修改一个或者多个文件的模块。我们可以使用该模块进行文本替换 并做文件备份。
二 使用
2.1 使用方法
fileinput.input (files=None, inplace=False, backup='', bufsize=0, mode='r', openhook=None)
files 为要读入的文件 或者文件列表 files=1.txt files=['1.txt','2.txt']
inplace 是原地替换与否,为True时表示重写原文件。默认为False。
backup 备份要替换的文件的后缀 backup='.bak' files='1.txt' 备份的文件名称为 1.txt.bak
mode 读写模式,默认为只读
bufsize 读写文件的缓冲区大小,默认为0 ,如果文件比较大 则需要调整该值。
-
import fileinput
-
def process(line):
-
do_something
-
-
for line in fileinput.input():
-
process(line) #处理文件的行 然后重新写入源文件
2.2 常用的函数:
-
fileinput.input() # 读取文件的内容
-
fileinput.filename() # 文件的名称
-
fileinput.lineno() # 当前读取行的数量
-
fileinput.filelineno() # 读取行的行号
-
fileinput.isfirstline() # 当前行是否是文件第一行
-
fileinput.isstdin() # 判断最后一行是否从stdin中读取
-
fileinput.close() # 关闭队列
2.3 例子
-
In [4]: for line in fileinput.input('1.txt',inplace=1,backup='.bak'):
-
...: print line.rstrip() + ' line'
-
...:
-
In [5]: pwd
-
Out[5]: u'/Users/yangyi'
-
In [6]: ls 1.txt*
-
1.txt
-
1.txt.bak
查看备份文件
-
In [7]: cat 1.txt.bak
-
AA:BB:CC
-
aa:30:6
-
cc:50:3
-
dd:20:7
-
bb:10:2
-
ee:40:4
-
ee:60:1
查看源文件,已经被修改了
-
In [8]: cat 1.txt
-
AA:BB:CC line
-
aa:30:6 line
-
cc:50:3 line
-
dd:20:7 line
-
bb:10:2 line
-
ee:40:4 line
-
ee:60:1 line
三 小结
本文是简单介绍了fileinput 的使用方式 ,具体更详细的内容可以参考 官方文档