【Python】代码行数统计
两级目录，可扩展为N级。
1 # Count the line of dir or file 2 3 import os, fnmatch, fileinput 4 5 def ChkFileType(lst): 6 tmp = lst 7 FileType = ['asm', 'c', 'cpp', 'h', 'ucf', 'v', 'java', 'm'] 8 for filename in tmp: 9 flg = 1 10 for types in FileType: 11 if fnmatch.fnmatch(filename.lower(), "*."+types): 12 flg = 0 13 break 14 if (flg): 15 lst.remove(filename) 16 17 def CountInsideFile(pathname): 18 if not os.path.exists(pathname): 19 print ("%s not exists." % pathname) 20 return -1 21 elif not os.path.isdir(pathname): 22 print ("%s not directory." % pathname) 23 return -1 24 # Get the files and check the type 25 lst = os.listdir(pathname) 26 ChkFileType(lst) 27 # build a tuple for fileinput 28 filegp = tuple(pathname+filename for filename in lst) 29 30 total = 0 31 with fileinput.input(files=filegp) as fin: 32 for line in fin: 33 total += 1 34 # Pay attention to close 35 fileinput.close() 36 print (total, pathname) 37 return total 38 39 40 if __name__ == "main": 41 desname = "count.txt" 42 En2Ch = {"compile":"编译原理", 43 "C":"C", 44 "C++":"C++", 45 "MIPS":"MIPS", 46 "NC4S":"NC4S", 47 "OS":"OS", 48 "pic":"计算机图形学", 49 "pattern":"模式识别", 50 "video":"视频标注", 51 "embed":"嵌入式", 52 "datastruct":"数据结构", 53 "database":"数据库", 54 "network":"网络", 55 "interface":"微机原理"} 56 # desname = sys.argv[1] 57 #if desname is None: 58 # desname = count.txt 59 basename = "E:\\python\\" 60 fin = open(basename+desname, "w") 61 62 ParrentPath = basename+"code\\" 63 LstDir = os.listdir(ParrentPath) 64 print (LstDir) 65 66 total = 0 67 for dirname in LstDir: 68 cnt = CountInsideFile(ParrentPath+dirname+"\\") 69 if cnt<0: 70 break 71 fin.write("%-12s %8d\n" % (dirname, cnt)) 72 total += cnt 73 # write an blank line in the end 74 fin.write("%-12s %8d\n\n" % ("total", total)) 75 fin.close()
