diff options
Diffstat (limited to 'shell/model/MeshModel.py')
-rw-r--r-- | shell/model/MeshModel.py | 108 |
1 files changed, 108 insertions, 0 deletions
diff --git a/shell/model/MeshModel.py b/shell/model/MeshModel.py new file mode 100644 index 0000000..2b47baf --- /dev/null +++ b/shell/model/MeshModel.py @@ -0,0 +1,108 @@ +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()] |