代码如下:
#!/usr/bin/python
”’ file : count.py author : mike e-mail : mike_zhang@live.com”’import sys,os
extens = [“.c”,”.cpp”,”.hpp”,”.h”]linescount = 0filescount = 0
def funcount(dirname): global extens,linescount,filescount for root,dirs,filenames in os.walk(dirname): for f in filenames: fname = os.path.join(root,f) try : ext = f[f.rindex(‘.’):] if(extens.count(ext) > 0): print ‘support’ filescount += 1 print fname l_count = len(open(fname).readlines()) print fname,” : “,l_count linescount += l_count else: print ext,” : not support” except: print “error occur!” pass
if len(sys.argv) > 1 : for m_dir in sys.argv[1:]: print m_dir funcount(m_dir)else : funcount(“.”) print “files count : “,filescountprint “lines count : “,linescount
raw_input(“press enter to continue”)
使用方法1、针对本目录
代码如下:
./count.py
2、统计多个目录
代码如下:
./count.py /tmp ~