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

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

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

以上这篇python遍历文件夹找出文件夹后缀为py的文件方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。