代码统计小程序

关于os.work

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)#一层一层打印出所有文件夹的名字

关于 sys.argv

在cmd中运行时如果.py文件中有sys.argv，则可得到一个列表，且列表的第一个元素是执行文件的文件名，后面几个元素是我们在.py旁边输入的字符，以空格分开为每个元素，所以如果编写代码统计的程序，我们可以利用sys.argv的这个特点。Sys.argv[ ]其实就是一个列表，里边的项为用户输入的参数，关键就是要明白这参数是从程序外部输入的。

代码统计程序

此程序只可在cmd下运行

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)