遍历电脑整个硬盘查找指定格式文件

#!/usr/bin/env python
# -*- encoding:utf8 -*-
import wx
import os, os.path, itertools
import re
import wmi
from threading import Thread, activeCount
import sys

reload(sys)
sys.setdefaultencoding('utf8')

def getdiskpart():
"""返回电脑磁盘分区情况
"""
diskpartlist = []
c = wmi.WMI()
for physical_disk in c.Win32_DiskDrive():
for partition in physical_disk.associators("Win32_DiskDriveToDiskPartition"):
for logical_disk in partition.associators("Win32_LogicalDiskToPartition"):
diskpartlist.append(logical_disk.Caption)
return diskpartlist

# filenamelist=[]

class MenuEventFrame(wx.Frame):
def __init__(self, parent, id):
wx.Frame.__init__(self, parent, id, u'文件查找', size=(400, 500))
self.initUI()
self.Center()
# self.search_thread

def initUI(self):
self.SetSizeHintsSz(wx.DefaultSize, wx.DefaultSize)
fgSizer2 = wx.FlexGridSizer(4, 2, 0, 0)
fgSizer2.AddGrowableCol(1)
fgSizer2.AddGrowableRow(3)
fgSizer2.SetFlexibleDirection(wx.BOTH)
fgSizer2.SetNonFlexibleGrowMode(wx.FLEX_GROWMODE_SPECIFIED)

fgSizer2.SetMinSize(wx.Size(2, 4))
self.m_staticText18 = wx.StaticText(self, wx.ID_ANY, u"路径", wx.DefaultPosition, wx.DefaultSize, 0)
self.m_staticText18.Wrap(-1)
fgSizer2.Add(self.m_staticText18, 0, wx.ALL, 5)

self.pathvalue = wx.TextCtrl(self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, 0)
fgSizer2.Add(self.pathvalue, 0, wx.ALL | wx.EXPAND, 5)

self.m_staticText19 = wx.StaticText(self, wx.ID_ANY, u"文件格式", wx.DefaultPosition, wx.DefaultSize, 0)
self.m_staticText19.Wrap(-1)
fgSizer2.Add(self.m_staticText19, 0, wx.ALL, 5)

self.fromatvalue = wx.TextCtrl(self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize, 0)
fgSizer2.Add(self.fromatvalue, 0, wx.ALL | wx.EXPAND, 5)

self.m_staticText20 = wx.StaticText(self, wx.ID_ANY, u"", wx.DefaultPosition, wx.DefaultSize, 0)
self.m_staticText20.Wrap(-1)
fgSizer2.Add(self.m_staticText20, 0, wx.ALL, 5)

self.button = wx.Button(self, wx.ID_ANY, u"查询", wx.DefaultPosition, wx.DefaultSize, 0)
fgSizer2.Add(self.button, 0, wx.ALL | wx.ALIGN_RIGHT, 5)

self.m_staticText21 = wx.StaticText(self, wx.ID_ANY, u"结果", wx.DefaultPosition, wx.DefaultSize, 0)
self.m_staticText21.Wrap(-1)
fgSizer2.Add(self.m_staticText21, 0, wx.ALL, 5)

self.m_textCtrl8 = wx.TextCtrl(self, wx.ID_ANY, wx.EmptyString, wx.DefaultPosition, wx.DefaultSize,
wx.HSCROLL | wx.TE_READONLY | wx.TE_MULTILINE)
fgSizer2.Add(self.m_textCtrl8, 0, wx.ALL | wx.EXPAND, 5)

self.SetSizer(fgSizer2)
self.Layout()
self.Centre(wx.BOTH)
self.Bind(wx.EVT_BUTTON, self.OnClick, self.button)

def work_func(self, searchfile, path, value, textctrl):
try:
print activeCount()
textctrl.Disable()
self.button.Disable()
searchfile(path, value, textctrl)
except Exception, e:
raise
finally:
textctrl.Enable()
self.button.Enable()

def OnClick(self, event):

self.m_textCtrl8.SetValue("")

if self.fromatvalue.GetValue() == '':
wx.MessageBox(u"请输入要查找的文件格式")
return

value = self.fromatvalue.GetValue().split()

if self.pathvalue.GetValue() == '':
for dirr in getdiskpart():
path = dirr
self.search_thread = Thread(target=self.work_func, args=(searchfile, path, value, self.m_textCtrl8))
self.search_thread.start()

else:
path = self.pathvalue.GetValue()
if not os.path.isdir(path):
wx.MessageBox(u'请输入正确的路径')
return
self.search_thread = Thread(target=self.work_func, args=(searchfile, path, value, self.m_textCtrl8))
self.search_thread.start()

def anyTrue(filename, sequence):
return True in itertools.imap(filename.endswith, sequence)

def searchfile(path, endswithformate, textctrl):
for basedir, dirlist, filenames in os.walk(path):
for x in filenames:
if anyTrue(x, endswithformate):
textctrl.WriteText(os.path.join(basedir, x) + os.linesep)

if __name__ == '__main__':
app = wx.App()
frame = MenuEventFrame(parent=None, id=-1)
frame.Show()
app.MainLoop()