python的音乐本地搜索统计的小工具（速度优化版）

加入了stackless微线程，速度大大的提高！！

#coding=gb2312

import os
import fnmatch
import wx
import re
import time
import stackless


def get_allfile_name(path):   #获得全部文件夹的文件名
    l=[]
    for p,d,f in os.walk(path):
        for i in f:
            l.append(os.path.join(p,i))

    return l


class Search_music(wx.Frame):
    def __init__(self):
        wx.Frame.__init__(self,None,-1,"音乐查找 v2.0",size=(300,330),style=wx.DEFAULT_FRAME_STYLE^wx.RESIZE_BORDER)
        panel=wx.Panel(self,-1)
        self.mymusicList=['mp3','wav','wma','cue']    
#------------------------------------------------------------------------------ 文件夹
        self.st1=wx.StaticText(panel,-1,"给我一个路径 ,我将翻透你的硬盘 :",(15,15))
        self.st1_pos_y=self.st1.GetPosition().y+self.st1.GetSize().height
        self.t1=wx.TextCtrl(panel,-1,"",style=0,size=(185,25),pos=(15,self.st1_pos_y+10))
        self.t1_pos_x=self.t1.GetPosition().x+self.t1.GetSize().width+10
        self.b1=wx.Button(panel,-1,"浏览",pos=(self.t1_pos_x,self.st1_pos_y+10),size=(45,25),style=0)
        self.Bind(wx.EVT_BUTTON, self.GetDir, self.b1)


#------------------------------------------------------------查询数据类型-----------------
        self.st2=wx.StaticText(panel,-1,"音乐查询类型 :",(15,self.st1_pos_y+45))
        self.st2_pos_y=self.st2.GetPosition().y+self.st2.GetSize().height
        self.st2_pos_x=self.st2.GetPosition().y+self.st2.GetSize().width
        lb=wx.CheckListBox(panel,-1,(15,self.st2_pos_y+10),wx.DefaultSize,self.mymusicList)
        lb.SetSelection(0)
        self.lb=lb
        pos_y=lb.GetPosition().y+lb.GetSize().height+10
        pos_x=lb.GetPosition().x+lb.GetSize().width+15
#------------------------------------------------------------------------------ 统计面板
        multiLabel=wx.StaticText(panel,-1,"统计结果",(self.st2_pos_x-10,self.st1_pos_y+45))
        multiLabel_pos_y=multiLabel.GetPosition().y+multiLabel.GetSize().height+15
        self.multiText=wx.TextCtrl(panel,-1,
            "",size=(140,190),style=wx.TE_MULTILINE,pos=(self.st2_pos_x-10,self.st2_pos_y+10))#创建一个文本控件
        self.multiText.SetInsertionPoint(0)#设置插入点

#------------------------------------------------------------------------------ 进度条
        self.gauage=wx.Gauge(panel,-1,100,(15,pos_y+40),(70,20))
        self.gauage.SetBezelFace(3)
        self.gauage.SetShadowWidth(3)                                
#-------------------------------------------------------------------------------总数据发送按钮        
        btn=wx.Button(panel,-1,'搜索', (15, pos_y+70))
        self.Bind(wx.EVT_BUTTON,self.OnTestButton,btn)      


    def OnTestButton(self,event):
        dict={}
        all_word=[]  #所有要显示在面板上的
        all_songs_num=0   #歌曲总数
        mycue_number=0



        for i in self.lb.GetCheckedStrings():#做一个字典{文件类型：[文件路径名]}
            dict[i]=[]

        if self.t1.GetValue()=='' or self.lb.GetCheckedStrings()==():
            wx.MessageBox('靓仔！信息不完整哦！','出错了',wx.ICON_EXCLAMATION)
            s=''
        elif not os.path.isdir(self.t1.GetValue()):
            wx.MessageBox('靓仔！路径不合法！','出错了',wx.ICON_EXCLAMATION)
            s=''
        else:
            self.gauage.SetValue(10) #标记进度条

            All_files=get_allfile_name(self.t1.GetValue())
            def my_stackless_task(my_str):
                for f in All_files:
                    if fnmatch.fnmatch(f,'*.%s'%my_str):
                        dict[my_str].append(f)

            self.gauage.SetValue(50) #标记进度条
           # if 'ape' in self.lb.GetCheckedStrings():

            for i in self.lb.GetCheckedStrings():
                stackless.tasklet(my_stackless_task)(i)
            stackless.run()

            self.gauage.SetValue(90)                    
            for i in self.lb.GetCheckedStrings():
                #print i,len(dict[i])
                if i!='cue':
                    all_songs_num=all_songs_num+len(dict[i])
                    all_word.append('%s有：%s \n'%(i,len(dict[i])))
                else:
                    the_cue_num=len(dict[i])

            if 'cue' in self.lb.GetCheckedStrings() and len(dict['cue'])!=0:
                #print dict['cue'] 
                for mycuefile in dict['cue']:
                    #print mycuefile
                    mycue_number=mycue_number+self.find_audionumber_in_ape(mycuefile)
                all_word.append('ape内曲目：%d'%(mycue_number))   

            all_word.insert(0, '歌曲【%s】总数（含ape内音频）：%d\n'%(str(self.lb.GetCheckedStrings()),(all_songs_num+mycue_number)))

            s="\n".join(all_word)  
            self.gauage.SetValue(100)#标记进度条  

            self.multiText.SetValue(s)
            self.gauage.SetValue(0)#标记进度条         

    def find_audionumber_in_ape(self,cuefile):  #对所有的cue文件进行操作
        try:
            #print cuefile
            myfile=open(r"%s"%cuefile,'r')
            all_str=myfile.readlines()
            myfile.close()
            all_str.reverse()
            for n in all_str:
                if re.search("^  TRACK",n):
                    break
            p=re.compile('.[0-9][0-9].')
            return int(p.search(n).group())

        except:
            errorfile=open('music_error.log','a')
            errorfile.write('\n 时间:%s    文件【%s】找不到了哇\n'%(time.ctime(time.time()),cuefile))
            errorfile.close() 
            return 0  


    def GetDir(self,event):
        self.t1.SetValue(wx.DirSelector("我的音乐目录",defaultPath="",style=0,pos=wx.DefaultPosition,parent=None))


if __name__=='__main__':

    app=wx.PySimpleApp()
    frame=Search_music()
    frame.Show()
    app.MainLoop()