下面是fileinput模块中的常用函数：
input() #返回能够用于for循环遍历的对象 filename() #返回当前文件的名称 lineno() #返回当前已经读取的行的数量（或者序号） filelineno() #返回当前读取的行的行号 isfirstline() #检查当前行是否是文件的第一行
创建测试文件test.txt：
# cat > test.txt << EOF Hello,Python www.gxlcms.com This is a test file EOF
利用fileinput实现文件内容替换，如：file_input.p（注意文件名，别写成fileinput.py）
#!/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()。
执行结果如下：
# python file_input.py #执行file_input.py # ls test.txt* test.txt test.txt_bak # cat test.txt Hello,LinuxEye www.gxlcms.com This is a test file # cat test.txt_bak Hello,Python www.gxlcms.com This is a test file
其他测试：
>>> import fileinput
>>> for line in fileinput.input('test.txt'):
... print fileinput.filename(),fileinput.lineno(),fileinput.filelineno()
...
test.txt 1 1
test.txt 2 2
test.txt 3 3
>>> import fileinput
>>> for line in fileinput.input('test.txt'):
... if fileinput.isfirstline():
... print line,
... else:
... break
...
Hello,LinuxEye
下载本文