统计指定后缀文件的代码行数

通过递归的方式获取文件夹内指定后缀类型所有文件的行数,用于统计代码行数

import os

def getFileNames(path, list_name, postfx):
    for file in os.listdir(path):
        file_path = os.path.join(path, file)
        if os.path.isdir(file_path):
            getFileNames(file_path, list_name, postfx)
        else:
            suffix = os.path.splitext(file_path)
            if len(suffix)==2: 
                for item in postfx:
                    if item==suffix[1]:
                        list_name.append(file_path)


if __name__=="__main__":
    path=R'D:\VisualStudio2017\src'
    list_name=[]
    postfix = {'.h', '.cpp'}
    getFileNames(path, list_name, postfix)
    lineNum=0
    for file in list_name:
        print(file)
        with open(file, 'r', encoding='gb2312') as f:
            for line in f:
                lineNum = lineNum + 1
        print(lineNum)
    print(len(list_name))
    

posted @   Oliver2022  阅读(24)  评论(0编辑  收藏  举报
相关博文:
阅读排行:
· winform 绘制太阳,地球,月球 运作规律
· 超详细:普通电脑也行Windows部署deepseek R1训练数据并当服务器共享给他人
· TypeScript + Deepseek 打造卜卦网站:技术与玄学的结合
· AI 智能体引爆开源社区「GitHub 热点速览」
· 写一个简单的SQL生成工具
点击右上角即可分享
微信分享提示