sugar-toolkit-gtk3/shell/home/MeshGroup.py

import random

import goocanvas

import conf
from sugar.canvas.IconItem import IconItem
from sugar.canvas.IconItem import IconColor
from sugar.presence import PresenceService
from home.IconLayout import IconLayout

class ActivityItem(IconItem):
	def __init__(self, activity, service):
		self._service = service
		self._activity = activity

		IconItem.__init__(self, icon_name=self.get_icon_name(),
						  color=self.get_color(), size=96)

	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 MeshGroup(goocanvas.Group):
	def __init__(self):
		goocanvas.Group.__init__(self)
		self._icon_layout = IconLayout(1200, 900)
		self._activities = {}

		self._pservice = PresenceService.get_instance()
		self._pservice.connect("service-appeared", self._service_appeared_cb)
		self._pservice.connect('activity-disappeared', self._activity_disappeared_cb)

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

	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):
		item = ActivityItem(activity, service)
		item.connect('clicked', self._activity_clicked_cb)
		self._icon_layout.add_icon(item)
		self.add_child(item)

		self._activities[item.get_id()] = item

	def _activity_disappeared_cb(self, pservice, activity):
		if self._activities.has_key(activity.get_id()):
			self.remove_child(self._activities[activity.get_id()])
			del self._activities[activity.get_id()]

	def _activity_clicked_cb(self, item):
		default_type = item.get_service().get_type()
		registry = conf.get_activity_registry()

		bundle_id = registry.get_activity_from_type(default_type).get_id()
		activity_id = item.get_id()

		self._shell.join_activity(bundle_id, activity_id)