#!/usr/bin/env python # # Demo RSS client using tracker as backend # Copyright (C) 2009, Nokia # # This program is free software; you can redistribute it and/or # modify it under the terms of the GNU General Public License # as published by the Free Software Foundation; either version 2 # of the License, or (at your option) any later version. # # This program is distributed in the hope that it will be useful, # but WITHOUT ANY WARRANTY; without even the implied warranty of # MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the # GNU General Public License for more details. # # You should have received a copy of the GNU General Public License # along with this program; if not, write to the Free Software # Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA # 02110-1301, USA. # import gtk import gobject import pango import dbus from dbus.mainloop.glib import DBusGMainLoop from tracker_backend import TrackerRSS try: from zeitgeist_backend import ZeitgeistBackend zg = ZeitgeistBackend () except ImportError: class MockZeitgeistBackend: def view_event (self, uri): pass def get_rss_by_usage (self): return [] def clean_events (self): pass zg = MockZeitgeistBackend () # uri, title, date, isread, pango.Weight # Be careful, we use this also in the code for the zg_model! posts_model = gtk.ListStore (str, str, str, bool, int) # uri, title, entries, enabled channels_model = gtk.ListStore (str, str, str, bool) ui = gtk.Builder () ui.add_from_file ("./rss_viewer.ui") channels_treeview = None sources_dialog = None # Convenience defines FALSE = "false" TRUE = "true" NORMAL = pango.WEIGHT_NORMAL BOLD = pango.WEIGHT_BOLD tracker = TrackerRSS () def populate_initial_posts (): results = tracker.get_post_sorted_by_date (10) for result in results: if (result[3] == FALSE): row = (result[0], result[1], result[2], False, BOLD) else: row = (result[0], result[1], result[2], True, NORMAL) posts_model.append (row) def create_posts_tree_view (): for treeview in [ui.get_object ("posts_treeview"), ui.get_object ("posts_treeview_by_usage")]: column_title = gtk.TreeViewColumn ("Title", gtk.CellRendererText (), text=1, weight=4) column_date = gtk.TreeViewColumn ("Date", gtk.CellRendererText (), text=2) treeview.append_column (column_title) treeview.append_column (column_date) treeview.connect ("cursor-changed", cursor_changed_cb) def create_channels_tree_view (): treeview = gtk.TreeView () toggle = gtk.CellRendererToggle () toggle.set_property ("activatable", True) def toggled (widget, path): it = channels_model.get_iter_from_string (path) uri, old_value = channels_model.get (it, 0, 3) if (old_value == True): tracker.mark_as_invisible (uri) else: tracker.mark_as_visible (uri) channels_model.set (it, 3, not old_value) toggle.connect ("toggled", toggled) column_enable = gtk.TreeViewColumn ("Enabled", toggle, active=3) column_title = gtk.TreeViewColumn ("Title", gtk.CellRendererText (), text=1) column_entries = gtk.TreeViewColumn ("Entries", gtk.CellRendererText (), text=2) treeview.append_column (column_enable) treeview.append_column (column_title) treeview.append_column (column_entries) return treeview def toggle_row_foreach (treemodel, path, iter): uri, readed, text = treemodel.get (iter, 0, 3 ,1) if (readed): #Mark as unread treemodel.set_value (iter, 3, False) treemodel.set_value (iter, 4, BOLD) tracker.set_is_read (uri, False) else: #Mark as readed treemodel.set_value (iter, 3, True) treemodel.set_value (iter, 4, NORMAL) tracker.set_is_read (uri, True) def clicked_toggle_cb (widget): selected = ui.get_object ("posts_treeview").get_selection () if (selected.count_selected_rows () == 0): return selected.selected_foreach (toggle_row_foreach) def cursor_changed_cb (tv): selection = tv.get_selection () if (selection.count_selected_rows () > 0): model, it = selection.get_selected () uri = model.get_value (it, 0) text = tracker.get_text_for_uri (uri) tv = ui.get_object ("text_view") tv.get_buffer ().set_text (text) zg.view_event (uri) def clicked_sources_cb (widget, dialog): # Dont do this all the time! if (len (channels_model) == 0): feeds = tracker.get_all_subscribed_feeds () for (uri, name, value, visible) in feeds: channels_model.append ((uri, name, value, visible)) channels_treeview.show_all () retval = dialog.run () if (retval == gtk.RESPONSE_ACCEPT): pass dialog.hide () def notification_addition (added_uris): print "%d add: %s" % (len(added_uris), [str(n) for n in added_uris]) for uri in added_uris: details = tracker.get_info_for_entry (uri) if (details): if (details[2]): model_row = (uri, details[0], details[1], details[2], NORMAL) else: model_row = (uri, details[0], details[1], details[2], BOLD) posts_model.prepend (model_row) def remove_uris (model, path, iter, user_data): uri = model.get (iter, 0) if (uri[0] in user_data): model.remove (iter) return True def notification_removal (removed_uris): print "%d remove: %s" % (len(removed_uris), [str(n) for n in removed_uris]) posts_model.foreach (remove_uris, removed_uris) def update_uri (model, path, iter, user_data): updated_uri = user_data [0] uri = model.get (iter, 0) if (uri[0] == updated_uri): model.set(iter, 1, user_data[1], 2, user_data[2], 3, user_data[3], 4, user_data[4]) return True def change_tab_cb (notebook, page, page_num): print "Change to tab", page_num if (page_num == 1): uris = zg.get_rss_by_usage () zg_model = gtk.ListStore (str, str, str, bool, int) for uri in uris: for row in posts_model: print uri, map (str,row) if row[0] == uri : print "Add to the model!" zg_model.append (row) ui.get_object ("posts_treeview_by_usage").set_model (zg_model) else: ui.get_object ("posts_treeview").set_model (posts_model) def notification_update (updated_uris, update_predicates): print "%d update: %s" % (len(updated_uris), [str(n) for n in updated_uris]) for uri in updated_uris: details = tracker.get_info_for_entry (uri) if (details): if (details[2]): model_row = (uri, details[0], details[1], details[2], NORMAL) else: model_row = (uri, details[0], details[1], details[2], BOLD) posts_model.foreach (update_uri, model_row) if __name__ == "__main__": dbus_loop = DBusGMainLoop(set_as_default=True) bus = dbus.SessionBus (dbus_loop) bus.add_signal_receiver (notification_addition, signal_name="SubjectsAdded", dbus_interface="org.freedesktop.Tracker1.Resources.Class", path="/org/freedesktop/Tracker1/Resources/Classes/mfo/FeedMessage") bus.add_signal_receiver (notification_removal, signal_name="SubjectsRemoved", dbus_interface="org.freedesktop.Tracker1.Resources.Class", path="/org/freedesktop/Tracker1/Resources/Classes/mfo/FeedMessage") bus.add_signal_receiver (notification_update, signal_name="SubjectsChanged", dbus_interface="org.freedesktop.Tracker1.Resources.Class", path="/org/freedesktop/Tracker1/Resources/Classes/mfo/FeedMessage") window = ui.get_object ("main_window") gobject.set_application_name ("Rss/tracker") create_posts_tree_view () ui.get_object("posts_treeview").set_model (posts_model) channels_treeview = create_channels_tree_view () channels_treeview.set_model (channels_model) dialog = gtk.Dialog ("Sources", window, gtk.DIALOG_MODAL|gtk.DIALOG_DESTROY_WITH_PARENT, (gtk.STOCK_CANCEL, gtk.RESPONSE_REJECT, gtk.STOCK_OK, gtk.RESPONSE_ACCEPT)) dialog.vbox.add (channels_treeview) ui.get_object ("button_toggle").connect ("clicked", clicked_toggle_cb) ui.get_object ("notebook1").connect ("switch-page", change_tab_cb) populate_initial_posts () window.show_all () def destroy (window): tracker.flush_to_file () gtk.main_quit () window.connect ("destroy", destroy) gtk.main ()