python

关注公众号 jb51net

关闭
首页 > 脚本专栏 > python > python os.walk函数查找windows文件

python关于os.walk函数查找windows文件方式

作者:愿此后再无WA

这篇文章主要介绍了python关于os.walk函数查找windows文件方式,具有很好的参考价值,希望对大家有所帮助,如有错误或未考虑完全的地方,望不吝赐教

前言

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

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

https://www.jb51.net/python/2959318dw.htm

直接上代码!!

查找文件

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。

不知道括号时中文还是英文的用[((].[))]表示。

总结

以上为个人经验,希望能给大家一个参考,也希望大家多多支持脚本之家。

您可能感兴趣的文章:
阅读全文