python os模块之实现多层目录文件查找

本文针对多层目录下文件的查找。

一、利用 os模块的基本操作，并利用递归的思想实现了目录多层查找。

代码如下：

import os


#dir_name: 处理文件的起始目录
def count_file(dir_name):
    #用于存目录的名字
    dir_list = []
    #将当前目录下的所有文件列出
    file_list = os.listdir(dir_name)
    #遍历当前文件名列表
    for file_name in file_list:
        #如果是目录
        if os.path.isdir(file_name):
            dir_list.append(dir_name + "/"+ file_name)
        else:
            read_file(file_name)
    # 处理目录
    for dir_path in dir_list:
        os.chdir(dir_path)
        count_file(dir_path)

#处理文件的函数
def read_file(file_name):
    with open(file_name) as f:
        text_list = f.readlines()
        print(file_name,":",text_list)


dir_name = input("请输入文件夹路径：")
count_file(dir_name)

文件处理函数并未实现什么功能，只是打印了文件名字和按行分割文件内容。

二、利用os模块的walk函数也可以达到遍历的效果

import os
def find_dir(dir_name):
    file_list = []
    for path_name, dir, files_name in os.walk(dir_name):
        for file in files_name:
            file_list.append(os.path.join(path_name, file))
    return file_list
file_list = find_dir(os.getcwd())
print(file_list)

以上代码实现了打印文件夹下所有文件的路径