Python Fileinput 模块介绍

作者博文地址：http://www.cnblogs.com/spiritman/

fileinput模块提供处理一个或多个文本文件的功能，可以通过使用for循环来读取一个或多个文本文件的所有行。

【默认格式】

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

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

【常用函数】

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

【使用实例】

1、使用fileinput查看文件所有行
 1 #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,
 ####################################################
13 #执行代码
 #python f_replace.py
 #输出结果
 Jack
 Rain
 Tom
 Suse

2、使用fileinput修改文件并备份原文件
 1 #ls ./
 2 user.txt
 3 #cat user.txt
 Jack
 Rain
 Tom
 Suse
 ####################################################
 #cat f_replace.py
 #!/usr/bin/env python
11 import fileinput
 for line in fileinput.input('user.txt',backup='.bak',inplace=1):
     line = line.replace('Rain','Jerry')
     print line,
 ####################################################
 #执行代码
17 #python f_replace.py
 #输出结果
19 #ls ./
20 user.txt user.txt.bak
21 #cat user.txt
 Jack
 Jerry
 Tom
 Suse

4、使用fileinput输出当前行号和行内容 
 1 #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判断是否为第一行
 1 #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
 1 #!/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)) 

作者博文地址：http://www.cnblogs.com/spiritman/