Python初学-----遍历文件夹获取文件夹下面文件的所有行数

一次特殊需求，需要获取项目里面代码文件的行数，一个脚本搞定

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