109 lines
3.3 KiB
Python
109 lines
3.3 KiB
Python
import gobject
|
|
|
|
import conf
|
|
from sugar.canvas.IconItem import IconColor
|
|
from sugar.presence import PresenceService
|
|
from model.BuddyModel import BuddyModel
|
|
|
|
class ActivityModel:
|
|
def __init__(self, activity, service):
|
|
self._service = service
|
|
self._activity = activity
|
|
|
|
def get_id(self):
|
|
return self._activity.get_id()
|
|
|
|
def get_icon_name(self):
|
|
registry = conf.get_activity_registry()
|
|
info = registry.get_activity_from_type(self._service.get_type())
|
|
|
|
return info.get_icon()
|
|
|
|
def get_color(self):
|
|
return IconColor(self._activity.get_color())
|
|
|
|
def get_service(self):
|
|
return self._service
|
|
|
|
class MeshModel(gobject.GObject):
|
|
__gsignals__ = {
|
|
'activity-added': (gobject.SIGNAL_RUN_FIRST,
|
|
gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])),
|
|
'activity-removed': (gobject.SIGNAL_RUN_FIRST,
|
|
gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])),
|
|
'buddy-added': (gobject.SIGNAL_RUN_FIRST,
|
|
gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])),
|
|
'buddy-moved': (gobject.SIGNAL_RUN_FIRST,
|
|
gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT,
|
|
gobject.TYPE_PYOBJECT])),
|
|
'buddy-removed': (gobject.SIGNAL_RUN_FIRST,
|
|
gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT]))
|
|
}
|
|
|
|
def __init__(self):
|
|
gobject.GObject.__init__(self)
|
|
|
|
self._activities = {}
|
|
self._buddies = {}
|
|
|
|
self._pservice = PresenceService.get_instance()
|
|
self._pservice.connect("service-appeared",
|
|
self._service_appeared_cb)
|
|
self._pservice.connect('activity-disappeared',
|
|
self._activity_disappeared_cb)
|
|
self._pservice.connect("buddy-appeared",
|
|
self._buddy_appeared_cb)
|
|
self._pservice.connect("buddy-disappeared",
|
|
self._buddy_disappeared_cb)
|
|
|
|
for service in self._pservice.get_services():
|
|
self._check_service(service)
|
|
|
|
def get_activities(self):
|
|
return self._activities
|
|
|
|
def get_buddies(self):
|
|
return self._buddies
|
|
|
|
def _buddy_activity_changed_cb(self, buddy, cur_activity):
|
|
buddy_model = self._buddies[buddy.get_name()]
|
|
activity_model = self._activities[cur_activity.get_id()]
|
|
self.emit('buddy-moved', buddy_model, activity_model)
|
|
|
|
def _buddy_appeared_cb(self, pservice, buddy):
|
|
model = BuddyModel(buddy=buddy)
|
|
model.connect('current-activity-changed',
|
|
self._buddy_activity_changed_cb)
|
|
self._buddies[model.get_name()] = model
|
|
self.emit('buddy-added', model)
|
|
|
|
def _buddy_disappeared_cb(self, pservice, buddy):
|
|
self.emit('buddy-removed', buddy)
|
|
del self._buddies[buddy.get_name()]
|
|
|
|
def _service_appeared_cb(self, pservice, service):
|
|
self._check_service(service)
|
|
|
|
def _check_service(self, service):
|
|
registry = conf.get_activity_registry()
|
|
if registry.get_activity_from_type(service.get_type()) != None:
|
|
activity_id = service.get_activity_id()
|
|
if not self.has_activity(activity_id):
|
|
activity = self._pservice.get_activity(activity_id)
|
|
if activity != None:
|
|
self.add_activity(activity, service)
|
|
|
|
def has_activity(self, activity_id):
|
|
return self._activities.has_key(activity_id)
|
|
|
|
def add_activity(self, activity, service):
|
|
model = ActivityModel(activity, service)
|
|
self._activities[model.get_id()] = model
|
|
self.emit('activity-added', model)
|
|
|
|
def _activity_disappeared_cb(self, pservice, activity):
|
|
if self._activities.has_key(activity.get_id()):
|
|
activity_model = self._activities[activity.get_id()]
|
|
self.emit('activity-removed', activity_model)
|
|
del self._activities[activity.get_id()]
|