python模块fileinput

      在python脚本语言中的fileinput模块可以对一个或多个文件的内容进行迭代，编历操作.

常用的函数:

     fileinput.input()          #读取文件的内容

     fileinput.filename()    #文件的名称

     fileinput.lineno()        #当前读取行的数量

     fileinput.filelineno()   #读取行的行号

     fileinput.isfirstline()   #当前行是否是文件第一行

     fileinput.isstdin()       #判断最后一行是否从stdin中读取

     fileinput.close()         #关闭队列

1.加载fileinput模块和使用input属性

格式:

input(files=None, inplace=0, backup=‘‘, bufsize=0, mode=‘r‘, openhook=None)
files:文件的路径列表,默认是stdin方式,多文件[‘1.txt‘,‘2.txt‘]写成列表

inplace:是否将标准输出的结果写回文件

backup:备份文件的扩展名,只定义扩展名

bufsize:缓冲区的大小,默认是0

mode:读写模式,默认为只读

openhook:控制打开的文件

2.备份文件内容

[root@node1 ~]# vim 1.py

#!/bin/env python
#! _*_ coding:utf8 _*
import fileinput
for line in fileinput.input(‘qwe.py‘,‘/root/1.txt‘,‘.back‘):     #.back是备份后文件的后缀
    print line,        #后面逗号表示不换行

[root@node1 ~]# python 1.py  qwe.py
[root@node1 ~]# ll qwe.py*
-rwxr-xr-x 1 root root 315 Jun 21 18:10 qwe.py
-rwxr-xr-x 1 root root 315 Jun 21 18:07 qwe.py.back
[root@node1 ~]#

3.格式化输出

[root@node1 ~]# vim 1.py

#!/bin/env python
#! _*_ coding:utf8 _*
import fileinput
for i in fileinput.input():
    print fileinput.filename(),‘|‘,‘Line Number:‘,‘|‘,fileinput.lineno(),‘|:‘,i.lstrip(),
[root@node1 ~]# python 1.py  qwe.py
qwe.py | Line Number: | 1 |: #!/bin/env python
qwe.py | Line Number: | 2 |: #!-*- coding:UTF-8 -*-
qwe.py | Line Number: | 3 |:  qwe.py | Line Number: | 4 |: def lines(file):
qwe.py | Line Number: | 5 |: for line for file:
qwe.py | Line Number: | 6 |: yield line
qwe.py | Line Number: | 7 |: yield ‘\n‘
qwe.py | Line Number: | 8 |:  qwe.py | Line Number: | 9 |: def blocks(file):
qwe.py | Line Number: | 10 |: blosk=[]
qwe.py | Line Number: | 11 |: for line in lines(file):
qwe.py | Line Number: | 12 |: if line.strip():
qwe.py | Line Number: | 13 |: block.append(line)
qwe.py | Line Number: | 14 |: elif block:
qwe.py | Line Number: | 15 |: yield ‘‘.join(block).strip()
qwe.py | Line Number: | 16 |: block=[]
4.修改文件内容

[root@node1 ~]# vim 1.py

def process(line):
    return line.rstrip()+‘   line‘
for line in fileinput.input([‘1.txt‘,‘2.txt‘],inplace=1):
    print process(line)
[root@node1 ~]# python 1.py 1.txt
[root@node1 ~]# cat 1.txt
1111   line
2222   line
3333   line
4444   line
[root@node1 ~]# cat 2.txt
777   line
888   line
999   line
[root@node1 ~]#

5.查找文件中的内容

[root@node1 ~]# vim 1.py

#!/bin/env python
#! _*_ coding:utf8 _*
import fileinput
import sys
import re

a=‘d{2}:d{2}:d{2}‘
for i in fileinput.input(‘/var/log/yum.log‘,backup=‘.back‘,inplace=1):
    if re.search(a,i):
        sys.stdout.write(line)
[root@node1 ~]# python 1.py
[root@node1 ~]# ll /var/log/yum.log*
-rw-------  1 root root     0 Jun 21 18:36 /var/log/yum.log
-rw-------  1 root root   640 Jun  3 11:39 /var/log/yum.log.back
[root@node1 ~]#

本文出自 “一起走过的日子” 博客，请务必保留此出处http://tongcheng.blog.51cto.com/6214144/1791475

原文：http://tongcheng.blog.51cto.com/6214144/1791475