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