下面是fileinput模块中的常用函数:
input() #返回能够用于for循环遍历的对象
filename() #返回当前文件的名称
lineno() #返回当前已经读取的行的数量(或者序号)
filelineno() #返回当前读取的行的行号
isfirstline() #检查当前行是否是文件的第一行
创建测试文件test.txt:
# cat test.txt
Hello,Python
www.linuxeye.com
This is a test file
利用fileinput实现文件内容替换
#!/usr/bin/env python
import fileinput
for line in fileinput.input('test.txt',backup='_bak',inplace=1):
print line.replace('Python','LinuxEye'),
fileinput.close()
inplace=1:标准输出会被重定向到打开文件;backup=’_bak’,:替换文件内容之前备份后缀以_bak结尾;另外,在调用fileinput.input()之后记得要fileinput.close()。
执行结果如下:
# ls test.txt*
test.txt test.txt_bak
# cat test.txt
Hello,LinuxEye
www.linuxeye.com
This is a test file
# cat test.txt_bak
Hello,Python
www.linuxeye.com
This is a test file