关于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)