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)