#!/usr/bin/env python import os import os.path import sys import gtk import gconf import dbus import md5 import urllib SERVICES = [ ( 'Files', 'system-file-manager' ), ( 'Folders', 'folder' ), ( 'Applications', 'application-x-executable' ), ( 'Conversations', 'face-smile-big' ), ( 'Contacts', 'x-office-address-book' ), ( 'Development Files', 'applications-development' ), ( 'Documents', 'x-office-document' ), ( 'Emails', 'email' ), ( 'Images', 'image-x-generic' ), ( 'Music', 'audio-x-generic' ), ( 'Text Files', 'text-x-generic' ), ( 'Videos', 'video-x-generic' ), ] class Tracker: def __init__(self): bus = dbus.SessionBus() obj = bus.get_object('org.freedesktop.Tracker', '/org/freedesktop/tracker') self.tracker = dbus.Interface(obj, 'org.freedesktop.Tracker') self.files = dbus.Interface(obj, 'org.freedesktop.Tracker.Files') self.search = dbus.Interface(obj, 'org.freedesktop.Tracker.Search') self.keywords = dbus.Interface(obj, 'org.freedesktop.Tracker.Keywords') def __getattr__(self, attr): return getattr(self.tracker, attr) class TrackerGUI: def __init__(self): self.gconf = gconf.client_get_default() self.icon_theme = gtk.icon_theme_get_default() self.tracker = Tracker() try: self.layout = int(os.environ['TRACKER_UI_LAYOUT']) except: self.layout = 1 try: self.max_results = int(os.environ['TRACKER_MAX_RESULTS']) except: self.max_results = 20 self.window = gtk.Window() self.window.set_title('Tracker') main_vbox = gtk.VBox() #main_vbox.set_border_width(12) main_vbox.set_spacing(6) hbox = gtk.HBox() hbox.set_spacing(6) hbox.set_border_width(12) label = gtk.Label('Search:') hbox.pack_start(label, expand=False, fill=False) self.search_entry = gtk.Entry() hbox.pack_start(self.search_entry, expand=True, fill=True) if self.layout == 1: self.service_combobox = gtk.ComboBox() store = gtk.ListStore(gtk.gdk.Pixbuf, str) self.service_combobox.set_model(store) cell = gtk.CellRendererPixbuf() self.service_combobox.pack_start(cell, False) self.service_combobox.add_attribute(cell, 'pixbuf', 0) cell = gtk.CellRendererText() self.service_combobox.pack_start(cell, False) self.service_combobox.add_attribute(cell, 'text', 1) for label, icon_name in SERVICES: pixbuf = self.icon_theme.load_icon(icon_name, gtk.ICON_SIZE_MENU, gtk.ICON_LOOKUP_USE_BUILTIN) store.append([pixbuf, label]) self.service_combobox.set_active(0) hbox.pack_start(self.service_combobox, expand=False, fill=True) self.search_button = gtk.Button(stock='gtk-find') self.search_button.set_data('offset', 0) hbox.pack_start(self.search_button, expand=False, fill=False) main_vbox.pack_start(hbox, expand=False, fill=False) hbox = gtk.HBox() if self.layout == 2: self.service_treeview = gtk.TreeView() store = gtk.ListStore(gtk.gdk.Pixbuf, str) self.service_treeview.set_model(store) #self.service_treeview.set_sensitive(False) self.service_treeview.set_headers_visible(False) self.service_treeview.set_row_separator_func(lambda model,iter: model[iter][1] == '') cell = gtk.CellRendererPixbuf() col = gtk.TreeViewColumn('', cell, pixbuf=0) self.service_treeview.append_column(col) cell = gtk.CellRendererText() col = gtk.TreeViewColumn('', cell, text=1) self.service_treeview.append_column(col) for label, icon_name in SERVICES: pixbuf = self.icon_theme.load_icon(icon_name, gtk.ICON_SIZE_MENU, gtk.ICON_LOOKUP_USE_BUILTIN) store.append([pixbuf, label]) store.append([None,'']) tags = self.tracker.keywords.GetList('Files') for tag,count in tags.items(): store.append([None, tag]) selection = self.service_treeview.get_selection() selection.select_iter(store.get_iter_first()) selection.connect('changed', self.on_service_selection_changed) sw = gtk.ScrolledWindow() sw.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC) sw.set_shadow_type(gtk.SHADOW_IN) sw.add(self.service_treeview) hbox.pack_start(sw, expand=False, fill=True) sw = gtk.ScrolledWindow() sw.set_policy(gtk.POLICY_NEVER, gtk.POLICY_ALWAYS) sw.set_shadow_type(gtk.SHADOW_IN) if self.layout in (1,2,3,): self.icon_view = gtk.IconView() store = gtk.ListStore(gtk.gdk.Pixbuf, str, str) self.icon_view.set_model(store) self.icon_view.set_pixbuf_column(0) self.icon_view.set_text_column(1) sw.add(self.icon_view) elif self.layout == 4: self.results_tree = gtk.TreeView() store = gtk.TreeStore(gtk.gdk.Pixbuf, str, str) self.results_tree.set_model(store) self.results_tree.set_headers_visible(False) cell = gtk.CellRendererPixbuf() col = gtk.TreeViewColumn('', cell, pixbuf=0) self.results_tree.append_column(col) cell = gtk.CellRendererText() col = gtk.TreeViewColumn('', cell, text=1) self.results_tree.append_column(col) self._fill_in_services(store) sw.add(self.results_tree) sw.set_policy(gtk.POLICY_AUTOMATIC, gtk.POLICY_AUTOMATIC) if self.layout == 3: self.notebook = gtk.Notebook() label = gtk.Label('Files') self.notebook.append_page(sw, label) main_vbox.pack_start(self.notebook, expand=True, fill=True) else: hbox.pack_start(sw, expand=True, fill=True) main_vbox.pack_start(hbox, expand=True, fill=True) if self.layout != 4: hbox = gtk.HBox() hbox.set_spacing(6) self.next_button = gtk.Button(stock='gtk-go-forward') self.next_button.get_children()[0].get_children()[0].get_children()[1].set_text('Next') self.next_button.set_sensitive(False) hbox.pack_end(self.next_button, expand=False, fill=False) image = gtk.image_new_from_stock('gtk-go-back', gtk.ICON_SIZE_BUTTON) self.previous_button = gtk.Button() self.previous_button.add(image) self.previous_button.set_sensitive(False) hbox.pack_end(self.previous_button, expand=False, fill=False) main_vbox.pack_start(hbox, expand=False, fill=False) self.statusbar = gtk.Statusbar() main_vbox.pack_start(self.statusbar, expand=False, fill=False) self.window.add(main_vbox) if self.layout == 3: self.search_button.connect('clicked', self.build_service_tabs) if self.layout == 4: self.search_button.connect('clicked', self.on_complete_search) self.results_tree.get_selection().connect('changed', self.on_result_selected) else: self.icon_view.connect('selection-changed', self.on_result_selected) self.search_button.connect('clicked', self.on_search) self.next_button.connect('clicked', self.on_search) self.previous_button.connect('clicked', self.on_search) self.window.connect('destroy', gtk.main_quit) self.window.resize(640, 480) self.window.show_all() def on_search(self, button): query = self.search_entry.get_text() offset = button.get_data('offset') if self.layout == 1: service = SERVICES[self.service_combobox.get_active()][0] elif self.layout == 2: store, iter = self.service_treeview.get_selection().get_selected() service = store[iter][1] elif self.layout == 3: service = self.notebook.get_tab_label(self.icon_view.get_parent()).get_text() results = self.tracker.search.Text(-1, service, query, offset, self.max_results + 1, False) if self.layout in (1,2,3,): store = self.icon_view.get_model() store.clear() for r in results: basename = os.path.basename(r) pixbuf = self._get_file_thumbnail_pixbuf(r) or self._get_mime_thumbnail(r) store.append([ pixbuf, basename, r ]) self.icon_view.get_parent().get_vscrollbar().get_adjustment().value = 0 if offset > 0: self.previous_button.set_sensitive(True) self.previous_button.set_data('offset', offset - self.max_results) else: self.previous_button.set_sensitive(False) if len(results) == self.max_results + 1: self.next_button.set_sensitive(True) self.next_button.set_data('offset', offset + self.max_results) else: self.next_button.set_sensitive(False) def on_complete_search(self, button): query = self.search_entry.get_text() store = self.results_tree.get_model() store.clear() self._fill_in_services(store) iter = store.get_iter_first() while iter: service = store[iter][1] done = False offset = 0 while not done: results = self.tracker.search.Text(-1, service, query, offset, 256, False) offset += len(results) for r in results: pixbuf = self._get_file_thumbnail_pixbuf(r) or self._get_mime_thumbnail(r) h = pixbuf.get_height() w = pixbuf.get_width() scale = max(h,w) h = int(h * 24. / scale) w = int(w * 24. / scale) pixbuf = pixbuf.scale_simple(w, h, gtk.gdk.INTERP_BILINEAR) store.append(iter, [pixbuf, r, r]) done = (len(results) < 256) iter = store.iter_next(iter) def on_service_selection_changed(self, selection): self.on_search(self.search_button) def _get_file_thumbnail_pixbuf(self, f): uri = 'file://%s' % urllib.pathname2url(f) thumb = md5.new() thumb.update(uri) thumb = os.path.expanduser('~/.thumbnails/normal/%s.png' % thumb.hexdigest()) try: return gtk.gdk.pixbuf_new_from_file(thumb) except: return None def _get_mime_thumbnail(self, f): return self.icon_theme.load_icon('image-loading', 32, gtk.ICON_LOOKUP_USE_BUILTIN) def _fill_in_services(self, store): for label, icon_name in SERVICES: pixbuf = self.icon_theme.load_icon(icon_name, gtk.ICON_SIZE_MENU, gtk.ICON_LOOKUP_USE_BUILTIN) store.append(None, [pixbuf, label, '']) def build_service_tabs(self, button): query = self.search_entry.get_text() services = set() done = False offset = 0 while not done: results = self.tracker.search.Text(-1, 'Files', query, offset, 256, False) print len(results),'results.',results for r in results: print r, self.tracker.files.GetServiceType(r) #services.add(self.tracker.files.GetServiceType(r)) offset += len(results) done = (len(results) < 256) print 'done:',done print services def on_result_selected(self, arg): if isinstance(arg, gtk.IconView): store = arg.get_model() selected = arg.get_selected_items()[0] path = store[selected][2] else: model, iter = arg.get_selected() path = model[iter][2] print 'Selected file: [%s]' % path self.statusbar.push(0, path) gui = TrackerGUI() gtk.main()