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

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