自动扫描所有本地磁盘，获取所有图片信息

我的开发环境：

python2.7.5            下载地址 http://www.python.org/download/releases/2.7.5/

WMI-1.4.6.win32.exe       下载地址 https://pypi.python.org/pypi/WMI

wxPython2.9-win32-docs-demos-2.9.4.0.exe       下载地址  http://www.wxpython.org/download.php

wxPython2.9-win32-2.9.4.0-py27.exe           下载地址  http://www.wxpython.org/download.php

用到了标准库WMI、wx，需要用户自己下载安装WMIba包、wxpython包。

功能：获取本地所有磁盘，一次扫描所有图片格式的文件，包括'.bmp'，'jpeg'，'.gif'，'.psd'，'.png'，'.jpg'格式。

         获取图片的全路径，通过logging模块将扫描结果写入文件scanpicture.log。用户也可导出该log文件，保存到自己选择的目录

         自定义扫描功能暂时未添加。

　　　扫描过程是通过threading.Thread模块开启线程实现。

效果截图：

重点：

1、如何获取本地所有磁盘信息？

    通过WMI模块，objWmi = wmi.WMI(); 产生一个对象，[disk.Caption for disk in objWmi.Win32_LogicalDisk(DriveType=3)]即本地的磁盘列表，如[u'C:', u'D:', u'E:', u'F:']，还可通过disk.FreeSpace, disk.Size分别获取磁盘可用空间、磁盘大小，单位是byte

2、如何快速全路径扫描？

     使用os.walk()即可

3、如何在扫描过程中停止扫描？

     使用线程实现。

下面是详细代码：

#! /usr/bin/env python
#coding: utf-8

import wmi
import wx
import logging
import time
import os
import threading
from shutil import copy


logging.basicConfig(level=logging.DEBUG,
                format='%(asctime)s %(message)s',
                datefmt='%Y-%m-%d_%H:%M:%S.%U',
                filename='scanpicture.log',
                filemode='w')

# 自定义日志Handler来接收日志数据
class MyLogHandler(logging.Handler):
    def __init__(self,obj):
        logging.Handler.__init__(self);
        self.Object = obj;

    def emit(self,record):
        if record.levelno<self.level: return;
        tstr = time.strftime('%Y-%m-%d_%H:%M:%S.%U');
        self.Object.AppendText("[%s] %s\n"%(tstr,record.getMessage()));

class MyFrame(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self,parent=None,title="扫描本地图片",
                          pos=(100,100),size=(1180,600));
        c = wmi.WMI();
        self.lstDisks = [disk.Caption for disk in c.Win32_LogicalDisk (DriveType=3)];
        self.buildUI();
        self.CenterOnScreen()
        
    def buildUI(self):
        self.box1 = wx.BoxSizer(wx.HORIZONTAL);
        self.text = wx.TextCtrl(parent=self,style=wx.TE_AUTO_SCROLL|wx.TE_MULTILINE);
        self.box1.Add(self.text,1,wx.ALL|wx.EXPAND,5,5);
        self.box2 = wx.BoxSizer(wx.VERTICAL);
        self.btn1 = wx.Button(parent=self,label="全盘扫描");
        self.box2.Add(self.btn1,1,wx.ALL|wx.EXPAND,5,0);
        self.btn2 = wx.Button(parent=self,label="停止扫描");
        self.box2.Add(self.btn2,1,wx.ALL|wx.EXPAND,5,0);
        self.btn3 = wx.Button(parent=self,label="自定义扫描");
        self.box2.Add(self.btn3,1,wx.ALL|wx.EXPAND,5,0);
        self.btn4 = wx.Button(parent=self,label="导出数据");
        self.box2.Add(self.btn4,1,wx.ALL|wx.EXPAND,5,0);
        self.box1.Add(self.box2,0,wx.NORMAL,0,0);
        self.SetSizer(self.box1);

        self.text.SetEditable(False);
        self.btn1.Enable();
        self.btn2.Disable();

        self.btn1.Bind(wx.EVT_BUTTON,self.OnButton1Event);
        self.btn2.Bind(wx.EVT_BUTTON,self.OnButton2Event);
        self.btn3.Bind(wx.EVT_BUTTON,self.OnButton3Event);
        self.btn4.Bind(wx.EVT_BUTTON,self.OnButton4Event);
            
        handler = MyLogHandler(self.text);
        logging.getLogger().addHandler(handler);

    def OnButton1Event(self,event):
        self.btn1.Disable();
        self.btn2.Enable();
        # 检测日志，超过指定日志大小，就清空日志列表
        if len(self.text.GetValue())>1024:
            self.text.SetValue('');
        self.scan = ScanPicture(self.lstDisks)
        # daemonic为True时，表示主线程结束时子线程也要跟着退出
        self.scan.setDaemon(True)
        self.scan.start()

    def OnButton2Event(self,event):
        self.scan.StopScan()
        self.btn1.Enable();
        self.btn2.Disable();
        print(os.getpid())

    def OnButton3Event(self,event):
        '''
        @summary: 自定义扫描
        '''
        pass

    def OnButton4Event(self,event):
        '''
        @summary: 将扫描结果导出，并保存到指定地方
        '''
        userPath = os.getcwd().decode('gb2312')
        print(userPath)
        dlg = wx.DirDialog(self, "Choose a directory:",
                          defaultPath=userPath,
                          style=wx.DD_DEFAULT_STYLE
                           #| wx.DD_DIR_MUST_EXIST
                           #| wx.DD_CHANGE_DIR
                           )
        if dlg.ShowModal() == wx.ID_OK:
            savepath = dlg.GetPath()
            if savepath == userPath:
                return
            try:
                copy(os.path.join(userPath,'scanpicture.log'), savepath)
                strResult = "导出文件成功!"
            except:
                strResult = "导出文件失败!"
            dlg = wx.MessageDialog(self, strResult,
                                   'A Message Box',
                                   wx.OK | wx.ICON_INFORMATION
                                   #wx.YES_NO | wx.NO_DEFAULT 
                                   #| wx.CANCEL | wx.ICON_INFORMATION
                                   )
            dlg.ShowModal()
            dlg.Destroy()

class ScanPicture(threading.Thread):
    def __init__(self, lstDisks):
        threading.Thread.__init__(self)  
        self.dicks = lstDisks
        self.flag = True
        
    def run(self):
        for path in self.dicks:
            self.GetScanResult(path)
        
    def GetScanResult(self, path):
        '''
           @summary: 获取扫描结果
        '''
        lstImgExt = ['.bmp','jpeg','.gif','.psd','.png','.jpg'];
        for root, dirs, files in os.walk(path, topdown=False):
            for onefile in files:
                if not self.flag:
                    return
                strPath = os.path.join(root, onefile);
                if strPath[-4:] in lstImgExt:
                    logging.error(strPath);
                    
    def StopScan(self):
        '''
        @summary: 设置flag标志位，停止扫描
        '''
        self.flag = False
    
class MyApp(wx.App):
    def OnInit(self):
        self.win = MyFrame();
        self.win.Show(True);
        self.SetTopWindow(self.win);
        return True;

if __name__ == "__main__":
    app = MyApp();
    app.MainLoop();

 

注：第一篇随笔，给自己加油，希望自己能加倍努力，实现理想。