git.maemo.org Git - mussorgsky/blob - src/edit_panel.py

   1 #!/usr/bin/env python2.5
   2 import hildon
   3 import gtk, gobject
   4 from mutagen_backend import MutagenBackend
   5 from player_backend import MediaPlayer
   6 import album_art_spec
   7 from album_art import MussorgskyAlbumArt
   8 import os
   9
  10 # Fields in the tuple!
  11 FILE_URI = 0
  12 ARTIST_KEY = 2
  13 TITLE_KEY = 3
  14 ALBUM_KEY = 4
  15 MIME_KEY = 5
  16
  17 class MussorgskyEditPanel (hildon.StackableWindow):
  18
  19     def __init__ (self, songs_list=None, albums_list=None, artists_list=None):
  20         hildon.StackableWindow.__init__ (self)
  21         self.set_title ("Edit")
  22         self.set_border_width (12)
  23         self.writer = MutagenBackend ()
  24         self.player = MediaPlayer ()
  25         self.album_art_retriever = MussorgskyAlbumArt ()
  26         self.albums_list = albums_list
  27         self.artists_list = artists_list
  28         self.add (self.__create_view ())
  29         if (songs_list):
  30             self.set_songs_list (songs_list)
  31         self.banner = None
  32
  33         self.artists_selector = None
  34         self.artists_dialog = None
  35
  36         self.albums_selector = None
  37         self.albums_dialog = None
  38         self.album_callback_id = -1
  39
  40
  41     def set_songs_list (self, songs_list):
  42         if (songs_list and len (songs_list) > 0):
  43             self.songs_list = songs_list
  44             self.set_data_in_view (songs_list [0])
  45             self.song_counter = 0
  46
  47     def press_back_cb (self, widget):
  48         if (self.player.is_playing ()):
  49             self.player.stop ()
  50
  51         if (self.banner and self.banner.get_property("visible")):
  52             self.banner.destroy ()
  53
  54         if (self.album_callback_id != -1):
  55             gobject.source_remove (self.album_callback_id)
  56             self.album_callback_id = -1
  57
  58         if self.__is_view_dirty ():
  59             print "Modified data. Save!"
  60             self.save_metadata ()
  61
  62         if (self.song_counter > 0):
  63             self.song_counter -= 1
  64             self.set_data_in_view (self.songs_list [self.song_counter])
  65
  66     def press_next_cb (self, widget):
  67         if (self.player.is_playing ()):
  68             self.player.stop ()
  69
  70         if (self.banner and self.banner.get_property("visible")):
  71             self.banner.destroy ()
  72
  73         if (self.album_callback_id != -1):
  74             gobject.source_remove (self.album_callback_id)
  75             self.album_callback_id = -1
  76
  77         if self.__is_view_dirty ():
  78             print "Modified data. Save!"
  79             self.save_metadata ()
  80
  81         if (self.song_counter < len (self.songs_list) -1):
  82             self.song_counter += 1
  83             self.set_data_in_view (self.songs_list [self.song_counter])
  84         else:
  85             self.destroy ()
  86
  87     def save_metadata (self):
  88         # Save the data in the online model to show the appropiate data
  89         # in the UI while tracker process the update.
  90         song = self.songs_list [self.song_counter]
  91
  92         new_song = (song[FILE_URI], song[1],
  93                     self.artist_entry.get_text (),
  94                     self.title_entry.get_text (),
  95                     self.album_entry.get_text (),
  96                     song[MIME_KEY])
  97         self.songs_list [self.song_counter] = new_song
  98         try:
  99             self.writer.save_metadata_on_file (new_song[FILE_URI],
 100                                                new_song[MIME_KEY],
 101                                                self.artist_entry.get_text (),
 102                                                self.title_entry.get_text (),
 103                                                self.album_entry.get_text ())
 104         except IOError, e:
 105             # This error in case of tracker returning unexistent files.
 106             # Uhm.... for instance after removing a memory card we are editing!
 107             dialog = gtk.MessageDialog (self,
 108                                         gtk.DIALOG_DESTROY_WITH_PARENT,
 109                                         gtk.MESSAGE_ERROR,
 110                                         gtk.BUTTONS_CLOSE,
 111                                         "%s" % str(e));
 112             dialog.run ()
 113
 114
 115
 116     def __is_view_dirty (self):
 117         """
 118         True if the data has been modified in the widgets
 119         """
 120         song = self.songs_list [self.song_counter]
 121
 122         return not (self.filename_data.get_text() == song[FILE_URI] and
 123                     self.artist_entry.get_text () == song[ARTIST_KEY] and
 124                     self.title_entry.get_text () == song[TITLE_KEY] and
 125                     self.album_entry.get_text () == song[ALBUM_KEY] )
 126
 127
 128     def __create_view (self):
 129         view_vbox = gtk.VBox (homogeneous=False, spacing = 12)
 130
 131         filename_row = gtk.HBox ()
 132         filename_label = gtk.Label ("Filename:")
 133         filename_row.pack_start (filename_label, expand=False, padding=12);
 134         self.filename_data = gtk.Label ("")
 135         filename_row.pack_start (self.filename_data, expand=True)
 136
 137         #play_button = gtk.Button (stock=gtk.STOCK_MEDIA_PLAY)
 138         play_button = hildon.Button (hildon.BUTTON_STYLE_NORMAL, hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
 139         img = gtk.image_new_from_stock (gtk.STOCK_MEDIA_PLAY, gtk.ICON_SIZE_BUTTON)
 140         play_button.set_image (img)
 141         play_button.connect ("clicked", self.clicked_play)
 142         filename_row.pack_start (play_button, expand=False, fill=False)
 143         view_vbox.pack_start (filename_row, expand=False);
 144
 145         central_panel = gtk.HBox (spacing=12)
 146
 147         table = gtk.Table (3, 2, False)
 148         table.set_col_spacings (12)
 149         table.set_row_spacings (12)
 150
 151         central_panel.pack_start (table, fill=True)
 152         view_vbox.pack_start (central_panel, expand=True, fill=True)
 153
 154         # Artist row
 155         button_artist = gtk.Button ("Artist:")
 156         if (not self.artists_list):
 157             button_artist.set_sensitive (False)
 158         button_artist.connect ("clicked", self.artist_selection_cb)
 159         table.attach (button_artist, 0, 1, 0, 1, 0, gtk.FILL|gtk.EXPAND)
 160         self.artist_entry = gtk.Entry()
 161         table.attach (self.artist_entry, 1, 2, 0, 1)
 162
 163         # Title row
 164         label_title = gtk.Label ("Title:")
 165         table.attach (label_title, 0, 1, 1, 2, 0)
 166         self.title_entry = gtk.Entry()
 167         table.attach (self.title_entry, 1, 2, 1, 2)
 168
 169         # Album row
 170         button_album = gtk.Button ("Album:")
 171         if (not self.albums_list):
 172             button_album.set_sensitive (False)
 173         button_album.connect ("clicked", self.album_selection_cb)
 174         table.attach (button_album, 0, 1, 2, 3, 0)
 175         self.album_entry = gtk.Entry()
 176         table.attach (self.album_entry, 1, 2, 2, 3)
 177
 178         # Album art space
 179         album_button = gtk.Button ()
 180         self.album_art = gtk.Image ()
 181         self.album_art.set_size_request (124, 124)
 182         album_button.add (self.album_art)
 183         album_button.connect ("clicked", self.clicked_album_art)
 184         central_panel.pack_start (album_button, expand=False, fill=False)
 185
 186         # Buttons row
 187         button_box = gtk.HButtonBox ()
 188         button_box.set_layout (gtk.BUTTONBOX_END)
 189
 190         back_button = hildon.Button (hildon.BUTTON_STYLE_NORMAL, hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
 191         img = gtk.image_new_from_stock (gtk.STOCK_GO_BACK, gtk.ICON_SIZE_BUTTON)
 192         back_button.set_image (img)
 193         back_button.connect ("clicked", self.press_back_cb)
 194         button_box.pack_start (back_button, expand=True, fill=True, padding=6)
 195
 196         next_button = hildon.Button (hildon.BUTTON_STYLE_NORMAL, hildon.BUTTON_ARRANGEMENT_HORIZONTAL)
 197         img = gtk.image_new_from_stock (gtk.STOCK_GO_FORWARD, gtk.ICON_SIZE_BUTTON)
 198         next_button.set_image (img)
 199         next_button.connect ("clicked", self.press_next_cb)
 200         button_box.pack_start (next_button, expand=True, fill=True, padding=6)
 201
 202         view_vbox.pack_start (button_box, expand=False, fill=True, padding=6)
 203
 204         return view_vbox
 205
 206
 207     def set_data_in_view (self, song):
 208         """
 209         Place in the screen the song information.
 210         Song is a tuple like (filename, 'Music', title, artist, album, mime)
 211         """
 212         assert len (song) == 6
 213         self.filename_data.set_text (song[FILE_URI])
 214         self.artist_entry.set_text (song[ARTIST_KEY])
 215         self.title_entry.set_text (song[TITLE_KEY])
 216         self.album_entry.set_text (song[ALBUM_KEY])
 217
 218         has_album = False
 219         if (song[ALBUM_KEY]):
 220             thumb = album_art_spec.getCoverArtThumbFileName (song[ALBUM_KEY])
 221             print "%s -> %s" % (song[ALBUM_KEY], thumb)
 222             if (os.path.exists (thumb)):
 223                 self.album_art.set_from_file (thumb)
 224                 has_album = True
 225             else:
 226                 self.album_callback_id = gobject.idle_add (self.retrieve_album_art, song[ARTIST_KEY], song[ALBUM_KEY])
 227
 228         if (not has_album):
 229             self.album_art.set_from_stock (gtk.STOCK_CDROM, gtk.ICON_SIZE_DIALOG)
 230
 231         if (not song[MIME_KEY] in self.writer.get_supported_mimes ()):
 232             print "show notification"
 233             self.banner = hildon.Banner ()
 234             self.banner.set_text ("Unsupported format (%s)" % song[MIME_KEY])
 235             self.banner.show_all ()
 236
 237     def retrieve_album_art (self, artist, album):
 238         print "trying to get the album art"
 239         (img, thumb) = self.album_art_retriever.get_album_art (artist, album)
 240         if (thumb):
 241             self.album_art.set_from_file (thumb)
 242         else:
 243             print "Unable to retrieve album art"
 244
 245         return False
 246
 247     def clicked_play (self, widget):
 248         if (self.player.is_playing ()):
 249             self.player.stop ()
 250         else:
 251             song = self.songs_list [self.song_counter]
 252             self.player.play ("file://" + song[FILE_URI])
 253
 254     def clicked_album_art (self, widget):
 255         print "implement me, please"
 256
 257     def album_selection_cb (self, widget):
 258         if (not self.albums_selector):
 259             self.albums_selector = hildon.hildon_touch_selector_new_text ()
 260             for album in self.albums_list :
 261                 self.albums_selector.append_text (album[0])
 262
 263         if (not self.albums_dialog):
 264             self.albums_dialog = hildon.PickerDialog (self)
 265             self.albums_dialog.set_title ("Choose album...")
 266             self.albums_dialog.set_selector (self.albums_selector)
 267
 268         response = self.albums_dialog.run ()
 269         if (response == gtk.RESPONSE_OK):
 270             print "Ok (%s)" % (self.albums_selector.get_current_text ())
 271             self.album_entry.set_text (self.albums_selector.get_current_text ())
 272         self.albums_dialog.hide ()
 273
 274     def artist_selection_cb (self, widget):
 275         if (not self.artists_selector):
 276             self.artists_selector = hildon.hildon_touch_selector_new_text ()
 277             for artist in self.artists_list :
 278                 self.artists_selector.append_text (artist[0])
 279
 280         if (not self.artists_dialog):
 281             self.artists_dialog = hildon.PickerDialog (self)
 282             self.artists_dialog.set_title ("Choose artist...")
 283             self.artists_dialog.set_selector (self.artists_selector)
 284
 285         response = self.artists_dialog.run ()
 286
 287         if (response == gtk.RESPONSE_OK):
 288             print "Ok (%s)" % (self.artists_selector.get_current_text ())
 289             self.artist_entry.set_text (str(self.artists_selector.get_current_text ()))
 290         self.artists_dialog.hide ()
 291
 292 # Testing porpuses
 293 if __name__ == "__main__":
 294
 295     TEST_DATA = [("/a/b/c/d.mp3", "Music", "", "title", "album", "audio/mpeg"),
 296                  ("/home/user/mufix/dejame.mp3", "Music", "", "title", "album 2", "a/b"),
 297                  ("/media/mmc1/Attachments/1-14 Una Rosa Es Una Rosa.m4a", "Music", "", "Una rosa es una rosa", "Ana, Jose, Nacho", "audio/mpeg"),
 298                  ("/home/user/mufix/3.mp2", "Music", "", "titlex", "album 3", "audio/mpeg")]
 299     ALBUMS = [["Album %d" % i] for i in range (0, 10)]
 300     ARTISTS = [["Artist %d" % i] for i in range (0, 10)]
 301     window = MussorgskyEditPanel (TEST_DATA, ALBUMS, ARTISTS)
 302     window.connect ("destroy", gtk.main_quit)
 303     window.show_all ()
 304     gtk.main ()