From f7a710e55bf433a378dee4bd05f6ceb59f2e9e2d Mon Sep 17 00:00:00 2001 From: Lance Stout Date: Fri, 10 Aug 2012 14:12:05 -0700 Subject: [PATCH] Add abort() method to kill the session and stop all processing without properly closing the stream. --- sleekxmpp/xmlstream/xmlstream.py | 14 ++++++++++++++ 1 file changed, 14 insertions(+) diff --git a/sleekxmpp/xmlstream/xmlstream.py b/sleekxmpp/xmlstream/xmlstream.py index 8f8e94fd..f56c360e 100644 --- a/sleekxmpp/xmlstream/xmlstream.py +++ b/sleekxmpp/xmlstream/xmlstream.py @@ -726,6 +726,20 @@ class XMLStream(object): self.event("disconnected", direct=True) return True + def abort(self): + self.session_started_event.clear() + self.stop.set() + if self._disconnect_wait_for_threads: + self._wait_for_threads() + try: + self.socket.shutdown(Socket.SHUT_RDWR) + self.socket.close() + self.filesocket.close() + except Socket.error: + pass + self.state.transition_any(['connected', 'disconnected'], 'disconnected', func=lambda: True) + self.event("killed", direct=True) + def reconnect(self, reattempt=True, wait=False, send_close=True): """Reset the stream's state and reconnect to the server.""" log.debug("reconnecting...")