5c7da12cd5
Need some theme love... (Code is an hack, but it's not worth implementing it right until we have more definite plans about this)
279 lines
8.5 KiB
Python
279 lines
8.5 KiB
Python
import pygtk
|
|
pygtk.require('2.0')
|
|
import gtk
|
|
import pango
|
|
import dbus
|
|
import cgi
|
|
import xml.sax.saxutils
|
|
import re
|
|
import gobject
|
|
|
|
import google
|
|
from sugar.presence.PresenceService import PresenceService
|
|
from sugar.presence import Service
|
|
from sugar.browser import BrowserActivity
|
|
|
|
_COLUMN_TITLE = 0
|
|
_COLUMN_ADDRESS = 1
|
|
_COLUMN_SUBTITLE = 2
|
|
_COLUMN_SERVICE = 3
|
|
|
|
class SearchHelper(object):
|
|
def __init__(self, activity_uid):
|
|
self.search_uid = activity_uid
|
|
self.found = False
|
|
|
|
class SearchModel(gtk.ListStore):
|
|
def __init__(self, activities_model, search_text):
|
|
gtk.ListStore.__init__(self, gobject.TYPE_STRING, gobject.TYPE_STRING,
|
|
gobject.TYPE_STRING, gobject.TYPE_PYOBJECT)
|
|
|
|
for row in activities_model:
|
|
title = row[_COLUMN_TITLE]
|
|
address = row[_COLUMN_ADDRESS]
|
|
if title.find(search_text) >= 0 or address.find(search_text) >= 0:
|
|
self.append([ title, address, row[_COLUMN_SUBTITLE], row[_COLUMN_SERVICE] ])
|
|
|
|
google.LICENSE_KEY = '1As9KaJQFHIJ1L0W5EZPl6vBOFvh/Vaf'
|
|
data = google.doGoogleSearch(search_text)
|
|
|
|
for result in data.results:
|
|
title = result.title
|
|
|
|
# FIXME what tags should we actually strip?
|
|
title = title.replace('<b>', '')
|
|
title = title.replace('</b>', '')
|
|
|
|
# FIXME I'm sure there is a better way to
|
|
# unescape these.
|
|
title = title.replace('"', '"')
|
|
title = title.replace('&', '&')
|
|
|
|
self.append([ title, result.URL, None, None ])
|
|
|
|
class ActivitiesModel(gtk.ListStore):
|
|
def __init__(self):
|
|
gtk.ListStore.__init__(self, gobject.TYPE_STRING, gobject.TYPE_STRING,
|
|
gobject.TYPE_STRING, gobject.TYPE_PYOBJECT)
|
|
|
|
def _filter_dupe_activities(self, model, path, it, user_data):
|
|
"""Search the list of list rows for an existing service that
|
|
has the activity ID we're looking for."""
|
|
helper = user_data
|
|
(service, ) = model.get(it, _COLUMN_SERVICE)
|
|
if not service:
|
|
return False
|
|
if service.get_activity_uid() == helper.search_uid:
|
|
helper.found = True
|
|
return True
|
|
return False
|
|
|
|
def add_activity(self, buddy, service):
|
|
# Web Activity check
|
|
activity_uid = service.get_activity_uid()
|
|
if activity_uid is None:
|
|
return
|
|
# Don't show dupes
|
|
helper = SearchHelper(activity_uid)
|
|
self.foreach(self._filter_dupe_activities, helper)
|
|
if helper.found == True:
|
|
return
|
|
|
|
# Only accept browser activities for now
|
|
if service.get_type() == BrowserActivity._BROWSER_ACTIVITY_TYPE:
|
|
escaped_title = service.get_one_property('Title')
|
|
escaped_uri = service.get_one_property('URI')
|
|
if escaped_title and escaped_uri:
|
|
title = xml.sax.saxutils.unescape(escaped_title)
|
|
address = xml.sax.saxutils.unescape(escaped_uri)
|
|
subtitle = 'Shared by %s' % buddy.get_nick_name()
|
|
self.append([ title, address, subtitle, service ])
|
|
|
|
class ActivitiesView(gtk.TreeView):
|
|
def __init__(self, model):
|
|
gtk.TreeView.__init__(self, model)
|
|
|
|
self.set_headers_visible(False)
|
|
|
|
theme = gtk.icon_theme_get_default()
|
|
size = 48
|
|
self._web_pixbuf = theme.load_icon('emblem-web', size, 0)
|
|
self._share_pixbuf = theme.load_icon('emblem-people', size, 0)
|
|
|
|
column = gtk.TreeViewColumn('')
|
|
self.append_column(column)
|
|
|
|
cell = gtk.CellRendererPixbuf()
|
|
column.pack_start(cell, False)
|
|
column.set_cell_data_func(cell, self._icon_cell_data_func)
|
|
|
|
cell = gtk.CellRendererText()
|
|
column.pack_start(cell)
|
|
column.set_cell_data_func(cell, self._cell_data_func)
|
|
|
|
self.connect('row-activated', self._row_activated_cb)
|
|
|
|
def _icon_cell_data_func(self, column, cell, model, it):
|
|
if model.get_value(it, _COLUMN_SERVICE) == None:
|
|
cell.set_property('pixbuf', self._web_pixbuf)
|
|
else:
|
|
cell.set_property('pixbuf', self._share_pixbuf)
|
|
|
|
def _cell_data_func(self, column, cell, model, it):
|
|
title = model.get_value(it, _COLUMN_TITLE)
|
|
subtitle = model.get_value(it, _COLUMN_SUBTITLE)
|
|
if subtitle is None:
|
|
subtitle = model.get_value(it, _COLUMN_ADDRESS)
|
|
|
|
markup = '<big><b>' + cgi.escape(title) + '</b></big>'
|
|
markup += '\n' + cgi.escape(subtitle)
|
|
|
|
cell.set_property('markup', markup)
|
|
cell.set_property('ellipsize', pango.ELLIPSIZE_END)
|
|
|
|
def _row_activated_cb(self, treeview, path, column):
|
|
bus = dbus.SessionBus()
|
|
proxy_obj = bus.get_object('com.redhat.Sugar.Browser', '/com/redhat/Sugar/Browser')
|
|
browser_shell = dbus.Interface(proxy_obj, 'com.redhat.Sugar.BrowserShell')
|
|
|
|
model = self.get_model()
|
|
address = model.get_value(model.get_iter(path), _COLUMN_ADDRESS)
|
|
service = model.get_value(model.get_iter(path), _COLUMN_SERVICE)
|
|
|
|
print 'Activated row %s' % address
|
|
|
|
if service is None:
|
|
browser_shell.open_browser(address)
|
|
else:
|
|
serialized_service = service.serialize()
|
|
browser_shell.open_browser(address, serialized_service)
|
|
|
|
class StartPage(gtk.HBox):
|
|
def __init__(self, ac_signal_object):
|
|
gtk.HBox.__init__(self)
|
|
|
|
self._ac_signal_object = ac_signal_object
|
|
self._ac_signal_object.connect("local-activity-started",
|
|
self._on_local_activity_started_cb)
|
|
self._ac_signal_object.connect("local-activity-ended",
|
|
self._on_local_activity_ended_cb)
|
|
|
|
self._pservice = PresenceService.get_instance()
|
|
self._pservice.connect("activity-announced", self._on_activity_announced_cb)
|
|
self._pservice.connect("new-service-adv", self._on_new_service_adv_cb)
|
|
self._pservice.start()
|
|
self._pservice.track_service_type(BrowserActivity._BROWSER_ACTIVITY_TYPE)
|
|
|
|
vbox = gtk.VBox()
|
|
|
|
search_box = gtk.HBox(False, 6)
|
|
search_box.set_border_width(24)
|
|
|
|
self._search_entry = gtk.Entry()
|
|
self._search_entry.connect('activate', self._search_entry_activate_cb)
|
|
search_box.pack_start(self._search_entry)
|
|
self._search_entry.show()
|
|
|
|
search_button = gtk.Button("Search")
|
|
search_button.connect('clicked', self._search_button_clicked_cb)
|
|
search_box.pack_start(search_button, False)
|
|
search_button.show()
|
|
|
|
vbox.pack_start(search_box, False, True)
|
|
search_box.show()
|
|
|
|
exp_space = gtk.Label('')
|
|
vbox.pack_start(exp_space)
|
|
exp_space.show()
|
|
|
|
self.pack_start(vbox)
|
|
vbox.show()
|
|
|
|
vbox = gtk.VBox()
|
|
|
|
self._search_close_box = gtk.HBox()
|
|
|
|
self._search_close_label = gtk.Label()
|
|
self._search_close_label.set_alignment(0.0, 0.5)
|
|
self._search_close_box.pack_start(self._search_close_label)
|
|
self._search_close_label.show()
|
|
|
|
close_image = gtk.Image()
|
|
close_image.set_from_stock (gtk.STOCK_CLOSE, gtk.ICON_SIZE_MENU)
|
|
close_image.show()
|
|
|
|
search_close_button = gtk.Button()
|
|
rcstyle = gtk.RcStyle();
|
|
rcstyle.xthickness = rcstyle.ythickness = 0;
|
|
search_close_button.modify_style (rcstyle);
|
|
search_close_button.add(close_image)
|
|
search_close_button.set_relief(gtk.RELIEF_NONE)
|
|
search_close_button.set_focus_on_click(False)
|
|
search_close_button.connect("clicked", self.__search_close_button_clicked_cb)
|
|
|
|
self._search_close_box.pack_start(search_close_button, False)
|
|
search_close_button.show()
|
|
|
|
vbox.pack_start(self._search_close_box, False)
|
|
|
|
sw = gtk.ScrolledWindow()
|
|
sw.set_size_request(320, -1)
|
|
sw.set_policy(gtk.POLICY_NEVER, gtk.POLICY_AUTOMATIC)
|
|
|
|
self._activities_model = ActivitiesModel()
|
|
|
|
self._activities = ActivitiesView(self._activities_model)
|
|
sw.add(self._activities)
|
|
self._activities.show()
|
|
|
|
vbox.pack_start(sw)
|
|
sw.show()
|
|
|
|
self.pack_start(vbox)
|
|
vbox.show()
|
|
|
|
def __search_close_button_clicked_cb(self, button):
|
|
self._search(None)
|
|
|
|
def _on_local_activity_started_cb(self, helper, activity_container, activity_id):
|
|
self._pservice.track_activity(activity_id)
|
|
print "new local activity %s" % activity_id
|
|
|
|
def _on_local_activity_ended_cb(self, helper, activity_container, activity_id):
|
|
self._pservice.untrack_activity(activity_id)
|
|
|
|
def _on_new_service_adv_cb(self, pservice, activity_id, short_stype):
|
|
if activity_id:
|
|
self._pservice.track_activity(activity_id)
|
|
self._pservice.track_service_type(short_stype)
|
|
|
|
def _on_activity_announced_cb(self, pservice, service, buddy):
|
|
print "Found new activity with type %s" % service.get_full_type()
|
|
self._activities_model.add_activity(buddy, service)
|
|
if self._activities.get_model() != self._activities_model:
|
|
self._search(self._last_search)
|
|
|
|
def _search_entry_activate_cb(self, entry):
|
|
self._search()
|
|
self._search_entry.set_text('')
|
|
|
|
def _search_button_clicked_cb(self, button):
|
|
self._search()
|
|
self._search_entry.set_text('')
|
|
|
|
def _search(self, text = None):
|
|
if text == None:
|
|
text = self._search_entry.get_text()
|
|
|
|
if text == None or len(text) == 0:
|
|
self._activities.set_model(self._activities_model)
|
|
self._search_close_box.hide()
|
|
else:
|
|
search_model = SearchModel(self._activities_model, text)
|
|
self._activities.set_model(search_model)
|
|
|
|
self._search_close_label.set_text('Search for %s' % (text))
|
|
self._search_close_box.show()
|
|
|
|
self._last_search = text
|