librarian.py

Go to the documentation of this file.

#!/usr/bin/python

import gtk,gtk.glade,gobject
import os,time,tempfile

from plai import utility
from plai import config

class Librarian:
        def __init__(self):
                xml=gtk.glade.XML(config.gladelibrarianfname)
        
                handlers={'on_AssistantWindow_prepare' : self._onPreparePage_onPreparePage,
                                  'on_AddFolder_clicked'           : self._onAddFolder_onAddFolder,
                                  'on_RemoveFolder_clicked'    : self._onRemoveFolder_onRemoveFolder,
                                  'on_AssistantWindow_close'   : self._onDone_onDone,
                                  'on_AssistantWindow_cancel'  : gtk.main_quit}
                                  
                xml.signal_autoconnect(handlers)
                
                f=xml.get_widget
                self._win_win                = f('AssistantWindow')             
                self._progresspage_progresspage       = f('ProgressPage')
                self._folderpage_folderpage         = f('FolderPage')
                self._view_view               = f('FolderList')         
                self._filelabel_filelabel          = f('FileLabel')
                self._titlelabel_titlelabel         = f('TitleLabel')
                self._artistlabel_artistlabel        = f('ArtistLabel')
                self._albumlabel_albumlabel         = f('AlbumLabel')
                self._durationlabel_durationlabel      = f('DurationLabel')
                self._progresspage_progresspage       = f('ProgressPage')
                self._bar_bar                = f('ProgressBar')
                self._folderlist_folderlist         = f('FolderList')
                self._launchplayertoggle_launchplayertoggle = f('LaunchPlayerToggle')
                
                self._CompletePage_CompletePage(xml.get_widget('WelcomePage'))
                self._CompletePage_CompletePage(xml.get_widget('WarningPage'))
                
                self._store_store=gtk.ListStore(gobject.TYPE_STRING)
                self._view_view.set_model(self._store_store)
                self._view_view.insert_column_with_attributes(0,'Folders',gtk.CellRendererText(),text=0)
                
                self._pulse_pulse=True
                self._numfiles_numfiles=-1
                self._curfile_curfile=0
                self._startedadding_startedadding=False
                self._starttime_starttime=-999
                (tempf,self._tempfname)=tempfile.mkstemp('.db','plaitemp')
                os.close(tempf)
                
        
        def _CompletePage(self,widget,done=True):
                self._win_win.set_page_complete(widget,done)
                
                
        def _onTimer(self):
                if not self._pulse_pulse: return False
                self._bar_bar.pulse()
                return True
                
        
        def _StopPulsing(self,prog=0):
                self._pulse_pulse=False
                self._bar_bar.set_fraction(prog)
                
        
        def _StartPulsing(self,txt=''):
                self._bar_bar.set_pulse_step(.01)
                self._bar_bar.set_text(txt)
                gobject.timeout_add(30,self._onTimer_onTimer)
                self._pulse_pulse=True
                

        def _onPipe(self,f,cond):
                if cond==gobject.IO_IN:
                        l=f.readline()[:-1]                     #strip newline at end
                        if l[-1]=='\x00': l=l[:-1]      #somehow these get in there (from wmas?) and gtk doesn't like them
                        l=unicode(l,'latin1')
                        self._HandlePipeText_HandlePipeText(l)
                        return True
                if cond==gobject.IO_ERR:
                        utility.ReportError('IO Error','')
                        return False
                if cond==gobject.IO_HUP:
                        if self._curfile_curfile!=self._numfiles_numfiles:
                                utility.ReportError('Unexpected HUP',repr(self._curfile_curfile)+'/'+repr(self._numfiles_numfiles))
                        return False
                utility.ReportError('Unknown condition','This should never happen')
                return False
                
        
        def _IncrementProgress(self):
                self._curfile_curfile=self._curfile_curfile+1
                frac=float(self._curfile_curfile)/self._numfiles_numfiles
                self._bar_bar.set_fraction(frac)
                elapsed=time.time()-self._starttime_starttime
                remain=int(((self._numfiles_numfiles-self._curfile_curfile)/(self._curfile_curfile+.001))*elapsed)+1
                self._bar_bar.set_text(utility.PrettyPrintTime(remain)+' remaining')            
                
                
        def _HandlePipeText(self,l):
                def sw(txt,widget,func=None):
                        if l.startswith(txt):
                                if widget: widget.set_text(l[len(txt):])
                                if func: func()
                                return True
                        return False
                                
                if sw('title - ', self._titlelabel_titlelabel) or \
                   sw('artist - ',self._artistlabel_artistlabel) or \
                   sw('album - ', self._albumlabel_albumlabel) or \
                   sw('fname - ', self._filelabel_filelabel,self._IncrementProgress_IncrementProgress): return
                
                if l.startswith('numfiles - '):
                        self._numfiles_numfiles=int(l[11:])
                        self._StopPulsing_StopPulsing(0)
                        self._starttime_starttime=time.time()
                elif l.startswith('length - '):
                        dur=int(l[9:])
                        self._durationlabel_durationlabel.set_text(utility.PrettyPrintTime(dur) if dur!=-999 else '')
                elif l.startswith('building onesongs table'):
                        self._StartPulsing_StartPulsing('Building index of single-song artists')
                elif l.startswith('done'):
                        self._StopPulsing_StopPulsing(1)
                        to=config.sqlfname
                        os.system('cp '+to+' '+to+'LIBRARIANBACKUP')
                        os.system('cp '+self._tempfname+' '+to)
                        self._bar_bar.set_text('DONE - total time '+utility.PrettyPrintTime(int(time.time()-self._starttime_starttime)))
                        self._CompletePage_CompletePage(self._progresspage_progresspage)
#                       self._win.set_current_page(self._win.get_current_page()+1)
                else:
                        utility.ReportError('Unexpected text in pipe',l)
                        
                
        def _onPreparePage(self,widget,page):
                def addfolder(model,path,it,folders):
                        folders.append(model.get_value(it,0))
                                        
                if page is self._progresspage_progresspage and not self._startedadding_startedadding:
                        self._startedadding_startedadding=True
                        cmd='plaibuildlib '+self._tempfname+' '
                        folders=[]
                        self._folderlist_folderlist.get_model().foreach(addfolder,folders)
                        for f in folders:
                                cmd=cmd+'"'+f+'" '
                                
                        # gobject.spawn_async ?
                        (w,r)=os.popen2(cmd)
                        gobject.io_add_watch(r,gobject.IO_IN | gobject.IO_ERR | gobject.IO_HUP, self._onPipe_onPipe)

                        self._StartPulsing_StartPulsing('Finding all the files')
                
                
        def _onAddFolder(self,widget):
                dlg=gtk.FileChooserDialog('Choose folder(s) to add to collection',self._win_win,
                                                                  gtk.FILE_CHOOSER_ACTION_SELECT_FOLDER,
                                                                  (gtk.STOCK_CANCEL, gtk.RESPONSE_REJECT,
                                                                   gtk.STOCK_OPEN, gtk.RESPONSE_ACCEPT))
                dlg.set_select_multiple(True)
                if dlg.run()==gtk.RESPONSE_ACCEPT:
                        dlg.hide()
                        for fname in dlg.get_filenames():
                                self._store_store.append((fname,))
                        self._CompletePage_CompletePage(self._folderpage_folderpage)
                else:
                        dlg.hide()


        def _onRemoveFolder(self,widget):
                (model,it)=self._view_view.get_selection().get_selected()
                model.remove(it)
                if model.iter_n_children(None)==0:
                        self._CompletePage_CompletePage(self._folderpage_folderpage,False)              
                        
                        
        def _onDone(self,widget):
                if self._launchplayertoggle_launchplayertoggle.get_active():
                        self._win_win.hide()
                        while gtk.events_pending(): gtk.main_iteration()
                        os.system('plai')
                gtk.main_quit()

---