sugar-toolkit-gtk3/shell/view/home/MeshBox.py

import random

import hippo
import gobject

from sugar.graphics.spreadbox import SpreadBox
from sugar.graphics.snowflakebox import SnowflakeBox
from sugar.graphics.canvasicon import CanvasIcon
from view.BuddyIcon import BuddyIcon
import conf

class ActivityView(SnowflakeBox):
	def __init__(self, shell, menu_shell, model):
		SnowflakeBox.__init__(self)

		self._shell = shell
		self._model = model
		self._icons = {}

		icon = CanvasIcon(icon_name=model.get_icon_name(),
						  color=model.get_color(), size=80)
		icon.connect('activated', self._clicked_cb)
		self.append(icon, hippo.PACK_FIXED)
		self.set_root(icon)

	def has_buddy_icon(self, name):
		return self._icons.has_key(name)

	def add_buddy_icon(self, name, icon):
		self._icons[name] = icon
		self.append(icon, hippo.PACK_FIXED)

	def remove_buddy_icon(self, name):
		icon = self._icons[name]
		self.remove(icon)
		del self._icons[name]

	def _clicked_cb(self, item):
		registry = conf.get_activity_registry()
		default_type = self._model.get_service().get_type()
		bundle = registry.get_activity_from_type(default_type)
		self._shell.join_activity(bundle.get_id(), self._model.get_id())

class MeshBox(SpreadBox):
	def __init__(self, shell, menu_shell):
		SpreadBox.__init__(self, background_color=0xe2e2e2ff)

		self._shell = shell
		self._menu_shell = menu_shell
		self._model = shell.get_model().get_mesh()
		self._buddies = {}
		self._activities = {}
		self._buddy_to_activity = {}

		for buddy_model in self._model.get_buddies():
			self._add_buddy(buddy_model)

		self._model.connect('buddy-added', self._buddy_added_cb)
		self._model.connect('buddy-removed', self._buddy_removed_cb)
		self._model.connect('buddy-moved', self._buddy_moved_cb)

		for activity_model in self._model.get_activities():
			self._add_activity(activity_model)

		self._model.connect('activity-added', self._activity_added_cb)
		self._model.connect('activity-removed', self._activity_removed_cb)

	def _buddy_added_cb(self, model, buddy_model):
		self._add_alone_buddy(buddy_model)

	def _buddy_removed_cb(self, model, buddy_model):
		self._remove_buddy(buddy_model) 

	def _buddy_moved_cb(self, model, buddy_model, activity_model):
		self._move_buddy(buddy_model, activity_model)

	def _activity_added_cb(self, model, activity_model):
		self._add_activity(activity_model)

	def _activity_removed_cb(self, model, activity_model):
		self._remove_activity(activity_model) 

	def _add_alone_buddy(self, buddy_model):
		icon = BuddyIcon(self._shell, self._menu_shell, buddy_model)
		icon.props.size = 80
		self.add(icon)

		self._buddies[buddy_model.get_name()] = icon

	def _remove_alone_buddy(self, buddy_model):
		icon = self._buddies[buddy_model.get_name()]
		self.remove(icon)
		del self._buddies[buddy_model.get_name()]

	def _remove_buddy(self, buddy_model):
		name = buddy_model.get_name()
		if self._buddies.has_key(name):
			self._remove_alone_buddy(buddy_model)
		else:
			for activity in self._activities.values():
				if activity.has_buddy_icon(name):
					activity.remove_buddy_icon(name)

	def _move_buddy(self, buddy_model, activity_model):
		name = buddy_model.get_name()

		self._remove_buddy(buddy_model)

		if activity_model == None:
			self._add_alone_buddy(buddy_model)
		else:
			activity = self._activities[activity_model.get_id()]

			icon = BuddyIcon(self._shell, self._menu_shell, buddy_model)
			icon.props.size = 60
			activity.add_buddy_icon(buddy_model.get_name(), icon)

	def _add_activity(self, activity_model):
		icon = ActivityView(self._shell, self._menu_shell, activity_model)
		self.add(icon)

		self._activities[activity_model.get_id()] = icon

	def _remove_activity(self, activity_model):
		icon = self._activities[activity_model.get_id()]
		self.remove(icon)
		del self._activities[activity_model.get_id()]