【Python】代码行数统计

两级目录，可扩展为N级。

 # Count the line of dir or file

 import os, fnmatch, fileinput

 def ChkFileType(lst):
     tmp = lst
     FileType = ['asm', 'c', 'cpp', 'h', 'ucf', 'v', 'java', 'm']
     for filename in tmp:
         flg = 1
         for types in FileType:
             if fnmatch.fnmatch(filename.lower(), "*."+types):
                flg = 0
                break
         if (flg):
             lst.remove(filename)

 def CountInsideFile(pathname):
     if not os.path.exists(pathname):
         print ("%s not exists." % pathname)
         return -1
     elif not os.path.isdir(pathname):
         print ("%s not directory."  % pathname)
         return -1
     # Get the files and check the type
     lst = os.listdir(pathname)
     ChkFileType(lst)
     # build a tuple for fileinput
     filegp = tuple(pathname+filename for filename in lst)

     total = 0
     with fileinput.input(files=filegp) as fin:
         for line in fin:
             total += 1
     # Pay attention to close
     fileinput.close()
     print (total, pathname)
     return total

 if __name__ == "main":
     desname = "count.txt"
     En2Ch = {"compile":"编译原理",
              "C":"C",
              "C++":"C++",
              "MIPS":"MIPS",
              "NC4S":"NC4S",
              "OS":"OS",
              "pic":"计算机图形学",
              "pattern":"模式识别",
              "video":"视频标注",
              "embed":"嵌入式",
              "datastruct":"数据结构",
              "database":"数据库",
              "network":"网络",
              "interface":"微机原理"}
     # desname = sys.argv[1]
     #if desname is None:
     #    desname = count.txt
     basename = "E:\\python\\"
     fin = open(basename+desname, "w")

     ParrentPath = basename+"code\\"
     LstDir = os.listdir(ParrentPath)
     print (LstDir)

     total = 0
     for dirname in LstDir:
         cnt = CountInsideFile(ParrentPath+dirname+"\\")
         if cnt<0:
             break
         fin.write("%-12s %8d\n" % (dirname, cnt))
         total += cnt
     # write an blank line in the end
     fin.write("%-12s %8d\n\n" % ("total", total))
     fin.close()