Python代码行统计

太过爱你忘了你带给我的痛 2022-04-23 03:16 212阅读 0赞

用py简单的实现了一个统计代码行的小工具

#/usr/bin/python
    import os
    
    #count the line of a single file
    def CountLine(path):
            tempfile = open(path)
            res = 0
            for lines in tempfile:
                    res += 1
            print "%s %d" %(path, res) #output the file path and lines
    		tempfile.close()
            return res
    
    #count the total line of a folder, sub folder included
    def TotalLine(path):
            total = 0
            for root, dirs, files in os.walk(path):
                    for item in files:
                            ext = item.split('.')
                            ext = ext[-1]  #get the postfix of the file
                            if(ext in ["cpp", "c", "h", "java", "py", "xml", "properties", "php"]):
                                    subpath = root + "/" + item
                                    total += CountLine(subpath)
            return total
    
    print "Input Path"
    path = "D:\drug\src"
    print TotalLine(path)

工具简单到，统计了源码中所有的行。包括空白行和注释行。

上面使用了os.walk来遍历目录。os.walk会直接列出一个目录下的所有文件，包括子目录的。

下面是用了os.path.listdir来实现。listdir仅仅会列出当前目录下的所有目录和文件，不会深入到子目录。

'''
    Created on 2013-3-11
    
    @author: naughty
    '''
    import os
    
    #===============================================================================
    # count the line of a single file
    #===============================================================================
    def CountLine(path):
        tempfile = open(path)
        res = 0
        for lines in tempfile:
            res += 1
        print "%s\t\t\t\t\t%d" % (path, res)
        tempfile.close()
        return res
    
    #===============================================================================
    # #count the total line of a folder, sub folder included
    #===============================================================================
    def TotalLine(path):
        total = 0
        for mfile in os.listdir(path):
            fullpath = path + "/" + mfile
            if os.path.isdir(fullpath):
                total += TotalLine(fullpath)
            elif os.path.isfile(fullpath):
                ext = mfile.split(".")
                ext = ext[-1] 
                if ext in ['c', 'cpp', 'java', 'py']:
                    total += CountLine(fullpath)   
        return total
    
    print "Input Path"
    path = "D:/drug/src"
    print TotalLine(path)