使用Python搜索所有驱动器中的文件

时间:2012-10-25 11:34:01

标签: python python-2.7

我使用python,我需要一个在所有驱动器中搜索我的文件的函数或库,我在Windows中只提供文件名为 F3 ,搜索计算机中的所有文件夹。 windows os,本地驱动器,我写了一个代码

import os
import win32api
paths = 'D:/'
def dir_list_folder(paths):
    for folderName in os.listdir(paths):
        if (folderName.find('.') == -1):
            folderPath = os.path.join(paths,folderName );
            dir_list_folder(folderPath);
        else:
            print ('Files is :'+ folderName );

它给了我一个好的结果但是有些类型给我一个错误,如果我不需要搜索.Zip或.RAR文件我怎么能这样做

3 个答案:

答案 0 :(得分:8)

在Windows上,您最好使用 os.walk 功能。 os.walk返回一个递归遍历源树的生成器。下面的示例显示了正则表达式搜索。

import os
import re
import win32api

def find_file(root_folder, rex):
    for root,dirs,files in os.walk(root_folder):
        for f in files:
            result = rex.search(f)
            if result:
                print os.path.join(root, f)
                break # if you want to find only one

def find_file_in_all_drives(file_name):
    #create a regular expression for the file
    rex = re.compile(file_name)
    for drive in win32api.GetLogicalDriveStrings().split('\000')[:-1]:
        find_file( drive, rex )


find_file_in_all_drivers( 'myfile\.doc' )

一些注意事项:

  1. 我正在使用正则表达式来搜索文件。为此,我提前编译RE,并将其作为参数传递。请记住规范化表达式 - 特别是如果文件名来自恶意用户。
  2. win32api.GetLogicalDriveStrings返回一个包含所有驱动程序的字符串,以0分隔。拆分它,然后切出最后一个元素
  3. 在步行期间,您可以从'dirs'中删除未打开的文件夹,例如'.git'或'.cvs'。例如,参见os.walk。 doc
  4. 为了保持样本简短,我没有传播'找到'。如果要打印所有文件,请删除“中断”。如果要在找到第一个文件后停止,请将“break”传播到find_file_in_all_drivers。

答案 1 :(得分:2)

import os
az = lambda: (chr(i)+":\\" for i in range(ord("A"), ord("Z") + 1))
for drv in az():
    for root, dirs, files in os.walk(drv):
        process_the_stuff()

答案 2 :(得分:1)

您需要为c驱动器指定驱动器。

def findall(directory):
    files=os.listdir(directory)
    for fl in files:
        path=os.path.join(directory,fl)
        if os.path.isdir(path):
            findall(path)
        else:
            dosomethingwithfile(path)
    return

基本上你遍历文件树。您必须以root目录将驱动器传递给此方法。例如。的findall( 'C:/')