Remove references to the old album_art file
[mussorgsky] / src / edit_panel.py
index 8f0f72f..3de1d5b 100755 (executable)
@@ -1,8 +1,10 @@
 #!/usr/bin/env python2.5
 import hildon
-import gtk
+import gtk, gobject
 from mutagen_backend import MutagenBackend
 from player_backend import MediaPlayer
+import album_art_spec
+import os
 
 # Fields in the tuple!
 FILE_URI = 0
@@ -15,24 +17,28 @@ class MussorgskyEditPanel (hildon.StackableWindow):
 
     def __init__ (self, songs_list=None, albums_list=None, artists_list=None):
         hildon.StackableWindow.__init__ (self)
-        self.set_title ("Edit")
         self.set_border_width (12)
+        self.song_counter = 0
+        self.album_callback_id = -1
+        self.album_change_handler = -1
+        self.artist_change_handler = -1
         self.writer = MutagenBackend ()
         self.player = MediaPlayer ()
-        self.albums_list = albums_list
-        self.artists_list = artists_list
-        self.add (self.__create_view ())
+        self.albums_list = [a [0] for a in albums_list]
+        self.artists_list = [a [0] for a in artists_list]
+        self.__create_view ()
         if (songs_list):
             self.set_songs_list (songs_list)
+        self.update_title ()
         self.banner = None
 
-        self.artists_selector = None
-        self.artists_dialog = None
 
-        self.albums_selector = None
-        self.albums_dialog = None
+    def update_title (self):
+        self.set_title ("Edit (%d/%d)" % (self.song_counter+1, len (self.songs_list)))
+
         
     def set_songs_list (self, songs_list):
+        if (songs_list and len (songs_list) > 0):
             self.songs_list = songs_list
             self.set_data_in_view (songs_list [0])
             self.song_counter = 0
@@ -44,6 +50,10 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         if (self.banner and self.banner.get_property("visible")):
             self.banner.destroy ()
 
+        if (self.album_callback_id != -1):
+            gobject.source_remove (self.album_callback_id)
+            self.album_callback_id = -1
+
         if self.__is_view_dirty ():
             print "Modified data. Save!"
             self.save_metadata ()
@@ -51,6 +61,9 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         if (self.song_counter > 0):
             self.song_counter -= 1
             self.set_data_in_view (self.songs_list [self.song_counter])
+            self.update_title ()
+        else:
+            self.destroy ()
 
     def press_next_cb (self, widget):
         if (self.player.is_playing ()):
@@ -59,6 +72,10 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         if (self.banner and self.banner.get_property("visible")):
             self.banner.destroy ()
 
+        if (self.album_callback_id != -1):
+            gobject.source_remove (self.album_callback_id)
+            self.album_callback_id = -1
+
         if self.__is_view_dirty ():
             print "Modified data. Save!"
             self.save_metadata ()
@@ -66,6 +83,7 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         if (self.song_counter < len (self.songs_list) -1):
             self.song_counter += 1
             self.set_data_in_view (self.songs_list [self.song_counter])
+            self.update_title ()
         else:
             self.destroy ()
 
@@ -75,17 +93,17 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         song = self.songs_list [self.song_counter]
 
         new_song = (song[FILE_URI], song[1],
-                    self.artist_entry.get_text (),
+                    self.artist_button.get_value (),
                     self.title_entry.get_text (),
-                    self.album_entry.get_text (),
+                    self.album_button.get_value (),
                     song[MIME_KEY])
         self.songs_list [self.song_counter] = new_song
         try:
             self.writer.save_metadata_on_file (new_song[FILE_URI],
                                                new_song[MIME_KEY],
-                                               self.artist_entry.get_text (),
+                                               self.artist_button.get_value (),
                                                self.title_entry.get_text (),
-                                               self.album_entry.get_text ())
+                                               self.album_button.get_value ())
         except IOError, e:
             # This error in case of tracker returning unexistent files.
             # Uhm.... for instance after removing a memory card we are editing!
@@ -105,9 +123,9 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         song = self.songs_list [self.song_counter]
 
         return not (self.filename_data.get_text() == song[FILE_URI] and
-                    self.artist_entry.get_text () == song[ARTIST_KEY] and
+                    self.artist_button.get_value () == song[ARTIST_KEY] and
                     self.title_entry.get_text () == song[TITLE_KEY] and
-                    self.album_entry.get_text () == song[ALBUM_KEY] )
+                    self.album_button.get_value () == song[ALBUM_KEY] )
         
 
     def __create_view (self):
@@ -119,7 +137,6 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         self.filename_data = gtk.Label ("")
         filename_row.pack_start (self.filename_data, expand=True)
 
-        #play_button = gtk.Button (stock=gtk.STOCK_MEDIA_PLAY)
         play_button = hildon.Button (hildon.BUTTON_STYLE_NORMAL, hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
         img = gtk.image_new_from_stock (gtk.STOCK_MEDIA_PLAY, gtk.ICON_SIZE_BUTTON)
         play_button.set_image (img)
@@ -133,40 +150,44 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         table.set_col_spacings (12)
         table.set_row_spacings (12)
 
-        central_panel.pack_start (table)
+        central_panel.pack_start (table, fill=True)
         view_vbox.pack_start (central_panel, expand=True, fill=True)
 
-        # Artist row
-        button_artist = gtk.Button ("Artist:")
-        if (not self.artists_list):
-            button_artist.set_sensitive (False)
-        button_artist.connect ("clicked", self.artist_selection_cb)
-        table.attach (button_artist, 0, 1, 0, 1, 0, gtk.FILL|gtk.EXPAND)
-        self.artist_entry = gtk.Entry()
-        table.attach (self.artist_entry, 1, 2, 0, 1)
-
         # Title row
         label_title = gtk.Label ("Title:")
-        table.attach (label_title, 0, 1, 1, 2, 0)
+        table.attach (label_title, 0, 1, 0, 1, 0)
         self.title_entry = gtk.Entry()
-        table.attach (self.title_entry, 1, 2, 1, 2)
+        table.attach (self.title_entry, 1, 2, 0, 1)
+
+        # Artist row
+        artist_selector = hildon.TouchSelectorEntry (text=True)
+        for a in self.artists_list:
+            artist_selector.append_text (a)
+        self.artist_button = hildon.PickerButton (hildon.BUTTON_STYLE_NORMAL,
+                                                  hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
+        self.artist_button.set_title ("Artist: ")
+        self.artist_button.set_selector (artist_selector)
+        table.attach (self.artist_button, 0, 2, 1, 2)
+
 
         # Album row
-        button_album = gtk.Button ("Album:")
-        if (not self.albums_list):
-            button_album.set_sensitive (False)
-        button_album.connect ("clicked", self.album_selection_cb)
-        table.attach (button_album, 0, 1, 2, 3, 0)
-        self.album_entry = gtk.Entry()
-        table.attach (self.album_entry, 1, 2, 2, 3)
+        album_selector = hildon.TouchSelectorEntry (text=True)
+        for a in self.albums_list:
+            album_selector.append_text (a)
+        self.album_button = hildon.PickerButton (hildon.BUTTON_STYLE_NORMAL,
+                                                 hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
+        self.album_button.set_title ("Album: ")
+        self.album_button.set_selector (album_selector)
+        table.attach (self.album_button, 0, 2, 2, 3) 
+        
 
         # Album art space
-        album_button = gtk.Button ()
+        album_art_button = gtk.Button ()
         self.album_art = gtk.Image ()
         self.album_art.set_size_request (124, 124)
-        album_button.add (self.album_art)
-        album_button.connect ("clicked", self.clicked_album_art)
-        central_panel.pack_start (album_button, expand=False, fill=False)
+        album_art_button.add (self.album_art)
+        album_art_button.connect ("clicked", self.clicked_album_art)
+        central_panel.pack_start (album_art_button, expand=False, fill=False)
         
         # Buttons row
         button_box = gtk.HButtonBox ()
@@ -186,8 +207,20 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         
         view_vbox.pack_start (button_box, expand=False, fill=True, padding=6)
         
-        return view_vbox
+        self.add (view_vbox)
+
+        #menu = hildon.AppMenu ()
+        #go_to = hildon.Button (hildon.BUTTON_STYLE_NORMAL,
+        #                       hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
+        #go_to.set_title ("Go to")
+        #go_to.connect ("clicked", self.go_to_cb)
+        #menu.append (go_to)
+        #menu.show_all ()
+        #self.set_app_menu (menu)
+        
 
+    def go_to_cb (self, widget):
+        pass
 
     def set_data_in_view (self, song):
         """
@@ -195,12 +228,52 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         Song is a tuple like (filename, 'Music', title, artist, album, mime)
         """
         assert len (song) == 6
+        
         self.filename_data.set_text (song[FILE_URI])
-        self.artist_entry.set_text (song[ARTIST_KEY])
         self.title_entry.set_text (song[TITLE_KEY])
-        self.album_entry.set_text (song[ALBUM_KEY])
+        
 
-        self.album_art.set_from_file ("/home/ivan/cover-sample.jpeg")
+        # Disconnect the value-change signal to avoid extra album art retrievals
+        if (self.album_button.handler_is_connected (self.album_change_handler)):
+            self.album_button.disconnect (self.album_change_handler)
+            
+        if (self.artist_button.handler_is_connected (self.artist_change_handler)):
+            self.artist_button.disconnect (self.artist_change_handler)
+
+        # Set values in the picker buttons
+        try:
+            self.artist_button.set_active (self.artists_list.index(song[ARTIST_KEY]))
+        except ValueError:
+            print "'%s' not in artist list!?" % (song[ARTIST_KEY])
+            self.artist_button.set_value ("")
+            
+        try:
+            self.album_button.set_active (self.albums_list.index (song[ALBUM_KEY]))
+        except ValueError:
+            print "'%s' is not in the album list!?" % (song[ALBUM_KEY])
+            self.album_button.set_value ("")
+
+        # Reconnect the signals!
+        self.album_change_handler = self.album_button.connect ("value-changed",
+                                                               self.album_selection_cb)
+
+        self.artist_change_handler = self.artist_button.connect ("value-changed",
+                                                                 self.artist_selection_cb)
+
+        # Set the album art given the current data
+        has_album = False
+        if (song[ALBUM_KEY]):
+            thumb = album_art_spec.getCoverArtThumbFileName (song[ALBUM_KEY])
+           print "%s -> %s" % (song[ALBUM_KEY], thumb)
+            if (os.path.exists (thumb)):
+                self.album_art.set_from_file (thumb)
+                has_album = True
+            else:
+                self.album_callback_id = gobject.idle_add (self.retrieve_album_art,
+                                                           song[ARTIST_KEY], song[ALBUM_KEY])
+                
+        if (not has_album):
+            self.album_art.set_from_stock (gtk.STOCK_CDROM, gtk.ICON_SIZE_DIALOG)
 
         if (not song[MIME_KEY] in self.writer.get_supported_mimes ()):
             print "show notification"
@@ -208,6 +281,16 @@ class MussorgskyEditPanel (hildon.StackableWindow):
             self.banner.set_text ("Unsupported format (%s)" % song[MIME_KEY])
             self.banner.show_all ()
 
+    def retrieve_album_art (self, artist, album):
+        print "trying to get the album art"
+        (img, thumb) = self.album_art_retriever.get_album_art (artist, album)
+        if (thumb):
+            self.album_art.set_from_file (thumb)
+        else:
+            print "Unable to retrieve album art"
+
+        return False
+        
     def clicked_play (self, widget):
         if (self.player.is_playing ()):
             self.player.stop ()
@@ -219,48 +302,38 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         print "implement me, please"
 
     def album_selection_cb (self, widget):
-        if (not self.albums_selector):
-            self.albums_selector = hildon.hildon_touch_selector_new_text ()
-            for album in self.albums_list :
-                self.albums_selector.append_text (album[0])
-
-        if (not self.albums_dialog):
-            self.albums_dialog = hildon.PickerDialog (self)
-            self.albums_dialog.set_title ("Choose album...")
-            self.albums_dialog.set_selector (self.albums_selector)
-
-        response = self.albums_dialog.run ()
-        if (response == gtk.RESPONSE_OK):
-            print "Ok (%s)" % (self.albums_selector.get_current_text ())
-            self.album_entry.set_text (self.albums_selector.get_current_text ())
-        self.albums_dialog.hide ()
+        """
+        On album change, add the album the local list of albums and the selector
+        if it doesn't exist already. So we show the new entry in the selector next time.
+        """
+        song = self.songs_list [self.song_counter]
+        if (not widget.get_value () in self.albums_list):
+            print "Inserting ", widget.get_value ()
+            widget.get_selector ().prepend_text (widget.get_value ())
+            self.albums_list.insert (0, widget.get_value ())
+        self.retrieve_album_art (song[ARTIST_KEY], widget.get_value ())
 
     def artist_selection_cb (self, widget):
-        if (not self.artists_selector):
-            self.artists_selector = hildon.hildon_touch_selector_new_text ()
-            for artist in self.artists_list :
-                self.artists_selector.append_text (artist[0])
-                
-        if (not self.artists_dialog):
-            self.artists_dialog = hildon.PickerDialog (self)
-            self.artists_dialog.set_title ("Choose artist...")
-            self.artists_dialog.set_selector (self.artists_selector)
-
-        response = self.artists_dialog.run ()
-
-        if (response == gtk.RESPONSE_OK):
-            print "Ok (%s)" % (self.artists_selector.get_current_text ())
-            self.artist_entry.set_text (str(self.artists_selector.get_current_text ()))
-        self.artists_dialog.hide ()
-
+        """
+        On artist change, add the artist the local list of artists and the selector
+        if it doesn't exist already. So we show the new entry in the selector next time
+        """
+        song = self.songs_list [self.song_counter]
+        if (not widget.get_value () in self.artists_list):
+            print "Inserting artist", widget.get_value ()
+            widget.get_selector ().prepend_text (widget.get_value ())
+            self.artists_list.insert (0, widget.get_value ())
+    
 # Testing porpuses
 if __name__ == "__main__":
 
-    TEST_DATA = [("/a/b/c/d.mp3", "Music", "", "title", "album", "audio/mpeg"),
-                 ("/home/user/mufix/dejame.mp3", "Music", "", "title", "album 2", "a/b"),
-                 ("/home/user/mufix/3.mp2", "Music", "", "titlex", "album 3", "audio/mpeg")]
-    ALBUMS = [["Album %d" % i] for i in range (0, 10)]
-    ARTISTS = [["Artist %d" % i] for i in range (0, 10)]
+    TEST_DATA = [("/home/user/Music/dylan.mp3", "Music", "Bob Dylan", "Subterranean homesick blues", "Bring it all back home", "audio/mpeg"),
+                 ("/home/user/mufix/a.mp3", "Music", "", "title", "Album 2", "a/b"),
+                ("/media/mmc1/Attachments/b.m4a", "Music", "", "b", "Album 9", "audio/mpeg"),
+                 ("/home/user/mufix/3.mp2", "Music", "", "titlex", "Album 3", "audio/mpeg")]
+    #TEST_DATA = []
+    ALBUMS = [["Album %d" % i] for i in range (0, 10)] + [["Bring it all back home"]]
+    ARTISTS = [["Artist %d" % i] for i in range (0, 10)] + [["Bob Dylan"]]
     window = MussorgskyEditPanel (TEST_DATA, ALBUMS, ARTISTS)
     window.connect ("destroy", gtk.main_quit)
     window.show_all ()