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

import random

import goocanvas

from sugar.canvas.IconItem import IconItem
import Theme

class FriendIcon(IconItem):
	def __init__(self, friend):
		IconItem.__init__(self, icon_name='stock-buddy',
						  color=friend.get_color(), size=96)
		self._friend = friend

	def get_friend(self):
		return self._friend

class FriendsGroup(goocanvas.Group):
	WIDTH = 1200.0 * 1.9
	HEIGHT = 900.0 * 1.9

	def __init__(self, icon_layout, data_model):
		goocanvas.Group.__init__(self)
		self._friend_to_child = {}

		self._icon_layout = icon_layout

		self._theme = Theme.get_instance()
		self._theme.connect("theme-changed", self.__theme_changed_cb)

		color = self._theme.get_home_friends_color()
		self._friends_rect = goocanvas.Rect(width=FriendsGroup.WIDTH,
											height=FriendsGroup.HEIGHT,
											line_width=0, fill_color=color,
											radius_x=60, radius_y=60)
		self.add_child(self._friends_rect)

		for friend in data_model:
			self.add_friend(friend)

		data_model.connect('friend-added', self.__friend_added_cb)
		data_model.connect('friend-removed', self.__friend_removed_cb)

	def __theme_changed_cb(self, theme):
		color = self._theme.get_home_friends_color()
		self._friends_rect.set_property("fill-color", color)

	def add_friend(self, friend):
		icon = FriendIcon(friend)
		self.add_child(icon)
		self._friend_to_child[friend] = icon
		self._icon_layout.add_icon(icon)

	def remove_friend(self, friend):
		icon = self._friend_to_child[friend]
		self._icon_layout.remove_icon(icon)
		self.remove_child(self.find_child(icon))
		del self._friend_to_child[friend]

	def __friend_added_cb(self, data_model, friend):
		self.add_friend(friend)

	def __friend_removed_cb(self, data_model, friend):
		self.remove_friend(friend)