Python中的fileinput模块和tempfile模块

Python中的fileinput模块和tempfile模块
fileinput模块提供处理一个或多个文本文件的功能, 可以通过使用for..in来循环读取一个或多个文本文件内容.
import fileinput
import sys
import glob
import string

for line in fileinput.input("test.txt"): #处理一个文本文件
print line

for line in fileinput.input(glob.glob("samples/*.txt")): #处理多个文本文件
if fileinput.isfirstline():
print "------ reading %s ------/n" % fileinput.filename() #获取文件名称
print str(fileinput.lineno()) + " " + string.upper(line) #获取行号和内容, 注意这里的行号是继前面文件行号来处理的,而不是重新开始计算

tempfile模块提供给你一个快速创建一个命名唯一的临时文件

import tempfile
import os

tempf = tempfile.mktemp(dir="D:/")

print "tempfile", "=>", tempf

file = open(tempf, "w+b")
file.write("*" * 1000)
file.seek(0)
print len(file.read()), "bytes"
file.close()

try:
os.remove(tempf) #用完以后要注意删除临时文件
except OSError:
pass

当然你还可以通过file = tempfile.TemporaryFile()直接为你创建一个临时文件,并且打开.

////////////

Python之fileinput 模块

fileinput模块允许你循环一个或多个文本文件的内容, 我们可以使用这个功能去分析处理Log

import fileinput

import sys
import re

pattern = "/d{4}-/d{2}-/d{2} /d{2}:/d{2}:/d{2}"
filePath = r"C:/seeUthere_errors.log"

for eachline in fileinput.input(filePath):
m = re.search(pattern, eachline)
if m:
sys.stdout.write("==>> ")
sys.stdout.write(eachline)

还可以使用 fileinput模块获得当前行的元信息 (meta information). 其中包括 isfirstline, filename, lineno

作者：Shane
出处：http://bluescorpio.cnblogs.com
本文版权归作者和博客园共有，欢迎转载，但未经作者同意必须保留此段声明，且在文章页面明显位置给出原文连接，否则保留追究法律责任的权利。