【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
" % (dirname, cnt))
        total += cnt
    # write an blank line in the end
    fin.write("%-12s %8d

" % ("total", total))
    fin.close()