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