python文件查找

最新推荐文章于 2023-10-02 21:02:53 发布

花浅雨越夏

最新推荐文章于 2023-10-02 21:02:53 发布

阅读量591

点赞数

分类专栏： python

本文链接：https://blog.csdn.net/zbo1301/article/details/50957009

版权

python 专栏收录该内容

16 篇文章 1 订阅

订阅专栏

<pre name="code" class="python">#!/usr/bin/env python 
#-*- encoding:utf8 -*-
import wx 
import os,os.path,itertools
import re
import wmi
from threading import Thread
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:
            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()