sugar-toolkit-gtk3/shell/ActivitiesModel.py

import gobject

from sugar.presence.PresenceService import PresenceService
from ActivityRegistry import ActivityRegistry

class ActivityInfo:
	def __init__(self, service):
		self._service = service
	
	def get_id(self):
		return self._service.get_activity_id()
		
	def get_type(self):
		return self._service.get_type()
	
	def get_title(self):
		return self._service.get_published_value('title')
	
	def get_service(self):
		return self._service

class ActivitiesModel(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]))
	}

	def __init__(self, registry):
		gobject.GObject.__init__(self)
		
		self._activities = {}
		self._registry = registry
		
		self._pservice = PresenceService()
		self._pservice.connect("service-appeared", self.__service_appeared_cb)

		for service in self._pservice.get_services():
			self.__check_service(service)

	def has_activity(self, activity_id):
		return self._activities.has_key(activity_id)

	def add_activity(self, service):
		activity_info = ActivityInfo(service)
		self._activities[activity_info.get_id()] = (activity_info)
		self.emit('activity-added', activity_info)

	def __iter__(self):
		activities = self._activities.values()
		return activities.__iter__()

	def __service_appeared_cb(self, pservice, service):
		self.__check_service(service)

	def __check_service(self, service):
		if self._registry.get_activity(service.get_type()) != None:
			if not self.has_activity(service.get_activity_id()):
				self.add_activity(service)