import gobject from sugar.presence import PresenceService from sugar.canvas.IconColor import IconColor import logging class Friend: def __init__(self, buddy): self._buddy = buddy def get_name(self): return self._buddy.get_name() def get_color(self): color = self._buddy.get_color() try: icolor = IconColor(color) except RuntimeError: icolor = IconColor() logging.info("Buddy %s doesn't have an allowed color; using a random color instead." % self.get_name()) return icolor class FriendsModel(gobject.GObject): __gsignals__ = { 'friend-added': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), 'friend-removed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])) } def __init__(self): gobject.GObject.__init__(self) self._friends = [] self._pservice = PresenceService.get_instance() self._pservice.connect("buddy-appeared", self.__buddy_appeared_cb) for buddy in self._pservice.get_buddies(): self.add_friend(buddy) def add_friend(self, buddy): friend = Friend(buddy) self._friends.append(friend) self.emit('friend-added', friend) def __iter__(self): return self._friends.__iter__() def __buddy_appeared_cb(self, pservice, buddy): self.add_friend(buddy)