一次特殊需求,需要获取项目里面代码文件的行数,一个脚本搞定
import os
from os import path
rootPath=os.getcwd()
print(rootPath)
fileArray=[]
def getAllJavaFile(fileList,rootPath):
files=os.listdir(rootPath) #获取文件列表
for fileName in files:
fullpath=path.join(rootPath,fileName)
if(path.isdir(fullpath)): #如果是文件夹 递归出所有的文件 添加到数组中
getAllJavaFile(fileList,fullpath)
else:
if '.java' in fullpath or '.kt' in fullpath: #按需过滤自己想要识别的文件
fileList.append(fullpath)
print(fullpath)
else:
pass
return fileList
lineCount=0
def readJavaFileLine(lineCount,fileArray):
length=len(fileArray)
for i in range(0,length):
print("path = ",fileArray[i])
count = len(open(fileArray[i],'rb').readlines())
lineCount=lineCount+count
return lineCount
getAllJavaFile(fileArray,rootPath)
print(readJavaFileLine(lineCount,fileArray))