用Python统计源码行数

最新推荐文章于 2024-08-23 16:45:27 发布

caz28

最新推荐文章于 2024-08-23 16:45:27 发布

阅读量192

点赞数

分类专栏： Python 文章标签： python 开发语言统计源码行数

本文链接：https://blog.youkuaiyun.com/caz28/article/details/131078496

版权

Python 专栏收录该内容

16 篇文章

订阅专栏

该代码定义了一个名为MyCodeCount的类，用于统计指定路径下Python文件的数量和总行数，以及总的目录和文件数量。通过os模块遍历目录，对.py文件进行特殊处理。程序首先初始化计数器，然后遍历指定路径，递归计算所有子目录和文件，特别地，如果文件是.py文件，还会计算其行数。最后，通过displaycountinfo方法显示统计结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import os

class MyCodeCount:
    def __init__(self):
        self.total_files_num = 0
        self.total_dirs_num = 0
        self.py_files_num = 0
        self.py_lines_num = 0
    def displaycountinfo(self):
        print("--------------------------")
        print("Total dirs:%d"%(self.total_dirs_num))
        print("Total files:%d"%(self.total_files_num))
        print("--------------------------")
        print("Python files:%d"%(self.py_files_num))
        print("Python lines:%d"%(self.py_lines_num))
        print("--------------------------")
    def countfilelines(self, path):
        try:
            with open(path,'r',encoding='utf8') as f:
                return len(f.readlines())
        except Exception as e:
            print(e)
            print(path)
            return 0
    def countdir(self, path):
        self.total_dirs_num += 1
    def countfile(self, path):
        self.total_files_num += 1
        ext_name = os.path.splitext(path)[1]
        low_ext_name = ext_name.lower()
        if low_ext_name == '.py':
            self.py_files_num += 1
            self.py_lines_num += self.countfilelines(path)
    
    def countcode(self, top):
        for root, dirs, files in os.walk(top,topdown=False):
            for name in files:
                self.countfile(os.path.join(root,name))
            for name in dirs:
                self.countdir(os.path.join(root,name))
        
def main():
    mcc = MyCodeCount()
    mcc.countcode("D:/Work/python")
    mcc.displaycountinfo()

if __name__ == '__main__':
    main()

运行结果：

--------------------------
Total dirs:0
Total files:1
--------------------------
Python files:1
Python lines:48
--------------------------

稍作修改也可以用于统计其他类型文件。