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