python fileinput_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 ~]#

本文参与腾讯云自媒体分享计划，欢迎正在阅读的你也加入，一起分享。