git.maemo.org Git - jamaendo/blob - jamaui/search.py

   1 import gtk
   2 import hildon
   3 import jamaendo
   4 from playerwindow import open_playerwindow
   5 from showartist import ShowArtist
   6 from showalbum import ShowAlbum
   7
   8 class SearchWindow(hildon.StackableWindow):
   9     def __init__(self):
  10         hildon.StackableWindow.__init__(self)
  11         self.set_title("Search")
  12
  13         vbox = gtk.VBox(False, 0)
  14
  15
  16         # Results list
  17         self.panarea = hildon.PannableArea()
  18         self.result_store = gtk.ListStore(str, int)
  19         #self.result_store.append(['red'])
  20         self.result_view = gtk.TreeView(self.result_store)
  21         col = gtk.TreeViewColumn('Name')
  22         self.result_view.append_column(col)
  23         cell = gtk.CellRendererText()
  24         col.pack_start(cell, True)
  25         col.add_attribute(cell, 'text', 0)
  26         self.result_view.set_search_column(0)
  27         col.set_sort_column_id(0)
  28         self.result_view.connect('row-activated', self.row_activated)
  29
  30         self.panarea.add(self.result_view)
  31         vbox.pack_start(self.panarea, True, True, 0)
  32
  33
  34         # Create selector for search mode
  35         self.mode_selector = hildon.TouchSelector(text=True)
  36
  37         self.mode_selector.append_text("Artists")
  38         self.mode_selector.append_text("Albums")
  39         self.mode_selector.append_text("Tracks")
  40         self.mode = hildon.PickerButton(gtk.HILDON_SIZE_FINGER_HEIGHT,
  41                                         hildon.BUTTON_ARRANGEMENT_VERTICAL)
  42         self.mode.set_title("Search for")
  43         self.mode.set_selector(self.mode_selector)
  44         self.mode_selector.connect("changed", self.mode_changed)
  45         #vbox.pack_start(self.mode, False)
  46         self.mode.set_active(1)
  47
  48
  49         # Search box
  50         hbox = gtk.HBox(False, 0)
  51         self.entry = hildon.Entry(gtk.HILDON_SIZE_FINGER_HEIGHT)
  52         self.entry.set_placeholder("Search")
  53         self.entry.connect('activate', self.on_search)
  54         btn = hildon.GtkButton(gtk.HILDON_SIZE_FINGER_HEIGHT)
  55         btn.set_label(">>")
  56         btn.connect('clicked', self.on_search)
  57         hbox.pack_start(self.mode, False)
  58         hbox.pack_start(self.entry, True, True, 0)
  59         hbox.pack_start(btn, False)
  60         vbox.pack_start(hbox, False)
  61
  62
  63         self.add(vbox)
  64
  65         self.idmap = {}
  66
  67     def mode_changed(self, selector, user_data):
  68         current_selection = selector.get_current_text()
  69         print current_selection
  70
  71     def on_search(self, w):
  72         mode = self.mode.get_active()
  73         txt = self.entry.get_text()
  74         self.result_store.clear()
  75         if mode == 0:
  76             for artist in jamaendo.search_artists(query=txt):
  77                 title = artist.name
  78                 self.idmap[artist.ID] = artist
  79                 self.result_store.append([title, artist.ID])
  80         elif mode == 1:
  81             for album in jamaendo.search_albums(query=txt):
  82                 title = "%s - %s" % (album.artist_name, album.name)
  83                 self.idmap[album.ID] = album
  84                 self.result_store.append([title, album.ID])
  85         elif mode == 2:
  86             for track in jamaendo.search_tracks(query=txt):
  87                 title = "%s - %s" % (track.artist_name, track.name)
  88                 self.idmap[track.ID] = track
  89                 self.result_store.append([title, track.ID])
  90
  91     def row_activated(self, treeview, path, view_column):
  92         treeiter = self.result_store.get_iter(path)
  93         title, _id = self.result_store.get(treeiter, 0, 1)
  94         item = self.idmap[_id]
  95         print _id, item
  96         self.open_item(item)
  97
  98     def open_item(self, item):
  99         if isinstance(item, jamaendo.Album):
 100             wnd = ShowAlbum(item)
 101             wnd.show_all()
 102         elif isinstance(item, jamaendo.Artist):
 103             wnd = ShowArtist(item)
 104             wnd.show_all()
 105         elif isinstance(item, jamaendo.Track):
 106             wnd = open_playerwindow()
 107             wnd.play_tracks([item])