import random import goocanvas from sugar.canvas.IconItem import IconItem from home.IconLayout import IconLayout from home.MyIcon import MyIcon from FriendPopup import FriendPopup from sugar.canvas.Grid import Grid class FriendIcon(IconItem): def __init__(self, shell, friend): IconItem.__init__(self, icon_name='stock-buddy', color=friend.get_color(), size=96) self._shell = shell self._friend = friend self._popup = None self.connect('popup', self._popup_cb) self.connect('popdown', self._popdown_cb) def get_friend(self): return self._friend def _popup_cb(self, icon, x1, y1, x2, y2): grid = Grid() if not self._popup: self._popup = FriendPopup(self._shell, grid, icon.get_friend()) [grid_x1, grid_y1] = grid.convert_from_screen(x1, y1) [grid_x2, grid_y2] = grid.convert_from_screen(x2, y2) if grid_x2 + self._popup.get_width() + 1 > Grid.ROWS: grid_x = grid_x1 - self._popup.get_width() + 1 else: grid_x = grid_x2 - 1 grid_y = grid_y1 if grid_y < 0: grid_y = 0 if grid_y + self._popup.get_width() > Grid.ROWS: grid_y = Grid.ROWS - self._popup.get_width() grid.set_constraints(self._popup, grid_x, grid_y, self._popup.get_width(), self._popup.get_height()) self._popup.show() def _popup_destroy_cb(self, popup): self._popup = None def _popdown_cb(self, friend): if self._popup: self._popup.connect('destroy', self._popup_destroy_cb) self._popup.popdown() class FriendsGroup(goocanvas.Group): def __init__(self, shell, friends): goocanvas.Group.__init__(self) self._shell = shell self._icon_layout = IconLayout(1200, 900) self._friends = friends me = MyIcon(100) me.translate(600 - (me.get_property('size') / 2), 450 - (me.get_property('size') / 2)) self.add_child(me) for friend in self._friends: self.add_friend(friend) friends.connect('friend-added', self._friend_added_cb) def add_friend(self, friend): icon = FriendIcon(self._shell, friend) self.add_child(icon) self._icon_layout.add_icon(icon) def _friend_added_cb(self, data_model, friend): self.add_friend(friend)