枚举文件夹所有后缀名，python脚本

import sys
import os

def addToList(fileName,el):
index=fileName.rfind(".")
if(index!=-1):
extension=fileName[index:]
try:
el[extension]
except Exception,message:
el[extension]=1

def findExtension(path,el):
if(os.path.exists(path)):
try:
items=os.listdir(path)
for item in items:
if(os.path.isfile(os.path.join(path,item))):
addToList(item,el)
else:
findExtension(os.path.join(path,item),el)
except Exception,message:
print message

if(len(sys.argv)<2):
print "Parameter not enough!"
print "Formate:countLines dir"
exit(1);

dir=sys.argv[1]
extensionList={}
findExtension(dir,extensionList)
print extensionList

评论

1 楼 colorfire 2011-02-28  

os.path.splitext(file)[1] 不就可以吗