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

import random

import hippo
import gobject

from sugar.graphics.spreadbox import SpreadBox
from sugar.graphics import style
from view.home.MyIcon import MyIcon
from view.home.FriendView import FriendView

class FriendsBox(SpreadBox, hippo.CanvasItem):
	__gtype_name__ = 'SugarFriendsBox'
	def __init__(self, shell, menu_shell):
		SpreadBox.__init__(self, background_color=0xe2e2e2ff)

		self._shell = shell
		self._menu_shell = menu_shell
		self._friends = {}

		self._my_icon = MyIcon()
		style.apply_stylesheet(self._my_icon, 'friends.MyIcon')
		self.append(self._my_icon, hippo.PACK_FIXED)

		friends = self._shell.get_model().get_friends()

		for friend in friends:
			self.add_friend(friend)

		friends.connect('friend-added', self._friend_added_cb)
		friends.connect('friend-removed', self._friend_removed_cb)

	def add_friend(self, buddy_info):
		icon = FriendView(self._shell, self._menu_shell, buddy_info)
		self.add(icon)

		self._friends[buddy_info.get_name()] = icon

	def _friend_added_cb(self, data_model, buddy_info):
		self.add_friend(buddy_info)

	def _friend_removed_cb(self, data_model, name):
		self.remove(self._friends[name])
		del self._friends[name]

	def do_allocate(self, width, height):
		SpreadBox.do_allocate(self, width, height)

		[icon_width, icon_height] = self._my_icon.get_allocation()
		self.move(self._my_icon, (width - icon_width) / 2,
				  (height - icon_height) / 2)