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

最新推荐文章于 2024-02-24 15:22:33 发布

原创最新推荐文章于 2024-02-24 15:22:33 发布 · 3.5k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#python

python 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一个简单的Python程序，该程序用于统计指定目录及其子目录下所有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()