代码统计小程序

import os
res = os.walk(r'D:\LearnPython\program2')
#for path, _, file_list in res:
    #print(path)#提取出根目录下所有的文件夹
    #print(file_list)#提取出文件夹里所有的文件
for _,i,_ in res:
    print(i)#一层一层打印出所有文件夹的名字

import os
import sys
def count_code_lines(filename):
    res = os.walk(filename)
    #print(res)
    count = 0
    for path, _, file_list in res:#提取出我们输入的文件夹里面的所有文件夹地址和文件名
        print(path)#文件夹地址
        print(file_list)#文件名
        for file in file_list:
            filename = os.path.join(path, file)#拼接出文件的路径
            if filename.endswith('py'):#识别哪些是python文件
                with open(filename, 'r', encoding='utf8') as fr:
                    file_count = 0
                    for i in fr:
                        if i.startswith('#') or i.startswith('\n'):#不计算注释和换行
                            continue
                        count += 1
                        file_count += 1
                    print(f'{filename}有{file_count}行')

    print(f'总共有{count}行')

if __name__ == '__main__':
    filename = sys.argv[1]#当运行cmd的时候直接将要统计代码的地址复制在本程序名旁边即可
    #filename = '目录or文件地址'
    #count_code_lines(r'D:\LearnPython\program2\10.03')
    count_code_lines(filename)