Fileinput 模块

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接： https://blog.csdn.net/qq_38970783/article/details/84834213

【默认格式】

fileinput.input (files=‘filename’, inplace=False, backup=’’, bufsize=0, mode=‘r’, openhook=None)

1 files: #文件的路径列表，默认是stdin方式，多文件[‘1.txt’,‘2.txt’,…]
2 inplace: #是否将标准输出的结果写回文件，默认不取代
3 backup: #备份文件的扩展名，只指定扩展名，如.bak。如果该文件的备份文件已存在，则会自动覆盖。
4 bufsize: #缓冲区大小，默认为0，如果文件很大，可以修改此参数，一般默认即可
5 mode:　　　　　　#读写模式，默认为只读
6 openhook:　　　 #该钩子用于控制打开的所有文件，比如说编码方式等;

【常用函数】

1 fileinput.input() #返回能够用于for循环遍历的对象
2 fileinput.filename() #返回当前文件的名称
3 fileinput.lineno() #返回当前已经读取的行的数量（或者序号）
4 fileinput.filelineno() #返回当前读取的行的行号
5 fileinput.isfirstline() #检查当前行是否是文件的第一行
6 fileinput.isstdin() #判断最后一行是否从stdin中读取
7 fileinput.close() #关闭队列

【使用实例】
1、使用fileinput查看文件所有行

#cat user.txt
Jack
Rain
Tom
Suse
####################################################
#cat f_replace.py
#!/usr/bin/env python
import fileinput
for line in fileinput.input(‘user.txt’):
print line,
####################################################
#执行代码
#python f_replace.py
#输出结果
Jack
Rain
Tom
Suse

2、使用fileinput修改文件并备份原文件

#ls ./
user.txt
#cat user.txt
Jack
Rain
Tom
Suse
####################################################
#cat f_replace.py
#!/usr/bin/env python
import fileinput
for line in fileinput.input(‘user.txt’,backup=’.bak’,inplace=1):
line = line.replace(‘Rain’,‘Jerry’)
print line,
####################################################
#执行代码
#python f_replace.py
#输出结果
#ls ./
user.txt user.txt.bak
#cat user.txt
Jack
Jerry
Tom
Suse

4、使用fileinput输出当前行号和行内容

#cat user.txt
Jack
Rain
Tom
Suse
####################################################
#cat f_replace.py
#!/usr/bin/env python
import fileinput
for line in fileinput.input(‘user.txt’):
lineno = fileinput.lineno()
print lineno,line,
####################################################
#执行代码
#python f_replace.py
#输出结果
1 Jack
2 Rain
3 Tom
4 Suse

5、使用fileinput判断是否为第一行

#cat user.txt
Jack
Rain
Tom
Suse
####################################################
#cat f_replace.py
#!/usr/bin/env python
import fileinput
for line in fileinput.input(‘user.txt’):
if fileinput.isfirstline():
print line,
break
####################################################
#执行代码
#python f_replace.py
#输出结果
Jack

6、使用fileinput过滤系统文件message

#!/user/bin/env python
#coding=UTF8
import os
import fileinput
import re
from sys import exit
logfiles = os.listdir("/var/log/")
pattern=“messages-\d”
msgfiles=[]
for log in logfiles:
if re.match(pattern,log):
print log
msgfiles.append("/var/log/"+log)
print(msgfiles)
while True:
find_str = raw_input(“please input search key word:”)
find_str = find_str.strip()
print(find_str)
if find_str ==’’:
continue
elif find_str.lower() ==“q” or find_str.lower() ==“quit”:
print(‘bye bye!’)
exit(0)
else:
for line in fileinput.input(msgfiles):
if find_str in line:
print(“filename:%s \nfileline:%s\ncontent:%s” %(fileinput.filename(),fileinput.filelineno(),line))