Python Fileinput 模块介绍
作者博文地址:http://www.cnblogs.com/spiritman/
fileinput模块提供处理一个或多个文本文件的功能,可以通过使用for循环来读取一个或多个文本文件的所有行。
【默认格式】
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查看文件所有行
1 #cat user.txt 2 Jack 3 Rain 4 Tom 5 Suse 6 #################################################### 7 #cat f_replace.py 8 #!/usr/bin/env python 9 import fileinput 10 for line in fileinput.input('user.txt'): 11 print line, 12 ####################################################
13 #执行代码 14 #python f_replace.py 15 #输出结果 16 Jack 17 Rain 18 Tom 19 Suse
2、使用fileinput修改文件并备份原文件
1 #ls ./
2 user.txt
3 #cat user.txt 4 Jack 5 Rain 6 Tom 7 Suse 8 #################################################### 9 #cat f_replace.py 10 #!/usr/bin/env python 11 import fileinput 12 for line in fileinput.input('user.txt',backup='.bak',inplace=1): 13 line = line.replace('Rain','Jerry') 14 print line, 15 #################################################### 16 #执行代码
17 #python f_replace.py 18 #输出结果
19 #ls ./
20 user.txt user.txt.bak
21 #cat user.txt 22 Jack 23 Jerry 24 Tom 25 Suse
4、使用fileinput输出当前行号和行内容
1 #cat user.txt 2 Jack 3 Rain 4 Tom 5 Suse 6 #################################################### 7 #cat f_replace.py 8 #!/usr/bin/env python 9 import fileinput 10 for line in fileinput.input('user.txt'): 11 lineno = fileinput.lineno() 12 print lineno,line, 13 #################################################### 14 #执行代码 15 #python f_replace.py 16 #输出结果 17 1 Jack 18 2 Rain 19 3 Tom 20 4 Suse
5、使用fileinput判断是否为第一行
1 #cat user.txt 2 Jack 3 Rain 4 Tom 5 Suse 6 #################################################### 7 #cat f_replace.py 8 #!/usr/bin/env python 9 import fileinput 10 for line in fileinput.input('user.txt'): 11 if fileinput.isfirstline(): 12 print line, 13 break 14 #################################################### 15 #执行代码 16 #python f_replace.py 17 #输出结果 18 Jack
6、使用fileinput过滤系统文件message
1 #!/user/bin/env python 2 #coding=UTF8 3 4 import os 5 import fileinput 6 import re 7 from sys import exit 8 logfiles = os.listdir("/var/log/") 9 pattern="messages-*\d*" 10 msgfiles=[] 11 for log in logfiles: 12 if re.match(pattern,log): 13 print log 14 msgfiles.append("/var/log/"+log) 15 print(msgfiles) 16 while True: 17 find_str = raw_input("please input search key word:") 18 find_str = find_str.strip() 19 print(find_str) 20 if find_str =='': 21 continue 22 elif find_str.lower() =="q" or find_str.lower() =="quit": 23 print('bye bye!') 24 exit(0) 25 else: 26 for line in fileinput.input(msgfiles): 27 if find_str in line: 28 print("filename:%s \nfileline:%s\ncontent:%s" %(fileinput.filename(),fileinput.filelineno(),line))
作者博文地址:http://www.cnblogs.com/spiritman/
http://www.cnblogs.com/spiritman/