import os
#输入任意一级的目录
dir_path = input('请输入文件路径:') #设置为全局变量,方便创建存储记录的count_lines.txt文本
'''统计行数'''
def count_Lines(file_name):
fp = open(file_name, 'r+',encoding = 'utf-8')
line_count = 0
#如果此处不用去处空行,可以直接用len()计算fp.readlines()的长度,就是行数
for line in fp.readlines():
if not line.split(): # 判断是否空行
line.strip() # 去除空行
continue
else:
line_count += 1
fp.close()
return line_count
'''记录行数'''
def mulu_Statics(file_path):
lines = count_Lines(file_path)
with open(dir_path +'\\'+'count_lines.txt', 'a+') as f: #新建存储记录文件,追加模式
f.write(os.path.split(file_path)[1] + '\t' + repr(lines)+ '\n')
f.close()
#os.path.split(file_path)[1]:把文件名和上级所有目录分开成两部分,list[1]表示取文件名
'''
遍历路径下所有的txt、csv文件
'''
def list_Dir(dir_path):
dir_files = os.listdir(dir_path) #获取的不仅仅是目录,如果是文件夹和文件同事存在,都获取
for file in dir_files:
file_path = os.path.join(dir_path,file)
#判断是文件而不是目录,且 是txt或者csv文件的时候才进行统计
if os.path.isfile(file_path) and os.path.splitext(file_path)[1] in ('.txt','.csv'):
if os.path.split(file_path)[1] == 'count_lines.txt': #控制只找txt和csv文件
continue
mulu_Statics(file_path)
if os.path.isdir(file_path):
list_Dir(file_path) #递归,遍历子目录
#os.listdir对于有文件夹和文件同时存在的情况,两者都会被遍历到
if __name__ =='__main__':
list_Dir(dir_path)