X-Git-Url: http://git.maemo.org/git/?p=theonering;a=blobdiff_plain;f=src%2Fchannel_manager.py;h=d1e99423e673110be5096d488d1baa317185a577;hp=90fdd990376fdbe3f5bd6337a2693115b6127361;hb=02a126316c2b761233aed2d194e61074fc69620b;hpb=e00e3978a660d399c5fe23cb4a1fd5da75fd4eb4 diff --git a/src/channel_manager.py b/src/channel_manager.py index 90fdd99..d1e9942 100644 --- a/src/channel_manager.py +++ b/src/channel_manager.py @@ -1,4 +1,5 @@ import weakref +import itertools import logging import telepathy @@ -18,53 +19,50 @@ class ChannelManager(object): self._callChannels = weakref.WeakValueDictionary() def close(self): - for chan in self._listChannels.values(): - chan.remove_from_connection()# so that dbus lets it die. - for chan in self._textChannels.values(): - chan.Close() - for chan in self._callChannels.values(): - chan.Close() + for chan in itertools.chain( + self._listChannels.values(), self._textChannels.values(), self._callChannels.values() + ): + try: + chan.close() + except Exception: + _moduleLogger.exception("Shutting down %r" % (chan, )) def channel_for_list(self, handle, suppress_handler=False): - if handle in self._listChannels: + try: chan = self._listChannels[handle] - else: - if handle.get_type() == telepathy.HANDLE_TYPE_GROUP: - chan = channel.contact_list.GroupChannel(self._connRef(), handle) - elif handle.get_type() == telepathy.HANDLE_TYPE_CONTACT: - chan = channel.contact_list.creat_contact_list_channel(self._connRef(), handle) - else: - _moduleLogger.warn("Unknown channel type %r" % handle.get_type()) + except KeyError, e: + if handle.get_type() != telepathy.HANDLE_TYPE_LIST: + raise telepathy.errors.NotImplemented("Only server lists are allowed") + _moduleLogger.debug("Requesting new contact list channel") + + chan = channel.contact_list.create_contact_list_channel(self._connRef(), handle) self._listChannels[handle] = chan self._connRef().add_channel(chan, handle, suppress_handler) return chan - def channel_for_text(self, handle, conversation=None, suppress_handler=False): - if handle in self._textChannels: + def channel_for_text(self, handle, suppress_handler=False): + try: chan = self._textChannels[handle] - else: + except KeyError, e: + if handle.get_type() != telepathy.HANDLE_TYPE_CONTACT: + raise telepathy.errors.NotImplemented("Only Contacts are allowed") _moduleLogger.debug("Requesting new text channel") - contact = handle.contact - if conversation is None: - client = self._connRef().msn_client - conversation = None - chan = channel.text.TextChannel(self._connRef(), conversation) + chan = channel.text.TextChannel(self._connRef(), handle) self._textChannels[handle] = chan self._connRef().add_channel(chan, handle, suppress_handler) return chan - def channel_forcall(self, handle, conversation=None, suppress_handler=False): - if handle in self._callChannels: + def channel_for_call(self, handle, suppress_handler=False): + try: chan = self._callChannels[handle] - else: + except KeyError, e: + if handle.get_type() != telepathy.HANDLE_TYPE_CONTACT: + _moduleLogger.warning("Using deprecated means to create a call") + raise telepathy.errors.NotImplemented("Not implementing depcrecated means") _moduleLogger.debug("Requesting new call channel") - contact = handle.contact - if conversation is None: - client = self._connRef().msn_client - conversation = None - chan = channel.call.CallChannel(self._connRef(), conversation) + chan = channel.call.CallChannel(self._connRef(), handle) self._callChannels[handle] = chan self._connRef().add_channel(chan, handle, suppress_handler) return chan