Search only after 3 or more characters
[mussorgsky] / src / browse_panel.py
1 #!/usr/bin/env python2.5
2 import hildon
3 import gtk, gobject
4 from edit_panel_tm import MussorgskyEditPanel
5 from utils import escape_html, Set
6
7 class MussorgskyBrowsePanel (hildon.StackableWindow):
8
9     def __init__ (self, songs_list):
10         hildon.StackableWindow.__init__ (self)
11         self.set_title ("Browse collection")
12         self.set_border_width (12)
13         self.__create_view ()
14
15         # Prepare cache of artists and albums
16         self.artist_set = Set ()
17         self.albums_set = Set ()
18
19         # (uri, "Music", artist, title, album, mimetype) + "string" + search_string
20         self.full_model = gtk.ListStore (str, str, str, str, str, str, str, str)
21         for (uri, category, artist, title, album, mime) in songs_list:
22             text = "<b>%s</b>\n<small>%s</small>" % (escape_html (title),
23                                                      escape_html (artist) + " / " + escape_html (album))
24             search_str = artist.lower () + " " + title.lower () + " " + album.lower ()
25             self.full_model.append ((uri, category, artist, title, album, mime, text, search_str))
26             self.artist_set.insert (artist)
27             self.albums_set.insert (album)
28             
29         self.filtered_model = self.full_model.filter_new ()
30         self.treeview.set_model (self.filtered_model)
31         self.kpid = self.connect ("key-press-event", self.key_pressed_cb)
32
33     def __create_view (self):
34         vbox = gtk.VBox (homogeneous=False)
35         
36         self.treeview = gtk.TreeView ()
37         self.treeview.connect ("row-activated", self.row_activated_cb)
38
39         desc_column = gtk.TreeViewColumn ("Song", gtk.CellRendererText (), markup=6)
40         desc_column.set_expand (True)
41         self.treeview.append_column (desc_column)
42
43         pannable_area = hildon.PannableArea ()
44         pannable_area.add (self.treeview)
45         
46         vbox.pack_start (pannable_area, expand=True)
47         
48         self.search_hbox = gtk.HBox ()
49         self.search_entry = hildon.Entry (gtk.HILDON_SIZE_FINGER_HEIGHT)
50         self.search_hbox.pack_start (self.search_entry, expand=True)
51         
52         self.search_close = gtk.Button (stock=gtk.STOCK_CLOSE)
53         self.search_hbox.pack_start (self.search_close, expand=False)
54         self.search_close.connect ("clicked", self.close_search_cb)
55         
56         # Hide it when the window is created
57         self.search_box_visible = False
58         self.search_hbox.set_no_show_all (True)
59         self.search_hbox.hide ()
60         vbox.pack_start (self.search_hbox, expand=False)
61         self.add (vbox)
62
63     def search_type (self, widget):
64         if len (widget.get_text()) == 0:
65             self.treeview.set_model (self.full_model)
66             return
67             
68         if (len (widget.get_text ()) < 3):
69             return
70         self.filtered_model.set_visible_func (self.entry_equals, widget)
71         self.filtered_model.refilter ()
72         self.treeview.set_model (self.filtered_model)
73
74     def close_search_cb (self, widget):
75         assert not self.search_box_visible
76         self.search_hbox.hide_all ()
77         self.search_entry.set_text ("")
78         self.search_box_visible = False
79         self.kpid = self.connect ("key-press-event", self.key_pressed_cb)
80
81     def key_pressed_cb (self, widget, event):
82         if (event.type == gtk.gdk.KEY_PRESS):
83             if (event.keyval == gtk.gdk.keyval_from_name ("Alt_L")):
84                 return
85             
86             if (not self.search_box_visible ):
87                 self.search_hbox.set_no_show_all (False)
88                 self.search_hbox.show_all ()
89                 
90             self.search_entry.grab_focus ()
91             self.search_entry.connect ("changed", self.search_type)
92             self.disconnect (self.kpid)
93     
94     def entry_equals (self, model, it, user_data):
95         t = user_data.get_text ()
96         return t.lower () in model.get_value (it, 7)
97
98     def row_activated_cb (self, treeview, path, view_colum):
99         edit_view = MussorgskyEditPanel ()
100         edit_view.set_artist_alternatives (self.artist_set.as_list ())
101         edit_view.set_album_alternatives (self.albums_set.as_list ())
102         edit_view.set_model (self.treeview.get_model (), self.treeview.get_model ().get_iter (path))
103         edit_view.show_all ()
104
105
106 if __name__ == "__main__":
107
108     import random
109     def get_random_path ():
110         path = "file://"
111         for i in range (0, random.randint (1, 8)):
112             path = path + "/" + ("x"* random.randint (4, 12))
113         return path
114             
115
116     songs = [(get_random_path (),
117               "Music",
118               "Artist%d" % i,
119               "Title <%d>" % i,
120               "album <%d>" % i,
121               "audio/mpeg") for i in range (0, 100)]
122
123     window = MussorgskyBrowsePanel (songs)
124     window.connect ("destroy", gtk.main_quit )
125     window.show_all ()
126     gtk.main ()