当前位置:主页 > 软件编程 > Python代码 >

python遍历文件夹找出文件夹后缀为py的文件方法

时间:2021-06-20 08:49:11 | 栏目:Python代码 | 点击:

大学毕业, 想看看大学写了多少行代码。

#coding=utf-8
import os
class Solution:
 def __init__(self):
  self.dirPath = []
 
 def numberOfCode(self,path):
  for dir in os.listdir(path):
   childDir = os.path.join(path,dir)
   if os.path.isdir(childDir):
    self.numberOfCode(childDir)
   else:
    if childDir[-2:] == "py":
     self.dirPath.append(childDir)
  return self.dirPath
 
 def setCode(self):
  with open("/home/code.py","ab+") as f:
   for file in self.dirPath:
    content = open(file,"r").read()
    f.write(content)
s = Solution()
s.numberOfCode("/home/py/")
s.setCode()

您可能感兴趣的文章:

相关文章