I got tired of writing C, so I wrote some prototype GUIs in Python.
So here's 4, all rolled into 1 script.  To switch between the four,
you need to set the TRACKER_UI_LAYOUT environment variable, so

TRACKER_UI_LAYOUT=1 ./tracker-gui-prototypes.py
Uses a GtkComboBox to select services; results in IconView

TRACKER_UI_LAYOUT=2 ./tracker-gui-prototypes.py
Uses a GtkTreeView + ListStore to select services; results in IconView

TRACKER_UI_LAYOUT=3 ./tracker-gui-prototypes.py
Uses a GtkNotebook to select services; results in IconView (broken...I
can't create the tabs until I know which services are relavant to the
results)

TRACKER_UI_LAYOUT=4 ./tracker-gui-prototypes.py
Just finds everything and puts it in a GtkTreeStore

You can also change the number of results returned at once by setting
the TRACKER_MAX_RESULTS environment variable,

TRACKER_UI_LAYOUT=1 TRACKER_MAX_RESULTS=50 ./tracker-gui-prototypes.py

Would show fifty at a time..default is 20.
#!/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()
_______________________________________________
tracker-list mailing list
[email protected]
http://mail.gnome.org/mailman/listinfo/tracker-list

Reply via email to