import goocanvas from frame.PanelWindow import PanelWindow from sugar.canvas.IconItem import IconItem from sugar.canvas.IconColor import IconColor from sugar.canvas.GridLayout import GridGroup from sugar.canvas.GridLayout import GridConstraints from sugar.presence import PresenceService class RightPanel(GridGroup): def __init__(self, shell, friends): GridGroup.__init__(self, 1, 10) self._shell = shell self._friends = friends self._activity_ps = None self._joined_hid = -1 self._left_hid = -1 self._pservice = PresenceService.get_instance() self._pservice.connect('activity-appeared', self.__activity_appeared_cb) shell.connect('activity-changed', self.__activity_changed_cb) def add(self, buddy): icon = IconItem(icon_name='stock-buddy', color=IconColor(buddy.get_color())) icon.connect('clicked', self.__buddy_clicked_cb, buddy) row = self.get_n_children() + 2 constraints = GridConstraints(0, row , 1, 1, 6) self._layout.set_constraints(icon, constraints) self.add_child(icon) def remove(self, buddy): pass def clear(self): pass def __activity_appeared_cb(self, pservice, activity_ps): activity = self._shell.get_current_activity() if activity_ps.get_id() == activity.get_id(): self._set_activity_ps(activity_ps) def _set_activity_ps(self, activity_ps): if self._activity_ps == activity_ps: return if self._joined_hid > 0: self._activity_ps.disconnect(self._joined_hid) self._joined_hid = -1 if self._left_hid > 0: self._activity_ps.disconnect(self._left_hid) self._left_hid = -1 self._activity_ps = activity_ps self.clear() if activity_ps != None: for buddy in activity_ps.get_joined_buddies(): self.add(buddy) self._joined_hid = activity_ps.connect( 'buddy-joined', self.__buddy_joined_cb) self._left_hid = activity_ps.connect( 'buddy-left', self.__buddy_left_cb) def __activity_changed_cb(self, group, activity): activity_ps = self._pservice.get_activity(activity.get_id()) self._set_activity_ps(activity_ps) def __buddy_joined_cb(self, activity, buddy): self.add(buddy) def __buddy_left_cb(self, activity, buddy): self.remove(buddy) def __buddy_clicked_cb(self, icon, buddy): self._friends.add_buddy(buddy)