python fileinput模块使用

fileinput模块可以对一个或多个文件中的内容进行迭代、遍历等操作。

该模块的input()函数有点类似文件readlines()方法，区别在于前者是一个迭代对象，需要用for循环迭代，后者是一次性读取所有行。

用fileinput对文件进行循环遍历，格式化输出，查找、替换等操作，非常方便。

【典型用法】

import fileinput
for line in fileinput.input():
process(line)

【基本格式】

fileinput.input([files[, inplace[, backup[, bufsize[, mode[, openhook]]]]]])

【默认格式】

fileinput.input (files=None, 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() #关闭队列

命令行方式:

#!/usr/bin/env python3
#coding:utf8

import fileinput

for line in fileinput.input():
    lineno = fileinput.filelineno()
    print(lineno,line)

1、处理单个文件

#!/usr/bin/env python3
#coding:utf8


import fileinput

for line in fileinput.input('./1.txt'):
    if fileinput.isfirstline():
        print( fileinput.filename())
    lineno = fileinput.filelineno()
    print(lineno,line)


2、处理多个文件

import fileinput

for line in fileinput.input(['./1.txt','./2.txt']):
    if fileinput.isfirstline():
        print( fileinput.filename())
    lineno = fileinput.filelineno()
    print(lineno,line)



3、利用fileinput实现文件内容替换，并将原文件作备份

import fileinput
for line in fileinput.input("1.txt", backup=".bak", inplace=True):
    line = line.replace("55", "kkk")
    print(line)


4、利用fileinput将CRLF文件转为LF

    import fileinput  
    import sys  
      
    for line in fileinput.input(inplace=True):  
        #将Windows/DOS格式下的文本文件转为Linux的文件  
        if line[-2:] == "\r\n":    
            line = line + "\n"  
        sys.stdout.write(line) 


5、利用fileinput对文件简单处理

 import sys  
 import fileinput  
  
 for line in fileinput.input("1.txt"):  
    sys.stdout.write('=> ')  
    sys.stdout.write(line)  


6、利用fileinput批处理文件

    import fileinput  
    import glob  
      
    for line in fileinput.input(glob.glob("test*.txt")):  
        if fileinput.isfirstline():  
            print （'-'*20, 'Reading %s...' % fileinput.filename(), '-'*20  ）
        print （str(fileinput.lineno()) + ': ' + line.upper()）