mainly UI work
[jamaendo] / jamaui / ui.py
1 # debugging hack - add . to path
2 import os, sys
3 local_module_dir = os.path.join(os.path.dirname(sys.argv[0]), '..')
4 if os.path.isdir(local_module_dir):
5     sys.path.append(local_module_dir)
6
7 import gtk
8 import gobject
9 import util
10 import logging
11 import gobject
12
13 from jamaendo.api import LocalDB, Query, Queries, refresh_dump
14 from jamaui.player import Player, Playlist
15
16 gobject.threads_init()
17
18 log = logging.getLogger(__name__)
19
20 try:
21     import hildon
22 except:
23     if util.platform == 'maemo':
24         log.critical('Using GTK widgets, install "python2.5-hildon" '
25             'for this to work properly.')
26     else:
27         log.critical('This ui only works in maemo')
28         sys.exit(1)
29
30 class RefreshDialog(object):
31     def __init__(self):
32         self.notebook = gtk.Notebook()
33         info = gtk.VBox()
34         info.pack_start(gtk.Label("Downloading complete DB from jamendo.com."), True, False)
35         info.pack_start(gtk.Label("This will download approximately 8 MB."), True, False)
36         self.force = hildon.GtkToggleButton(gtk.HILDON_SIZE_FINGER_HEIGHT)
37         self.force.set_label("Force refresh")
38
39         info.pack_start(self.force, True, False)
40         self.notebook.append_page(info)
41
42         pcont = gtk.VBox()
43         self.progress = gtk.ProgressBar()
44         pcont.pack_start(self.progress, True, False)
45         self.notebook.append_page(pcont,
46                                   gtk.Label("Updating Database"))
47         self.progress.set_fraction(0)
48         self.progress.set_orientation(gtk.PROGRESS_LEFT_TO_RIGHT)
49         self.progress.set_text("Downloading...")
50
51         self.notebook.append_page(gtk.Label("Database refreshed."))
52
53         self.dialog = hildon.WizardDialog(None, "Refresh", self.notebook)
54         self.notebook.connect("switch-page", self.on_switch)
55         self.dialog.set_forward_page_func(self.forward_func)
56
57         self.refresher = None
58
59     def on_complete(self, status):
60         hildon.hildon_gtk_window_set_progress_indicator(self.dialog, 0)
61         if status:
62             self.progress.set_fraction(1)
63             self.progress.set_text("DB up to date.")
64         else:
65             self.progress.set_fraction(0)
66             self.progress.set_text("Download failed.")
67
68     def on_progress(self, percent):
69         if percent < 100:
70             self.progress.set_text("Downloading...")
71         self.progress.set_fraction(percent/100.0)
72
73     def on_switch(self, notebook, page, num):
74         if num == 1:
75             hildon.hildon_gtk_window_set_progress_indicator(self.dialog, 1)
76             refresh_dump(self.on_complete, self.on_progress, force=self.force.get_active())
77         elif self.refresher:
78             # cancel download
79             pass
80         return True
81
82     def forward_func(self, notebook, current, userdata):
83         #page = notebook.get_nth_page(current)
84         if current == 0:
85             return True
86         else:
87             return False
88
89     def show_all(self):
90         self.dialog.show_all()
91
92     def run(self):
93         self.dialog.run()
94
95     def hide(self):
96         self.dialog.hide()
97
98 class PlayerWindow(hildon.StackableWindow):
99     def __init__(self, playlist=None):
100         hildon.StackableWindow.__init__(self)
101         self.set_title("jamaendo")
102
103         self.playlist = Playlist(playlist)
104         self.player = Player()
105         #self.player.play(playlist)
106
107         vbox = gtk.VBox()
108
109         hbox = gtk.HBox()
110
111         cover = gtk.Image()
112
113         vbox2 = gtk.VBox()
114
115         playlist_pos = gtk.Label("0/0 songs")
116         track = gtk.Label("Track name")
117         progress = hildon.GtkHScale()
118         artist = gtk.Label("Artist")
119         album = gtk.Label("Album")
120
121         vbox2.pack_start(playlist_pos, False)
122         vbox2.pack_start(track, False)
123         vbox2.pack_start(progress, True, True)
124         vbox2.pack_start(artist, False)
125         vbox2.pack_start(album, False)
126
127         hbox.pack_start(cover, True, True, 0)
128         hbox.pack_start(vbox2, True, True, 0)
129
130         vbox.pack_start(hbox, True, True, 0)
131
132         btns = gtk.HButtonBox()
133         btns.set_property('layout-style', gtk.BUTTONBOX_SPREAD)
134
135         vbox.pack_end(btns, False, True, 0)
136
137         self.add_stock_button(btns, gtk.STOCK_MEDIA_PREVIOUS, self.on_prev)
138         self.add_stock_button(btns, gtk.STOCK_MEDIA_PLAY, self.on_play)
139         self.add_stock_button(btns, gtk.STOCK_MEDIA_PAUSE, self.on_pause)
140         self.add_stock_button(btns, gtk.STOCK_MEDIA_STOP, self.on_stop)
141         self.add_stock_button(btns, gtk.STOCK_MEDIA_NEXT, self.on_next)
142
143         self.add(vbox)
144
145     def add_stock_button(self, btns, stock, cb):
146         btn = hildon.GtkButton(gtk.HILDON_SIZE_FINGER_HEIGHT)
147         btn.set_relief(gtk.RELIEF_NONE)
148         btn.set_image(gtk.image_new_from_stock(stock, gtk.ICON_SIZE_SMALL_TOOLBAR))
149         btn.connect('clicked', cb)
150         btns.add(btn)
151
152     def on_play(self, button):
153         self.player.play(self.playlist)
154     def on_pause(self, button):
155         self.player.pause()
156     def on_prev(self, button):
157         self.player.prev()
158     def on_next(self, button):
159         self.player.next()
160     def on_stop(self, button):
161         self.player.stop()
162
163 class SearchWindow(hildon.StackableWindow):
164     def __init__(self):
165         hildon.StackableWindow.__init__(self)
166         self.set_title("Search")
167
168         vbox = gtk.VBox(False, 0)
169
170         hbox = gtk.HBox()
171         self.entry = hildon.Entry(gtk.HILDON_SIZE_FINGER_HEIGHT)
172         self.entry.set_placeholder("Search")
173         self.entry.connect('activate', self.on_search)
174         btn = hildon.GtkButton(gtk.HILDON_SIZE_FINGER_HEIGHT)
175         btn.set_label("Go")
176         btn.connect('clicked', self.on_search)
177         hbox.pack_start(self.entry, True, True, 0)
178         hbox.pack_start(btn, False)
179
180         self.results = hildon.TouchSelector(text=True)
181         self.results.connect("changed", self.selection_changed)
182
183         vbox.pack_start(hbox, False)
184         vbox.pack_start(self.results, True, True, 0)
185
186         self.add(vbox)
187
188         self.idmap = {}
189
190         self.pwnd = None
191
192     def on_search(self, w):
193         txt = self.entry.get_text()
194         print "Search for: %s" % (txt)
195         db = LocalDB()
196         db.connect()
197         for album in db.search_albums(txt):
198             title = "%s - %s" % (album['artist'], album['name'])
199             self.idmap[title] = album
200             print "Found %s" % (album)
201             self.results.append_text(title)
202
203     def selection_changed(self, results, userdata):
204         current_selection = results.get_current_text()
205
206         album = self.idmap[current_selection]
207         selected = [album['id']]
208         print "Selected: %s" % (selected)
209         album = None
210         db = LocalDB()
211         db.connect()
212         for a in db.get_albums(selected):
213             album = a
214             break
215
216         if album:
217             tracks = album['tracks']
218             print "Playing: %s" % (tracks)
219             self.pwnd = PlayerWindow(tracks)
220             self.pwnd.show_all()
221
222 class PlaylistsWindow(hildon.StackableWindow):
223     def __init__(self):
224         hildon.StackableWindow.__init__(self)
225         self.set_title("Playlists")
226
227         label = gtk.Label("playlists")
228         vbox = gtk.VBox(False, 0)
229         vbox.pack_start(label, True, True, 0)
230         self.add(vbox)
231
232 class FavoritesWindow(hildon.StackableWindow):
233     def __init__(self):
234         hildon.StackableWindow.__init__(self)
235         self.set_title("Favorites")
236
237         label = gtk.Label("favorites")
238         vbox = gtk.VBox(False, 0)
239         vbox.pack_start(label, True, True, 0)
240         self.add(vbox)
241
242 class Jamaui(object):
243     _BG = 'bg.png' # /opt/jamaendo/data/bg.png
244
245     def __init__(self):
246         self.app = None
247         self.menu = None
248         self.window = None
249
250     def create_window(self):
251         self.app = hildon.Program()
252         self.window = hildon.StackableWindow()
253         self.app.add_window(self.window)
254
255         self.window.set_title("jamaendo")
256         self.window.connect("destroy", self.destroy)
257
258     def create_menu(self):
259         self.menu = hildon.AppMenu()
260
261         #search = hildon.GtkButton(gtk.HILDON_SIZE_AUTO)
262         #search.set_label("Search")
263         #search.connect("clicked", self.on_search)
264         #self.menu.append(search)
265
266         player = hildon.GtkButton(gtk.HILDON_SIZE_AUTO)
267         player.set_label("Open player")
268         player.connect("clicked", self.on_player)
269         self.menu.append(player)
270
271         refresh = hildon.GtkButton(gtk.HILDON_SIZE_AUTO)
272         refresh.set_label("Refresh")
273         refresh.connect("clicked", self.on_refresh)
274         self.menu.append(refresh)
275
276         menu_about = hildon.GtkButton(gtk.HILDON_SIZE_AUTO)
277         menu_about.set_label("About")
278         menu_about.connect("clicked", self.show_about, self.window)
279         self.menu.append(menu_about)
280         gtk.about_dialog_set_url_hook(self.open_link, None)
281
282         self.menu.show_all()
283         self.window.set_app_menu(self.menu)
284
285     def find_resource(self, name):
286         if os.path.isfile(os.path.join('data', name)):
287             return os.path.join('data', name)
288         elif os.path.isfile(os.path.join('/opt/jaemendo', name)):
289             return os.path.join('/opt/jaemendo', name)
290         elif os.path.isfile(os.path.join('/usr/share/jaemendo', name)):
291             return os.path.join('/usr/share/jaemendo', name)
292         else:
293             return name
294
295     def setup_widgets(self):
296         background, mask = gtk.gdk.pixbuf_new_from_file(self.find_resource(self._BG)).render_pixmap_and_mask()
297         self.window.realize()
298         self.window.window.set_back_pixmap(background, False)
299
300         bbox = gtk.HButtonBox()
301         alignment = gtk.Alignment(xalign=0.0, yalign=0.8, xscale=1.0)
302         alignment.add(bbox)
303         bbox.set_property('layout-style', gtk.BUTTONBOX_SPREAD)
304         self.bbox = bbox
305         self.window.add(alignment)
306
307         self.add_mainscreen_button("Search", "Search for artists/albums", self.on_search)
308         self.add_mainscreen_button("Playlists", "Browse playlists", self.on_playlists)
309         self.add_mainscreen_button("Favorites", "Your favorite albums", self.on_favorites)
310
311         self.window.show_all()
312
313     def add_mainscreen_button(self, title, subtitle, callback):
314         btn = hildon.Button(gtk.HILDON_SIZE_THUMB_HEIGHT,
315                             hildon.BUTTON_ARRANGEMENT_VERTICAL)
316         btn.set_text(title, subtitle)
317         btn.set_property('width-request', 225)
318         btn.connect('clicked', callback)
319         self.bbox.add(btn)
320
321     def destroy(self, widget):
322         gtk.main_quit()
323
324     def show_about(self, w, win):
325         dialog = gtk.AboutDialog()
326         dialog.set_website("http://github.com/krig")
327         dialog.set_website_label("http://github.com/krig")
328         dialog.set_name("Jamaendo")
329         dialog.set_authors(("Kristoffer Gronlund (Purple Scout AB)",))
330         dialog.set_comments("Media player for jamendo.com")
331         dialog.set_version('')
332         dialog.run()
333         dialog.destroy()
334
335     def open_link(self, d, url, data):
336         import webbrowser
337         webbrowser.open_new(url)
338
339
340     def on_refresh(self, button):
341         dialog = RefreshDialog()
342         dialog.show_all()
343         dialog.run()
344         dialog.hide()
345
346     def on_search(self, button):
347         self.searchwnd = SearchWindow()
348         self.searchwnd.show_all()
349
350     def on_playlists(self, button):
351         self.playlistswnd = PlaylistsWindow()
352         self.playlistswnd.show_all()
353
354     def on_favorites(self, button):
355         self.favoriteswnd = FavoritesWindow()
356         self.favoriteswnd.show_all()
357
358     def on_player(self, button):
359         self.playerwnd = PlayerWindow()
360         self.playerwnd.show_all()
361
362     '''
363     def on_search(self, button):
364         if self.searchbar:
365             self.searchbar.show()
366         else:
367             self.searchstore = gtk.ListStore(gobject.TYPE_STRING)
368             iter = self.searchstore.append()
369             self.searchstore.set(iter, 0, "Test1")
370             iter = self.searchstore.append()
371             self.searchstore.set(iter, 0, "Test2")
372             self.searchbar = hildon.FindToolbar("Search", self.searchstore, 0)
373             self.searchbar.set_active(0)
374             self.window.add_toolbar(self.searchbar)
375             self.searchbar.show()
376     '''
377
378     def run(self):
379         self.create_window()
380         self.create_menu()
381         self.setup_widgets()
382         self.window.show_all()
383         gtk.main()
384
385 if __name__=="__main__":
386     ui = Jamaui()
387     ui.run()
388