python 基于os.walk函数查找windows文件

前言

OS模块是与操作系统交互的模块。这其中一个非常实际且常用的功能莫过于帮助我们查找需要的文件。

由于前面我发布过一篇比较详细的OS模块讲解文章，这里我就不再多述。

python 畅玩os模块 顺便批量修改个图片尺寸_lishuaigell的博客-CSDN博客https://blog.csdn.net/lishuaigell/article/details/122507508

直接上代码！！

目录

前言

查找文件

其他需求

查找目录

提升速度

根据文件格式查找文件

模糊查找

---

查找文件

import os


# 文件查找
def file_search():
    # os.walk的三个返回值分别是 路径 目录列表 文件列表  它会不断的进入每个目录获取信息
    for dir_path, listdir, files in os.walk(path):
        for file in files:
            info = dir_path + os.sep + file
            print(info)
            # 如果名字符合则用列表储存起来
            # 可以把 in 换成 == 吗？ 不建议，因为可能传入的文件名不带格式的，但是实际文件是有格式后缀的，如XX.txt
            if target_name in file:
                result.append(info)
    print()
    output()


# 文件输出
def output():
    if result:
        print("文件出现在如下位置：")
        for i in result:
            print(i)
    else:
        print("未查询到文件！")


if __name__ == '__main__':
    path = input("请输入查找路径(默认在C盘下寻找,可直接回车):")
    path = 'C:\\' if not path else path  # 若有输出路径则使用给出的，否则使用默认的C盘
    # 如果给出的路径存在才进行查找
    if os.path.exists(path):
        target_name = input("请输入需要查找的文件名：")
        result = []
        file_search()
    else:
        print("您输入的路径不存在！")

其他需求

查找目录

我们可以使用os.path模块下的basename方法，通过路径来截取目录名，再看是否匹配。

那样的话更方便，还不用看文件列表了。

def file_search():
    # os.walk的三个返回值分别是 路径 目录列表 文件列表  它会不断的进入每个目录获取信息
    for dir_path, listdir, files in os.walk(path):
        if os.path.basename(dir_path) == target_name:
            result.append(dir_path)
    print()
    output()

提升速度

print方法会消耗不少时间，适当的时候我们可以不查看当前的路径信息，直接等结果就好了。另外，如果你知道有多少个文件的话可以设置一个终止条件，当找到的文件个数满足条件时，直接终止程序。

根据文件格式查找文件

如果知道文件格式，可以根据目标文件的后缀匹配。如：我想在某个找所有MP3文件或其他，就可以这么做

# 文件查找
def file_search():
    # os.walk的三个返回值分别是 路径 目录列表 文件列表  它会不断的进入每个目录获取信息
    for dir_path, listdir, files in os.walk(path):
        for file in files:
            info = dir_path + os.sep + file
            # endswith 以···结尾  如果要查MP3文件，那么 target_name = MP3
            if file.endswith(target_name):
                result.append(info)
    print()
    output()

模糊查找

如果你知道某个比较重要的关键词可以用str1 in str2语句判断是否是符合条件的文件；如果你知道更多特征可以使用正则表达式查找。

比如我想找这几个文件，但我只知道Redis 和 （）两个关键字符串，那么应该这么写。

# 文件查找
def file_search():
    # os.walk的三个返回值分别是 路径 目录列表 文件列表  它会不断的进入每个目录获取信息
    for dir_path, listdir, files in os.walk(path):
        for file in files:
            info = dir_path + os.sep + file
            #  target_name = '.*[Rr]edis.*[（(].[）)]'
            ret = re.match(target_name, file)
            if ret:
                result.append(info)
    print()
    output()

其中，.表示任意单个字符，*表示*前面的那个字符能够出现任意多次。.*表示任意字符出现任意多次。我不知道redis的r是大写还是小写所以用[Rr]表示，[Rr]表示单个字符可选R也可选r。不知道括号时中文还是英文的用[（(].[）)]表示。