Python程序——遍历搜索文件

关键字 搜索文件

很简单的一个小程序
搜寻路径为列表：默认搜寻D盘，可以自定义目录
利用os模块中的walk方法 遍历磁盘，找出包含关键字的目录和文件
引入了time模块计算搜索时长

# -*- coding: utf-8 -*-
# @Author  : LuoXian
# @Date    : 2020/2/12 11:49
# Software : PyCharm
# version： Python 3.8
# @File    : 遍历搜寻.py
import os
import time


# 利用os.walk遍历搜素路径
def search_file(path):
    dst_files = os.walk(path)
    for i in dst_files:
        for each_dir in i[1]:
            if find_file in each_dir:
                print(f'Find dir :\t{i[0]}\\{each_dir}')
        for each_file in i[2]:
            if find_file in each_file:
                print(f'Find file :\t{i[0]}\\{each_file}')


if __name__ == '__main__':
    find_file = input('请输入你要找的文件信息 :')
    print('搜素中...')
    t1 = time.time()
    # 搜素路径，默认为D盘
    all_path = ['D:\\']
    for each_path in all_path:
        search_file(each_path)
    t2 = time.time()
    # 搜素时长
    find_time = t2 - t1
    # 打印结果
    print('\n{:.2f}s\n搜素完毕！'.format(find_time))