Python

落月成孤倚

python实现递归统计指定目录下，指定后缀名文件的行数：

1. import os
   
2. def countLine(filename):
   
3.      fp = open(filename, 'r+',encoding = 'utf-8')
   
4.      line_count = 0
   
5.      for line in fp.readlines():
   
6.          if not line.split():        # 判断是否为空行
   
7.              line.strip()                # 去除空行
   
8.              continue
   
9.          else:
   
10.              line_count += 1
    
11.      fp.close()
    
12.      return line_count

复制代码

1.     def listdir(dir):
   
2.      files = os.listdir(dir)  #列出目录下的所有文件和目录
   
3.      for file in files:
   
4.          filepath = os.path.join(dir, file)
   
5.          # print("filename " + filepath)
   
6.          if os.path.isdir(filepath):  #如果filepath是目录，递归遍历子目录
   
7.              listdir(filepath)
   
8.          elif os.path:   #如果filepath是文件，直接统计行数
   
9.              file_type = file.split('.')                                                # 只统计规范命名文件：[文件名.扩展名]
   
10.              if len(file_type) > 1 :
    
11.                  if file_type[1] in ["js", "css", "hml", "json"]:  # 要统计的文件后缀名
    
12.                      line = countLine(filepath)
    
13.                      dirStr._count = dirStr._count + line
    
14.                      print("filename " + filepath + " lines is " + str(line))

复制代码