from sugar.presence import PresenceService from sugar.graphics.iconcolor import IconColor import gobject _NOT_PRESENT_COLOR = "#888888,#BBBBBB" class BuddyModel(gobject.GObject): __gsignals__ = { 'appeared': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([])), 'disappeared': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([])), 'color-changed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])), 'icon-changed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([])), 'current-activity-changed': (gobject.SIGNAL_RUN_FIRST, gobject.TYPE_NONE, ([gobject.TYPE_PYOBJECT])) } def __init__(self, name=None, buddy=None): if name and buddy: raise RuntimeError("Must specify only _one_ of name or buddy.") gobject.GObject.__init__(self) self._ba_handler = None self._pc_handler = None self._dis_handler = None self._bic_handler = None self._cac_handler = None self._pservice = PresenceService.get_instance() self._buddy = None # If given just a name, try to get the buddy from the PS first if not buddy: self._name = name # FIXME: use public key, not name buddy = self._pservice.get_buddy_by_name(self._name) # If successful, copy properties from the PS buddy object if buddy: self.__update_buddy(buddy) else: # Otherwise, connect to the PS's buddy-appeared signal and # wait for the buddy to appear self._ba_handler = self._pservice.connect('buddy-appeared', self.__buddy_appeared_cb) self._name = name # Set color to 'inactive'/'disconnected' self.__set_color_from_string(_NOT_PRESENT_COLOR) def __set_color_from_string(self, color_string): self._color = IconColor(color_string) def get_name(self): return self._name def get_color(self): return self._color def get_buddy(self): return self._buddy def is_present(self): if self._buddy: return True return False def get_current_activity(self): if self._buddy: return self._buddy.get_current_activity() return None def __update_buddy(self, buddy): if not buddy: raise ValueError("Buddy cannot be None.") self._buddy = buddy self._name = self._buddy.get_name() self.__set_color_from_string(self._buddy.get_color()) self._pc_handler = self._buddy.connect('property-changed', self.__buddy_property_changed_cb) self._dis_handler = self._buddy.connect('disappeared', self.__buddy_disappeared_cb) self._bic_handler = self._buddy.connect('icon-changed', self.__buddy_icon_changed_cb) self._cac_handler = self._buddy.connect('current-activity-changed', self.__buddy_current_activity_changed_cb) def __buddy_appeared_cb(self, pservice, buddy): # FIXME: use public key rather than buddy name if self._buddy or buddy.get_name() != self._name: return if self._ba_handler: # Once we have the buddy, we no longer need to # monitor buddy-appeared events self._pservice.disconnect(self._ba_handler) self._ba_handler = None self.__update_buddy(buddy) self.emit('appeared') def __buddy_property_changed_cb(self, buddy, keys): if not self._buddy: return if 'color' in keys: self.__set_color_from_string(self._buddy.get_color()) self.emit('color-changed', self.get_color()) def __buddy_disappeared_cb(self, buddy): if buddy != self._buddy: return self._buddy.disconnect(self._pc_handler) self._buddy.disconnect(self._dis_handler) self._buddy.disconnect(self._bic_handler) self._buddy.disconnect(self._cac_handler) self.__set_color_from_string(_NOT_PRESENT_COLOR) self.emit('disappeared') self._buddy = None def __buddy_icon_changed_cb(self, buddy): self.emit('icon-changed') def __buddy_current_activity_changed_cb(self, buddy, activity=None): if not self._buddy: return if activity: self.emit('current-activity-changed', activity) else: self.emit('current-activity-changed')