Remove references to the old album_art file

[mussorgsky] / src / download_dialog.py

diff --git a/src/download_dialog.py b/src/download_dialog.py
index 2a61af9..d467799 100644 (file)
--- a/src/download_dialog.py
+++ b/src/download_dialog.py
@@ -1,22 +1,27 @@
 #!/usr/bin/env python2.5
 import gtk, gobject
-from album_art import MussorgskyAlbumArt
+from album_art_thread import MussorgskyAlbumArt
+from utils import escape_html
 
 class MussorgskyAlbumArtDownloadDialog (gtk.Dialog):
 
-    def __init__ (self, parent):
+    def __init__ (self, parent, downloader=None):
         gtk.Dialog.__init__ (self,
                              "Downloading album art", parent,
                              gtk.DIALOG_DESTROY_WITH_PARENT,
                              (gtk.STOCK_CANCEL, gtk.RESPONSE_REJECT)
                         )
-        self.downloader = MussorgskyAlbumArt ()
+        if (downloader):
+            self.downloader = downloader
+        else:
+            self.downloader = MussorgskyAlbumArt ()
+            
         self.set_title ("Downloading album art")
         self.connect ("response", self.handle_response)
-        self.create_view ()
+        self.__create_view ()
         self.cancel = False
 
-    def create_view (self):
+    def __create_view (self):
 
         hbox = gtk.HBox (homogeneous=False)
 
@@ -26,8 +31,8 @@ class MussorgskyAlbumArtDownloadDialog (gtk.Dialog):
         hbox.pack_start (self.album_art, expand=False, fill=True)
 
         labels = gtk.VBox ()
-        self.status_label = gtk.Label ("")
-        labels.pack_start (self.status_label)
+        self.previous_label = gtk.Label ("")
+        labels.pack_start (self.previous_label)
         self.current_label = gtk.Label ("")
         labels.pack_start (self.current_label)
 
@@ -36,27 +41,45 @@ class MussorgskyAlbumArtDownloadDialog (gtk.Dialog):
         self.vbox.add (hbox)
 
 
-    def do_the_job (self, artist_albums):
-        TOTAL = len (artist_albums)
+    def do_the_job (self, artist_albums_model):
+        """
+        each row: ("Visible text", pixbuf, Artist, Album)
+        """
+        TOTAL = len (artist_albums_model)
         current = 1
-        
-        for (artist, album) in artist_albums:
+
+        it = artist_albums_model.get_iter_first ()
+        while (it):
             while (gtk.events_pending()):
                 gtk.main_iteration()
 
             if (self.cancel):
                 break
+
+            artist = artist_albums_model.get_value (it, 2)
+            album = artist_albums_model.get_value (it, 3)
+            
+            self.current_label.set_markup ("<small>Trying: %s - %s</small>" % (escape_html(artist),
+                                                                                   escape_html(album)))
             
             try:
+                while (gtk.events_pending()):
+                    gtk.main_iteration()
+
+                if (self.cancel):
+                    break
+                
                 (image, thumb) = self.downloader.get_album_art (artist, album)
-            except LookupError, e:
+                if thumb:
+                        pixbuf = gtk.gdk.pixbuf_new_from_file_at_size (thumb, 64, 64)
+                        artist_albums_model.set_value (it, 1, pixbuf)
+            except Exception, e:
                 print "Error processing %s - %s" % (artist, album)
                 print str(e)
-                self.album_art.set_from_stock (gtk.STOCK_CDROM, gtk.ICON_SIZE_DIALOG)
-                continue
-
-            self.status_label.set_text ("Retrieved (%d/%d)" % (current, TOTAL))
-            self.current_label.set_markup ("<b>%s - %s</b>" % (artist, album))
+                thumb = None
+            
+            self.set_title ("Downloading album art (%d/%d)" % (current, TOTAL))
+            self.previous_label.set_markup ("<b>%s - %s</b>" % (escape_html(artist), escape_html(album)))
               
             if (thumb):
                 self.album_art.set_from_file (thumb)
@@ -64,6 +87,7 @@ class MussorgskyAlbumArtDownloadDialog (gtk.Dialog):
                 self.album_art.set_from_stock (gtk.STOCK_CDROM, gtk.ICON_SIZE_DIALOG)
 
             current += 1
+            it = artist_albums_model.iter_next (it)
 
         
     def handle_response (self, widget, response_id):
@@ -74,18 +98,27 @@ class MussorgskyAlbumArtDownloadDialog (gtk.Dialog):
 
 if __name__ == "__main__":
 
-    PAIRS_NO = [("Led Zeppelin", "Led Zeppelin IV"),
-             ("Pink Floyd", "The Wall"),
-             ("Deep purple", "Made in Japan"),
-             ("", "Freakin' out"),
-             ("Dinah Washington", "")]
-
-    PAIRS = [ ("Artist %d" % i, "Album %d" %i) for i in range (0, 100)]
+    import time
+    import random
+    class MockDownloader:
+        def __init__ (self):
+            self.alt = [("../hendrix.jpeg", "../hendrix-thumb.jpeg"),
+                        ("../hoover.jpeg", "../hoover-thumb.jpeg"),
+                        ("../backbeat.jpeg", "../backbeat-thumb.jpeg"),
+                        ("../dylan.jpeg", "../dylan-thumb.jpeg")]
+            self.counter = 0
+        def get_album_art (self, artist, album, force=False):
+            time.sleep (3)
+            return  self.alt [random.randint (0, len (self.alt)-1)]
+
+    PAIRS_store = gtk.ListStore (str, gtk.gdk.Pixbuf, str, str)
+    for i in range (0, 100):
+        PAIRS_store.append (("blablabal", None, "Artist %d" % i, "Album %d" %i))
 
     def clicked_button (self):
-        aadd = MussorgskyAlbumArtDownloadDialog (w)
+        aadd = MussorgskyAlbumArtDownloadDialog (w, MockDownloader ())
         aadd.show_all ()
-        aadd.do_the_job (PAIRS)
+        aadd.do_the_job (PAIRS_store)
         
     w = gtk.Window ()
     box = gtk.VBox ()