python读取某个目录下的特定文件

需求:

读取/home/netcat/的所有以py结尾的文件,打印出文件名及行号

 

代码:

#!/usr/bin/env python

-^- coding: utf-8 -^-

import glob #遍历目录文件
import fileinput #同时读入多个文件
import sys

filelist=glob.glob(‘/home/netcat/*.py’) #把符合的文件放到列表中
for line in fileinput.input(filelist): #一行一行的读入文件
if fileinput.isfirstline(): #判断是否是第一行
sys.stdout.write(“\n”)
sys.stdout.write(“—reading file: %s —“%fileinput.filename()) #打印出文件名
sys.stdout.write(“\n”)
sys.stdout.write(“%d %s”%(fileinput.filelineno(),line)) #打印行号和内容