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

最新推荐文章于 2023-04-17 10:19:15 发布

zbo1301

最新推荐文章于 2023-04-17 10:19:15 发布

阅读量1.2k

点赞数

分类专栏： python

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

版权

python 专栏收录该内容

16 篇文章 1 订阅

订阅专栏

#!/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()