sugar-toolkit-gtk3/shell/frame/BottomPanel.py

import gtk
import goocanvas
import logging

import conf
from sugar.canvas.IconItem import IconItem
from sugar.canvas.IconColor import IconColor
from sugar.presence import PresenceService
from frame.Panel import Panel

class ActivityItem(IconItem):
	def __init__(self, activity, size):
		icon_name = activity.get_icon()
		if not icon_name:
			act_type = activity.get_type()
			raise RuntimeError("Activity %s did not have an icon!" % act_type)
		IconItem.__init__(self, icon_name=icon_name,
						  color=IconColor('white'), size=size)
		self._activity = activity

	def get_bundle_id(self):
		return self._activity.get_id()

class InviteItem(IconItem):
	def __init__(self, invite, size):
		IconItem.__init__(self, icon_name=invite.get_icon(),
						  color=invite.get_color(), size=size)
		self._invite = invite

	def get_activity_id(self):
		return self._invite.get_activity_id()

	def get_bundle_id(self):
		return self._invite.get_bundle_id()

class ActivityBar(goocanvas.Group):
	def __init__(self, shell, invites, height):
		goocanvas.Group.__init__(self)

		self._shell = shell
		self._height = height

		registry = conf.get_activity_registry()
		for activity in registry.list_activities():
			if activity.get_show_launcher():
				self.add_activity(activity)

		for invite in invites:
			self.add_invite(invite)
		invites.connect('invite-added', self.__invite_added_cb)

	def __activity_clicked_cb(self, icon):
		self._shell.start_activity(icon.get_bundle_id())

	def __invite_clicked_cb(self, icon):
		self._shell.join_activity(icon.get_bundle_id(),
								  icon.get_activity_id())
	
	def __invite_added_cb(self, invites, invite):
		self.add_invite(invite)

	def add_activity(self, activity):
		# Need an icon to show up on the bar
		if not activity.get_icon():
			name = activity.get_name()
			logging.info("Activity %s did not have an icon.  Won't show it." % name)
			return

		item = ActivityItem(activity, self._height)
		item.connect('clicked', self.__activity_clicked_cb)

		icon_size = self._height
		x = (icon_size + 6) * self.get_n_children()
		item.set_property('x', x)

		self.add_child(item)

	def add_invite(self, invite):
		item = InviteItem(invite, self._height)
		item.connect('clicked', self.__invite_clicked_cb)

		icon_size = self._height
		x = (icon_size + 6) * self.get_n_children()
		item.set_property('x', x)

		self.add_child(item)

class BottomPanel(Panel):
	def __init__(self, shell, invites):
		Panel.__init__(self)

		self._shell = shell
		self._invites = invites

	def construct(self):
		Panel.construct(self)

		root = self.get_root()

		activity_bar = ActivityBar(self._shell, self._invites,
								   self.get_height())
		root.add_child(activity_bar)