import os


def fn(path, tail2):
    key = dict()
    for i in os.listdir(path):
        sub_path = os.path.join(path, i)
        if os.path.isdir(sub_path):  # 递归遍历子目录下文件及目录
            key.update({i: dict()})  # 父级标签
            key[i].update(fn(sub_path, tail2))
        elif os.path.isfile(sub_path):  # 读取目录下文件
            tail1 = i.split('.')[-1]  # 取出后缀
            key.update({i: list()})  # 子级标签
            # 读取后缀为txt的目标文件内容
            if tail1 == tail2:
                with open(sub_path, "r", encoding="utf-8") as f:
                    text = f.readlines()
                for j in text:
                    key[i].append(j.strip())
    return key


if __name__ == '__main__':
    path = "/home/klein/.cache/pip/http"
    key = fn(path, "txt")
    print(key)

        当一个目录下有多个层级目录或者多个文件时，例如python安装包的缓存文件，会耗费很多时间找到目标文件，所以实现了一个递归方法。该方法可以递归遍历目录和目录下的文件，并查看目标文件的内容。

  • os.path.isdir(path):判断路径是否为目录
  • os.path.isfile(path):判断路径是否为文件

        效果如下：

python 递归目录 python递归遍历目录_后缀