Fixes MB#8912, crashes when saving song metadata
[mussorgsky] / src / edit_panel_tm.py
index 7646f72..47e9807 100644 (file)
@@ -3,22 +3,27 @@ import hildon
 import gtk, gobject
 from mutagen_backend import MutagenBackend
 from player_backend import MediaPlayer
+from utils import escape_html
 import album_art_spec
 import os
 
 # Fields in the tuple!
-FILE_URI = 0
-ARTIST_KEY = 2
-TITLE_KEY = 3
-ALBUM_KEY = 4
-MIME_KEY = 5
+# Shared with browse_panel
+URI_COLUMN = 0
+ARTIST_COLUMN = 2
+TITLE_COLUMN = 3
+ALBUM_COLUMN = 4
+MIME_COLUMN = 5
+UI_COLUMN = 6
+SEARCH_COLUMN = 7
+
+THEME_PATH = "/usr/share/themes/default"
 
 class MussorgskyEditPanel (hildon.StackableWindow):
 
     def __init__ (self):
         hildon.StackableWindow.__init__ (self)
         self.set_border_width (12)
-        self.album_callback_id = -1
         self.album_change_handler = -1
         self.artist_change_handler = -1
         self.writer = MutagenBackend ()
@@ -27,6 +32,7 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         self.data_loaded = False
         self.artist_list = None
         self.albums_list = None
+        self.current = None
         self.connect ("delete-event", self.close_function)
 
     def close_function (self, widget, event):
@@ -47,6 +53,7 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         return self.model.get (self.current, 0, 1, 2, 3, 4, 5)
 
     def set_model (self, model, current=None):
+        assert type(model) == gtk.TreeModelFilter
         try:
             if self.artists_list or self.albums_list:
                 pass
@@ -91,10 +98,6 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         if (self.player.is_playing ()):
             self.player.stop ()
 
-        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 ()
@@ -112,17 +115,12 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         if (self.player.is_playing ()):
             self.player.stop ()
 
-        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 ()
 
         self.current = self.model.iter_next (self.current)
         if (not self.current):
-            print "Destroy"
             self.destroy ()
         else:
             self.set_data_in_view (self.get_current_row ())
@@ -136,23 +134,26 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         # 0 - filename -> doesn't change
         # 1 - "Music"  -> doesn't change
         # 5 - mimetype -> doesn't change
-        if (type (self.model) == gtk.TreeModelFilter):
-            print "Ye un filtered"
-            m = self.model.get_model ()
-            c = self.model.convert_iter_to_child_iter (self.current)
-        else:
-            # Very unlikely
-            m = self.model
-            c = self.current
+        m = self.model.get_model ()
+        c = self.model.convert_iter_to_child_iter (self.current)
+
+        artist = self.artist_button.get_value ()
+        title = self.title_entry.get_text ()
+        album = self.album_button.get_value ()
+        text = "<b>%s</b>\n<small>%s</small>" % (escape_html (title),
+                                                 escape_html (artist) + " / " + escape_html (album))
+        search_str = artist.lower () + " " + title.lower () + " " + album.lower ()
 
+        uri, mime = m.get (c, URI_COLUMN, MIME_COLUMN)
         m.set (c,
-               2, self.artist_button.get_value (),
-               3, self.title_entry.get_text (),
-               4, self.album_button.get_value ())
-        new_song = self.get_current_row ()
+               ARTIST_COLUMN, artist,
+               TITLE_COLUMN, title,
+               ALBUM_COLUMN, album,
+               UI_COLUMN, text,
+               SEARCH_COLUMN, search_str)
         try:
-            self.writer.save_metadata_on_file (new_song[FILE_URI],
-                                               new_song[MIME_KEY],
+            self.writer.save_metadata_on_file (uri,#new_song[URI_COLUMN],
+                                               mime, #new_song[MIME_COLUMN],
                                                self.artist_button.get_value (),
                                                self.title_entry.get_text (),
                                                self.album_button.get_value ())
@@ -160,12 +161,6 @@ class MussorgskyEditPanel (hildon.StackableWindow):
             # This error in case of tracker returning unexistent files.
             # Uhm.... for instance after removing a memory card we are editing!
             pass
-            #dialog = gtk.MessageDialog (self,
-            #                            gtk.DIALOG_DESTROY_WITH_PARENT,
-            #                            gtk.MESSAGE_ERROR,
-            #                            gtk.BUTTONS_CLOSE,
-            #                            "%s" % str(e));
-            #dialog.run ()
         
 
     def __is_view_dirty (self):
@@ -174,19 +169,16 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         """
         song = self.get_current_row ()
 
-        return not (self.filename_data.get_text() == song[FILE_URI] and
-                    self.artist_button.get_value () == song[ARTIST_KEY] and
-                    self.title_entry.get_text () == song[TITLE_KEY] and
-                    self.album_button.get_value () == song[ALBUM_KEY] )
+        return not (self.filename_data.get_text() == song[URI_COLUMN] and
+                    self.artist_button.get_value () == song[ARTIST_COLUMN] and
+                    self.title_entry.get_text () == song[TITLE_COLUMN] and
+                    self.album_button.get_value () == song[ALBUM_COLUMN] )
         
 
     def __create_view (self):
         view_vbox = gtk.VBox (homogeneous=False, spacing = 12)
 
         filename_row = gtk.HBox ()
-        filename_label = gtk.Label ()
-        filename_label.set_markup ("<small>Filename:</small>")
-        filename_row.pack_start (filename_label, expand=False, padding=12);
         self.filename_data = gtk.Label ("")
         filename_row.pack_start (self.filename_data, expand=True)
 
@@ -209,7 +201,7 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         table.attach (self.title_entry, 1, 2, 0, 1)
 
         # Artist row
-        self.artist_button = hildon.PickerButton (gtk.HILDON_SIZE_FINGER_HEIGHT,
+        self.artist_button = hildon.PickerButton (gtk.HILDON_SIZE_THUMB_HEIGHT,
                                                   hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
         self.artist_button.set_title ("Artist: ")
         # Set data will set the selector
@@ -217,11 +209,11 @@ class MussorgskyEditPanel (hildon.StackableWindow):
 
 
         # Album row
-        self.album_button = hildon.PickerButton (gtk.HILDON_SIZE_FINGER_HEIGHT,
+        self.album_button = hildon.PickerButton (gtk.HILDON_SIZE_THUMB_HEIGHT,
                                                  hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
         self.album_button.set_title ("Album: ")
         # set_data will set the selector
-        table.attach (self.album_button, 0, 2, 2, 3) 
+        table.attach (self.album_button, 0, 2, 2, 3)
         
 
         # Album art space
@@ -231,8 +223,8 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         
         # Buttons row
         button_box = gtk.Toolbar ()
-
-        play_button = gtk.ToolButton (gtk.image_new_from_stock (gtk.STOCK_MEDIA_PLAY, gtk.ICON_SIZE_BUTTON))
+        play_image = os.path.join (THEME_PATH, "mediaplayer", "Play.png")
+        play_button = gtk.ToolButton (gtk.image_new_from_file (play_image))
         play_button.connect ("clicked", self.clicked_play)                   
         play_button.set_expand (True)
         button_box.insert (play_button, -1)
@@ -241,12 +233,14 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         separator.set_expand (True)
         button_box.insert  (separator, -1)
 
-        back_button = gtk.ToolButton (gtk.image_new_from_stock (gtk.STOCK_GO_BACK, gtk.ICON_SIZE_BUTTON))
+        back_image = os.path.join (THEME_PATH, "mediaplayer", "Back.png")
+        back_button = gtk.ToolButton (gtk.image_new_from_file (back_image))
         back_button.connect ("clicked", self.press_back_cb)
         back_button.set_expand (True)
         button_box.insert (back_button, -1)
 
-        next_button = gtk.ToolButton (gtk.image_new_from_stock (gtk.STOCK_GO_FORWARD, gtk.ICON_SIZE_BUTTON))
+        next_image = os.path.join (THEME_PATH, "mediaplayer", "Forward.png")
+        next_button = gtk.ToolButton (gtk.image_new_from_file (next_image))
         next_button.connect ("clicked", self.press_next_cb)
         next_button.set_expand (True)
         button_box.insert (next_button, -1)
@@ -263,8 +257,8 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         """
         assert len (song) == 6
         
-        self.filename_data.set_markup ("<small>" + song[FILE_URI] + "</small>")
-        self.title_entry.set_text (song[TITLE_KEY])
+        self.filename_data.set_markup ("<small>" + song[URI_COLUMN] + "</small>")
+        self.title_entry.set_text (song[TITLE_COLUMN])
         
 
         # Disconnect the value-change signal to avoid extra album art retrievals
@@ -276,17 +270,17 @@ class MussorgskyEditPanel (hildon.StackableWindow):
 
         # Set values in the picker buttons
         try:
-            self.artist_button.set_active (self.artists_list.index(song[ARTIST_KEY]))
+            self.artist_button.set_active (self.artists_list.index(song[ARTIST_COLUMN]))
         except ValueError:
-            print "'%s' not in artist list!?" % (song[ARTIST_KEY])
+            print "'%s' not in artist list!?" % (song[ARTIST_COLUMN])
             self.artist_button.set_value ("")
         except AttributeError:
             print "WARNING: Use set_artist_alternatives method to set a list of artists"
             
         try:
-            self.album_button.set_active (self.albums_list.index (song[ALBUM_KEY]))
+            self.album_button.set_active (self.albums_list.index (song[ALBUM_COLUMN]))
         except ValueError:
-            print "'%s' is not in the album list!?" % (song[ALBUM_KEY])
+            print "'%s' is not in the album list!?" % (song[ALBUM_COLUMN])
             self.album_button.set_value ("")
         except AttributeError:
             print "WARNING: Use set_album_alternatives method to set a list of artists"
@@ -299,9 +293,9 @@ class MussorgskyEditPanel (hildon.StackableWindow):
                                                                  self.artist_selection_cb)
 
         # Set the album art given the current data
-        self.set_album_art (song[ALBUM_KEY])
+        self.set_album_art (song[ALBUM_COLUMN])
 
-        if (not song[MIME_KEY] in self.writer.get_supported_mimes ()):
+        if (not song[MIME_COLUMN] in self.writer.get_supported_mimes ()):
             self.artist_button.set_sensitive (False)
             self.album_button.set_sensitive (False)
             self.title_entry.set_sensitive (False)
@@ -327,8 +321,8 @@ class MussorgskyEditPanel (hildon.StackableWindow):
         if (self.player.is_playing ()):
             self.player.stop ()
         else:
-            song = self.songs_list [self.song_counter]
-            self.player.play ("file://" + song[FILE_URI])
+            song = self.get_current_row ()
+            self.player.play ("file://" + song[URI_COLUMN])
 
     def album_selection_cb (self, widget):
         """
@@ -356,19 +350,21 @@ class MussorgskyEditPanel (hildon.StackableWindow):
 # Testing porpuses
 if __name__ == "__main__":
 
-    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 = [("/a/b/c/%d.mp3" %i, "Music",
+                  "Artist %d" % i, "Title %d" % i, "Album %d" % (i*100),
+                  "audio/mpeg",
+                  "artist %d album %d" % (i, i*100),
+                  "text to be searched artist %d album %d" % (i, i*100)) for i in range (0, 4)]
 
-    model = gtk.ListStore (str, str, str, str, str, str)
+    model = gtk.ListStore (str, str, str, str, str, str, str, str)
     for t in TEST_DATA:
+        print t
         model.append (t)
 
     window = MussorgskyEditPanel ()
-    window.set_artist_alternatives (["", "Bob Dylan"])
-    window.set_album_alternatives (["", "Bring it all back home", "Album 2", "Album 9", "Album 3"])
-    window.set_model (model)
+    window.set_artist_alternatives (["Artist %d" % i for i in range (0, 4)])
+    window.set_album_alternatives (["Album %d" % (i*100) for i in range (0, 4)])
+    window.set_model (model.filter_new ())
     window.connect ("destroy", gtk.main_quit)
     window.show_all ()
     gtk.main ()